forked from mirror/gorm
Optimize: code logic db.scanIntoStruct() (#5633)
This commit is contained in:
parent
d71caef7d9
commit
f78f635fae
46
scan.go
46
scan.go
|
@ -66,30 +66,32 @@ func (db *DB) scanIntoStruct(rows Rows, reflectValue reflect.Value, values []int
|
||||||
db.RowsAffected++
|
db.RowsAffected++
|
||||||
db.AddError(rows.Scan(values...))
|
db.AddError(rows.Scan(values...))
|
||||||
|
|
||||||
joinedSchemaMap := make(map[*schema.Field]interface{}, 0)
|
joinedSchemaMap := make(map[*schema.Field]interface{})
|
||||||
for idx, field := range fields {
|
for idx, field := range fields {
|
||||||
if field != nil {
|
if field == nil {
|
||||||
if len(joinFields) == 0 || joinFields[idx][0] == nil {
|
continue
|
||||||
db.AddError(field.Set(db.Statement.Context, reflectValue, values[idx]))
|
|
||||||
} else {
|
|
||||||
joinSchema := joinFields[idx][0]
|
|
||||||
relValue := joinSchema.ReflectValueOf(db.Statement.Context, reflectValue)
|
|
||||||
if relValue.Kind() == reflect.Ptr {
|
|
||||||
if _, ok := joinedSchemaMap[joinSchema]; !ok {
|
|
||||||
if value := reflect.ValueOf(values[idx]).Elem(); value.Kind() == reflect.Ptr && value.IsNil() {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
relValue.Set(reflect.New(relValue.Type().Elem()))
|
|
||||||
joinedSchemaMap[joinSchema] = nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
db.AddError(joinFields[idx][1].Set(db.Statement.Context, relValue, values[idx]))
|
|
||||||
}
|
|
||||||
|
|
||||||
// release data to pool
|
|
||||||
field.NewValuePool.Put(values[idx])
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if len(joinFields) == 0 || joinFields[idx][0] == nil {
|
||||||
|
db.AddError(field.Set(db.Statement.Context, reflectValue, values[idx]))
|
||||||
|
} else {
|
||||||
|
joinSchema := joinFields[idx][0]
|
||||||
|
relValue := joinSchema.ReflectValueOf(db.Statement.Context, reflectValue)
|
||||||
|
if relValue.Kind() == reflect.Ptr {
|
||||||
|
if _, ok := joinedSchemaMap[joinSchema]; !ok {
|
||||||
|
if value := reflect.ValueOf(values[idx]).Elem(); value.Kind() == reflect.Ptr && value.IsNil() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
relValue.Set(reflect.New(relValue.Type().Elem()))
|
||||||
|
joinedSchemaMap[joinSchema] = nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
db.AddError(joinFields[idx][1].Set(db.Statement.Context, relValue, values[idx]))
|
||||||
|
}
|
||||||
|
|
||||||
|
// release data to pool
|
||||||
|
field.NewValuePool.Put(values[idx])
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue