gorm/callbacks/scan.go

package callbacks

import (
	"database/sql"
	"reflect"
	"strings"

	"github.com/jinzhu/gorm"
	"github.com/jinzhu/gorm/schema"
)

func Scan(rows *sql.Rows, db *gorm.DB) {
	columns, _ := rows.Columns()
	values := make([]interface{}, len(columns))

	switch dest := db.Statement.Dest.(type) {
	case map[string]interface{}, *map[string]interface{}:
		for idx, _ := range columns {
			values[idx] = new(interface{})
		}

		if rows.Next() {
			db.RowsAffected++
			rows.Scan(values...)
		}

		mapValue, ok := dest.(map[string]interface{})
		if ok {
			if v, ok := dest.(*map[string]interface{}); ok {
				mapValue = *v
			}
		}

		for idx, column := range columns {
			mapValue[column] = *(values[idx].(*interface{}))
		}
	case *[]map[string]interface{}:
		for idx, _ := range columns {
			values[idx] = new(interface{})
		}

		for rows.Next() {
			db.RowsAffected++
			rows.Scan(values...)

			v := map[string]interface{}{}
			for idx, column := range columns {
				v[column] = *(values[idx].(*interface{}))
			}
			*dest = append(*dest, v)
		}
	case *int, *int64, *uint, *uint64:
		for rows.Next() {
			db.RowsAffected++
			rows.Scan(dest)
		}
	default:
		switch db.Statement.ReflectValue.Kind() {
		case reflect.Slice, reflect.Array:
			isPtr := db.Statement.ReflectValue.Type().Elem().Kind() == reflect.Ptr
			db.Statement.ReflectValue.Set(reflect.MakeSlice(db.Statement.ReflectValue.Type(), 0, 0))
			fields := make([]*schema.Field, len(columns))
			joinFields := make([][2]*schema.Field, len(columns))

			for idx, column := range columns {
				if field := db.Statement.Schema.LookUpField(column); field != nil && field.Readable {
					fields[idx] = field
				} else if names := strings.Split(column, "__"); len(names) > 1 {
					if rel, ok := db.Statement.Schema.Relationships.Relations[names[0]]; ok {
						if field := rel.FieldSchema.LookUpField(strings.Join(names[1:], "__")); field != nil && field.Readable {
							joinFields[idx] = [2]*schema.Field{rel.Field, field}
							continue
						}
					}
					values[idx] = &sql.RawBytes{}
				} else {
					values[idx] = &sql.RawBytes{}
				}
			}

			for rows.Next() {
				elem := reflect.New(db.Statement.Schema.ModelType).Elem()
				for idx, field := range fields {
					if field != nil {
						values[idx] = field.ReflectValueOf(elem).Addr().Interface()
					} else if joinFields[idx][0] != nil {
						relValue := joinFields[idx][0].ReflectValueOf(elem)
						values[idx] = joinFields[idx][1].ReflectValueOf(relValue).Addr().Interface()
					}
				}

				db.RowsAffected++
				if err := rows.Scan(values...); err != nil {
					db.AddError(err)
				}

				if isPtr {
					db.Statement.ReflectValue.Set(reflect.Append(db.Statement.ReflectValue, elem.Addr()))
				} else {
					db.Statement.ReflectValue.Set(reflect.Append(db.Statement.ReflectValue, elem))
				}
			}
		case reflect.Struct:
			for idx, column := range columns {
				if field := db.Statement.Schema.LookUpField(column); field != nil && field.Readable {
					values[idx] = field.ReflectValueOf(db.Statement.ReflectValue).Addr().Interface()
				} else if names := strings.Split(column, "__"); len(names) > 1 {
					if rel, ok := db.Statement.Schema.Relationships.Relations[names[0]]; ok {
						relValue := rel.Field.ReflectValueOf(db.Statement.ReflectValue)
						if field := rel.FieldSchema.LookUpField(strings.Join(names[1:], "__")); field != nil && field.Readable {
							values[idx] = field.ReflectValueOf(relValue).Addr().Interface()
							continue
						}
					}
					values[idx] = &sql.RawBytes{}
				} else {
					values[idx] = &sql.RawBytes{}
				}
			}

			if rows.Next() {
				db.RowsAffected++
				if err := rows.Scan(values...); err != nil {
					db.AddError(err)
				}
			}
		}
	}

	if db.RowsAffected == 0 && db.Statement.RaiseErrorOnNotFound {
		db.AddError(gorm.ErrRecordNotFound)
	}
}