forked from mirror/gorm
134 lines
3.7 KiB
Go
134 lines
3.7 KiB
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)
|
|
}
|
|
}
|