mirror of https://github.com/go-gorm/gorm.git
fix m2m nested preload
This commit is contained in:
parent
57c72125b3
commit
13ac0839c4
|
@ -288,7 +288,7 @@ func (scope *Scope) handleManyToManyPreload(field *Field, conditions []interface
|
||||||
// assign find results
|
// assign find results
|
||||||
var (
|
var (
|
||||||
indirectScopeValue = scope.IndirectValue()
|
indirectScopeValue = scope.IndirectValue()
|
||||||
fieldsSourceMap = map[string]reflect.Value{}
|
fieldsSourceMap = map[string][]reflect.Value{}
|
||||||
foreignFieldNames = []string{}
|
foreignFieldNames = []string{}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -301,13 +301,21 @@ func (scope *Scope) handleManyToManyPreload(field *Field, conditions []interface
|
||||||
if indirectScopeValue.Kind() == reflect.Slice {
|
if indirectScopeValue.Kind() == reflect.Slice {
|
||||||
for j := 0; j < indirectScopeValue.Len(); j++ {
|
for j := 0; j < indirectScopeValue.Len(); j++ {
|
||||||
object := indirect(indirectScopeValue.Index(j))
|
object := indirect(indirectScopeValue.Index(j))
|
||||||
fieldsSourceMap[toString(getValueFromFields(object, foreignFieldNames))] = object.FieldByName(field.Name)
|
key := toString(getValueFromFields(object, foreignFieldNames))
|
||||||
|
fieldsSourceMap[key] = append(fieldsSourceMap[key], object.FieldByName(field.Name))
|
||||||
}
|
}
|
||||||
} else if indirectScopeValue.IsValid() {
|
} else if indirectScopeValue.IsValid() {
|
||||||
fieldsSourceMap[toString(getValueFromFields(indirectScopeValue, foreignFieldNames))] = indirectScopeValue.FieldByName(field.Name)
|
key := toString(getValueFromFields(indirectScopeValue, foreignFieldNames))
|
||||||
|
fieldsSourceMap[key] = append(fieldsSourceMap[key], indirectScopeValue.FieldByName(field.Name))
|
||||||
}
|
}
|
||||||
|
|
||||||
for source, link := range linkHash {
|
for source, link := range linkHash {
|
||||||
fieldsSourceMap[source].Set(reflect.Append(fieldsSourceMap[source], link...))
|
for i, field := range fieldsSourceMap[source] {
|
||||||
|
//If not 0 this means Value is a pointer and we already added preloaded models to it
|
||||||
|
if fieldsSourceMap[source][i].Len() != 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
field.Set(reflect.Append(fieldsSourceMap[source][i], link...))
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1118,6 +1118,81 @@ func TestNestedManyToManyPreload3(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestNestedManyToManyPreload3ForStruct(t *testing.T) {
|
||||||
|
type (
|
||||||
|
Level1 struct {
|
||||||
|
ID uint
|
||||||
|
Value string
|
||||||
|
}
|
||||||
|
Level2 struct {
|
||||||
|
ID uint
|
||||||
|
Value string
|
||||||
|
Level1s []Level1 `gorm:"many2many:level1_level2;"`
|
||||||
|
}
|
||||||
|
Level3 struct {
|
||||||
|
ID uint
|
||||||
|
Value string
|
||||||
|
Level2ID sql.NullInt64
|
||||||
|
Level2 Level2
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
DB.DropTableIfExists(&Level1{})
|
||||||
|
DB.DropTableIfExists(&Level2{})
|
||||||
|
DB.DropTableIfExists(&Level3{})
|
||||||
|
DB.DropTableIfExists("level1_level2")
|
||||||
|
|
||||||
|
if err := DB.AutoMigrate(&Level3{}, &Level2{}, &Level1{}).Error; err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
level1Zh := Level1{Value: "zh"}
|
||||||
|
level1Ru := Level1{Value: "ru"}
|
||||||
|
level1En := Level1{Value: "en"}
|
||||||
|
|
||||||
|
level21 := Level2{
|
||||||
|
Value: "Level2-1",
|
||||||
|
Level1s: []Level1{level1Zh, level1Ru},
|
||||||
|
}
|
||||||
|
|
||||||
|
level22 := Level2{
|
||||||
|
Value: "Level2-2",
|
||||||
|
Level1s: []Level1{level1Zh, level1En},
|
||||||
|
}
|
||||||
|
|
||||||
|
wants := []*Level3{
|
||||||
|
{
|
||||||
|
Value: "Level3-1",
|
||||||
|
Level2: level21,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Value: "Level3-2",
|
||||||
|
Level2: level22,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Value: "Level3-3",
|
||||||
|
Level2: level21,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, want := range wants {
|
||||||
|
if err := DB.Save(&want).Error; err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var gots []*Level3
|
||||||
|
if err := DB.Preload("Level2.Level1s", func(db *gorm.DB) *gorm.DB {
|
||||||
|
return db.Order("level1.id ASC")
|
||||||
|
}).Find(&gots).Error; err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !reflect.DeepEqual(gots, wants) {
|
||||||
|
t.Errorf("got %s; want %s", toJSONString(gots), toJSONString(wants))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestNestedManyToManyPreload4(t *testing.T) {
|
func TestNestedManyToManyPreload4(t *testing.T) {
|
||||||
type (
|
type (
|
||||||
Level4 struct {
|
Level4 struct {
|
||||||
|
|
Loading…
Reference in New Issue