tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Handle Associations with pointer of pointer, close #3130

Browse Source
This commit is contained in:
Jinzhu 2020-07-10 22:53:03 +08:00
parent d4b462a351
commit 1f05cb7e55
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
association.go
View File

@ -30,7 +30,10 @@ func (db *DB) Association(column string) *Association {
association.Error = fmt.Errorf("%w: %v", ErrUnsupportedRelation, column) association.Error = fmt.Errorf("%w: %v", ErrUnsupportedRelation, column)
} }
db.Statement.ReflectValue = reflect.Indirect(reflect.ValueOf(db.Statement.Model)) db.Statement.ReflectValue = reflect.ValueOf(db.Statement.Model)
for db.Statement.ReflectValue.Kind() == reflect.Ptr {
db.Statement.ReflectValue = db.Statement.ReflectValue.Elem()
}
} else { } else {
association.Error = err association.Error = err
} }

5
tests/associations_belongs_to_test.go
View File

@ -18,7 +18,10 @@ func TestBelongsToAssociation(t *testing.T) {
// Find // Find
var user2 User var user2 User
DB.Find(&user2, "id = ?", user.ID) DB.Find(&user2, "id = ?", user.ID)
DB.Model(&user2).Association("Company").Find(&user2.Company) pointerOfUser := &user2
if err := DB.Model(&pointerOfUser).Association("Company").Find(&user2.Company); err != nil {
t.Errorf("failed to query users, got error %#v", err)
}
user2.Manager = &User{} user2.Manager = &User{}
DB.Model(&user2).Association("Manager").Find(user2.Manager) DB.Model(&user2).Association("Manager").Find(user2.Manager)
CheckUser(t, user2, user) CheckUser(t, user2, user)