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

Fix tests

Browse Source
This commit is contained in:
sgsv 2024-05-09 17:03:03 +02:00
parent 17517b05ec
commit 6e680b39ed
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
association.go
View File

@ -384,11 +384,6 @@ func (association *Association) saveAssociation(clear bool, values ...interface{
) )
appendToRelations := func(source, rv reflect.Value, clear bool) { appendToRelations := func(source, rv reflect.Value, clear bool) {
if !rv.CanAddr() {
association.Error = ErrInvalidValue
return
}
switch association.Relationship.Type { switch association.Relationship.Type {
case schema.HasOne, schema.BelongsTo: case schema.HasOne, schema.BelongsTo:
switch rv.Kind() { switch rv.Kind() {
@ -401,6 +396,10 @@ func (association *Association) saveAssociation(clear bool, values ...interface{
} }
} }
case reflect.Struct: case reflect.Struct:
if !rv.CanAddr() {
association.Error = ErrInvalidValue
return
}
association.Error = association.Relationship.Field.Set(association.DB.Statement.Context, source, rv.Addr().Interface()) association.Error = association.Relationship.Field.Set(association.DB.Statement.Context, source, rv.Addr().Interface())
if association.Relationship.Field.FieldType.Kind() == reflect.Struct { if association.Relationship.Field.FieldType.Kind() == reflect.Struct {
@ -438,6 +437,10 @@ func (association *Association) saveAssociation(clear bool, values ...interface{
appendToFieldValues(reflect.Indirect(rv.Index(i)).Addr()) appendToFieldValues(reflect.Indirect(rv.Index(i)).Addr())
} }
case reflect.Struct: case reflect.Struct:
if !rv.CanAddr() {
association.Error = ErrInvalidValue
return
}
appendToFieldValues(rv.Addr()) appendToFieldValues(rv.Addr())
} }