rows_to_map.go

package dbal

import (
	"database/sql"
)

// MappedSQLRow ...
// Improve this to not use interface{}
type MappedSQLRow map[string]interface{}

// MappedSQLRows ...
type MappedSQLRows map[int]MappedSQLRow

// RowsToMap takes the current sql.Rows and maps each column and value to a
// map[string]interface{}.
func RowsToMap(rows *sql.Rows) (MappedSQLRows, error) {
	columns, cErr := rows.Columns()
	if cErr != nil {
		return nil, cErr
	}
	count := len(columns)
	values := make([]interface{}, count)
	valuePtrs := make([]interface{}, count)

	finalResult := MappedSQLRows{}
	resultID := 0
	for rows.Next() {
		for i := range columns {
			valuePtrs[i] = &values[i]
		}
		scanErr := rows.Scan(valuePtrs...)
		if scanErr != nil {
			return nil, scanErr
		}

		tmpStruct := map[string]interface{}{}

		for i, col := range columns {
			var v interface{}
			val := values[i]
			b, ok := val.([]uint8)
			if ok {
				v = []byte(b)
			} else {
				v = val
			}
			tmpStruct[col] = v
		}

		finalResult[resultID] = tmpStruct
		resultID++
	}

	return finalResult, nil
}