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

Allows foreign keys to be saved without saving the assoication when specified #1628

This commit is contained in:
joe-at-startupmedia 2017-10-04 08:19:16 +00:00 committed by Jinzhu
parent aa3fd6de13
commit 9235b47ea2
1 changed files with 26 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
callback_save.go
View File

@ -11,35 +11,34 @@ func commitOrRollbackTransactionCallback(scope *Scope) {
} }
func saveFieldAsAssociation(scope *Scope, field *Field) (bool, *Relationship) { func saveFieldAsAssociation(scope *Scope, field *Field) (bool, *Relationship) {
if scope.changeableField(field) && !field.IsBlank && !field.IsIgnored { if scope.changeableField(field) && !field.IsBlank && !field.IsIgnored {
if value, ok := field.TagSettings["SAVE_ASSOCIATIONS"]; !ok || (value != "false" && value != "skip") { if value, ok := field.TagSettings["SAVE_ASSOCIATIONS"]; !ok || (value != "false" && value != "skip") {
if relationship := field.Relationship; relationship != nil { return true, field.Relationship
return true, relationship }
} return false, field.Relationship
} }
} return false, nil
return false, nil
} }
func saveBeforeAssociationsCallback(scope *Scope) { func saveBeforeAssociationsCallback(scope *Scope) {
if !scope.shouldSaveAssociations() { for _, field := range scope.Fields() {
return ok, relationship := saveFieldAsAssociation(scope, field);
} if relationship != nil && relationship.Kind == "belongs_to" {
for _, field := range scope.Fields() { fieldValue := field.Field.Addr().Interface()
if ok, relationship := saveFieldAsAssociation(scope, field); ok && relationship.Kind == "belongs_to" { if ok && scope.shouldSaveAssociations() {
fieldValue := field.Field.Addr().Interface() scope.Err(scope.NewDB().Save(fieldValue).Error)
scope.Err(scope.NewDB().Save(fieldValue).Error) }
if len(relationship.ForeignFieldNames) != 0 { if len(relationship.ForeignFieldNames) != 0 {
// set value's foreign key // set value's foreign key
for idx, fieldName := range relationship.ForeignFieldNames { for idx, fieldName := range relationship.ForeignFieldNames {
associationForeignName := relationship.AssociationForeignDBNames[idx] associationForeignName := relationship.AssociationForeignDBNames[idx]
if foreignField, ok := scope.New(fieldValue).FieldByName(associationForeignName); ok { if foreignField, ok := scope.New(fieldValue).FieldByName(associationForeignName); ok {
scope.Err(scope.SetColumn(fieldName, foreignField.Field.Interface())) scope.Err(scope.SetColumn(fieldName, foreignField.Field.Interface()))
} }
} }
} }
} }
} }
} }
func saveAfterAssociationsCallback(scope *Scope) { func saveAfterAssociationsCallback(scope *Scope) {
@ -47,7 +46,7 @@ func saveAfterAssociationsCallback(scope *Scope) {
return return
} }
for _, field := range scope.Fields() { for _, field := range scope.Fields() {
if ok, relationship := saveFieldAsAssociation(scope, field); ok && if ok, relationship := saveFieldAsAssociation(scope, field); ok && relationship != nil &&
(relationship.Kind == "has_one" || relationship.Kind == "has_many" || relationship.Kind == "many_to_many") { (relationship.Kind == "has_one" || relationship.Kind == "has_many" || relationship.Kind == "many_to_many") {
value := field.Field value := field.Field