mirror
/
gorm
mirror of https://github.com/go-gorm/gorm.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

support to preload all children in multiple levels associations

This commit is contained in:
ruozhixian 2021-03-11 16:36:49 +08:00
parent 912360097a
commit c575a4e719
1 changed files with 19 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
callbacks/query.go
View File

@ -185,6 +185,19 @@ func Preload(db *gorm.DB) {
} }
} else { } else {
preloadFields := strings.Split(name, ".") preloadFields := strings.Split(name, ".")
if preloadFields[0] == clause.Associations {
for _, rel := range db.Statement.Schema.Relationships.Relations {
if rel.Schema == db.Statement.Schema {
if _, ok := preloadMap[rel.Name]; !ok {
preloadMap[rel.Name] = map[string][]interface{}{}
}
if value := strings.TrimPrefix(strings.TrimPrefix(name, preloadFields[0]), "."); value != "" {
preloadMap[rel.Name][value] = db.Statement.Preloads[name]
}
}
}
} else {
if _, ok := preloadMap[preloadFields[0]]; !ok { if _, ok := preloadMap[preloadFields[0]]; !ok {
preloadMap[preloadFields[0]] = map[string][]interface{}{} preloadMap[preloadFields[0]] = map[string][]interface{}{}
} }
@ -194,6 +207,7 @@ func Preload(db *gorm.DB) {
} }
} }
} }
}
preloadNames := make([]string, 0, len(preloadMap)) preloadNames := make([]string, 0, len(preloadMap))
for key := range preloadMap { for key := range preloadMap {