|
@@ -1,6 +1,9 @@
|
|
|
package gorm
|
|
|
|
|
|
-import "reflect"
|
|
|
+import (
|
|
|
+ "reflect"
|
|
|
+ "strings"
|
|
|
+)
|
|
|
|
|
|
func beginTransactionCallback(scope *Scope) {
|
|
|
scope.Begin()
|
|
@@ -10,33 +13,79 @@ func commitOrRollbackTransactionCallback(scope *Scope) {
|
|
|
scope.CommitOrRollback()
|
|
|
}
|
|
|
|
|
|
-func saveFieldAsAssociation(scope *Scope, field *Field) (bool, *Relationship) {
|
|
|
+func saveAssociationCheck(scope *Scope, field *Field) (autoUpdate bool, autoCreate bool, saveReference bool, r *Relationship) {
|
|
|
+ checkTruth := func(value interface{}) bool {
|
|
|
+ if v, ok := value.(bool); ok && !v {
|
|
|
+ return false
|
|
|
+ }
|
|
|
+
|
|
|
+ if v, ok := value.(string); ok {
|
|
|
+ v = strings.ToLower(v)
|
|
|
+ if v == "false" || v != "skip" {
|
|
|
+ return false
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return true
|
|
|
+ }
|
|
|
+
|
|
|
if scope.changeableField(field) && !field.IsBlank && !field.IsIgnored {
|
|
|
- if field.Relationship != nil {
|
|
|
- if value, ok := field.TagSettings["SAVE_ASSOCIATIONS"]; (!ok || (value != "false" && value != "skip")) && scope.allowSaveAssociations() {
|
|
|
- return true, field.Relationship
|
|
|
+ if r = field.Relationship; r != nil {
|
|
|
+ autoUpdate, autoCreate, saveReference = true, true, true
|
|
|
+
|
|
|
+ if value, ok := scope.Get("gorm:save_associations"); ok {
|
|
|
+ autoUpdate = checkTruth(value)
|
|
|
+ autoCreate = autoUpdate
|
|
|
+ } else if value, ok := field.TagSettings["SAVE_ASSOCIATIONS"]; ok {
|
|
|
+ autoUpdate = checkTruth(value)
|
|
|
+ autoCreate = autoUpdate
|
|
|
+ }
|
|
|
+
|
|
|
+ if value, ok := scope.Get("gorm:association_autoupdate"); ok {
|
|
|
+ autoUpdate = checkTruth(value)
|
|
|
+ } else if value, ok := field.TagSettings["ASSOCIATION_AUTOUPDATE"]; ok {
|
|
|
+ autoUpdate = checkTruth(value)
|
|
|
+ }
|
|
|
+
|
|
|
+ if value, ok := scope.Get("gorm:association_autocreate"); ok {
|
|
|
+ autoCreate = checkTruth(value)
|
|
|
+ } else if value, ok := field.TagSettings["ASSOCIATION_AUTOCREATE"]; ok {
|
|
|
+ autoCreate = checkTruth(value)
|
|
|
+ }
|
|
|
+
|
|
|
+ if value, ok := field.TagSettings["ASSOCIATION_SAVE_REFERENCE"]; ok {
|
|
|
+ saveReference = checkTruth(value)
|
|
|
}
|
|
|
- return false, field.Relationship
|
|
|
}
|
|
|
}
|
|
|
- return false, nil
|
|
|
+
|
|
|
+ return
|
|
|
}
|
|
|
|
|
|
func saveBeforeAssociationsCallback(scope *Scope) {
|
|
|
for _, field := range scope.Fields() {
|
|
|
- if allowSaveAssociation, relationship := saveFieldAsAssociation(scope, field); relationship != nil && relationship.Kind == "belongs_to" {
|
|
|
+ autoUpdate, autoCreate, saveReference, relationship := saveAssociationCheck(scope, field)
|
|
|
+
|
|
|
+ if relationship != nil && relationship.Kind == "belongs_to" {
|
|
|
fieldValue := field.Field.Addr().Interface()
|
|
|
+ newScope := scope.New(fieldValue)
|
|
|
|
|
|
- if allowSaveAssociation {
|
|
|
+ if newScope.PrimaryKeyZero() {
|
|
|
+ if autoCreate {
|
|
|
+ scope.Err(scope.NewDB().Save(fieldValue).Error)
|
|
|
+ }
|
|
|
+ } else if autoUpdate {
|
|
|
scope.Err(scope.NewDB().Save(fieldValue).Error)
|
|
|
}
|
|
|
|
|
|
- if len(relationship.ForeignFieldNames) != 0 {
|
|
|
- // set value's foreign key
|
|
|
- for idx, fieldName := range relationship.ForeignFieldNames {
|
|
|
- associationForeignName := relationship.AssociationForeignDBNames[idx]
|
|
|
- if foreignField, ok := scope.New(fieldValue).FieldByName(associationForeignName); ok {
|
|
|
- scope.Err(scope.SetColumn(fieldName, foreignField.Field.Interface()))
|
|
|
+ if saveReference {
|
|
|
+ if len(relationship.ForeignFieldNames) != 0 {
|
|
|
+ // set value's foreign key
|
|
|
+ for idx, fieldName := range relationship.ForeignFieldNames {
|
|
|
+ associationForeignName := relationship.AssociationForeignDBNames[idx]
|
|
|
+ if foreignField, ok := scope.New(fieldValue).FieldByName(associationForeignName); ok {
|
|
|
+ scope.Err(scope.SetColumn(fieldName, foreignField.Field.Interface()))
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
}
|
|
@@ -46,8 +95,9 @@ func saveBeforeAssociationsCallback(scope *Scope) {
|
|
|
|
|
|
func saveAfterAssociationsCallback(scope *Scope) {
|
|
|
for _, field := range scope.Fields() {
|
|
|
- if allowSaveAssociation, relationship := saveFieldAsAssociation(scope, field); relationship != nil &&
|
|
|
- (relationship.Kind == "has_one" || relationship.Kind == "has_many" || relationship.Kind == "many_to_many") {
|
|
|
+ autoUpdate, autoCreate, saveReference, relationship := saveAssociationCheck(scope, field)
|
|
|
+
|
|
|
+ if relationship != nil && (relationship.Kind == "has_one" || relationship.Kind == "has_many" || relationship.Kind == "many_to_many") {
|
|
|
value := field.Field
|
|
|
|
|
|
switch value.Kind() {
|
|
@@ -57,44 +107,59 @@ func saveAfterAssociationsCallback(scope *Scope) {
|
|
|
elem := value.Index(i).Addr().Interface()
|
|
|
newScope := newDB.NewScope(elem)
|
|
|
|
|
|
- if relationship.JoinTableHandler == nil && len(relationship.ForeignFieldNames) != 0 {
|
|
|
- for idx, fieldName := range relationship.ForeignFieldNames {
|
|
|
- associationForeignName := relationship.AssociationForeignDBNames[idx]
|
|
|
- if f, ok := scope.FieldByName(associationForeignName); ok {
|
|
|
- scope.Err(newScope.SetColumn(fieldName, f.Field.Interface()))
|
|
|
+ if saveReference {
|
|
|
+ if relationship.JoinTableHandler == nil && len(relationship.ForeignFieldNames) != 0 {
|
|
|
+ for idx, fieldName := range relationship.ForeignFieldNames {
|
|
|
+ associationForeignName := relationship.AssociationForeignDBNames[idx]
|
|
|
+ if f, ok := scope.FieldByName(associationForeignName); ok {
|
|
|
+ scope.Err(newScope.SetColumn(fieldName, f.Field.Interface()))
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
- }
|
|
|
|
|
|
- if relationship.PolymorphicType != "" {
|
|
|
- scope.Err(newScope.SetColumn(relationship.PolymorphicType, relationship.PolymorphicValue))
|
|
|
+ if relationship.PolymorphicType != "" {
|
|
|
+ scope.Err(newScope.SetColumn(relationship.PolymorphicType, relationship.PolymorphicValue))
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
- if allowSaveAssociation {
|
|
|
+ if newScope.PrimaryKeyZero() {
|
|
|
+ if autoCreate {
|
|
|
+ scope.Err(newDB.Save(elem).Error)
|
|
|
+ }
|
|
|
+ } else if autoUpdate {
|
|
|
scope.Err(newDB.Save(elem).Error)
|
|
|
}
|
|
|
|
|
|
- if joinTableHandler := relationship.JoinTableHandler; joinTableHandler != nil && !newScope.PrimaryKeyZero() {
|
|
|
- scope.Err(joinTableHandler.Add(joinTableHandler, newDB, scope.Value, newScope.Value))
|
|
|
+ if !scope.New(newScope.Value).PrimaryKeyZero() && saveReference {
|
|
|
+ if joinTableHandler := relationship.JoinTableHandler; joinTableHandler != nil {
|
|
|
+ scope.Err(joinTableHandler.Add(joinTableHandler, newDB, scope.Value, newScope.Value))
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
default:
|
|
|
elem := value.Addr().Interface()
|
|
|
newScope := scope.New(elem)
|
|
|
- if len(relationship.ForeignFieldNames) != 0 {
|
|
|
- for idx, fieldName := range relationship.ForeignFieldNames {
|
|
|
- associationForeignName := relationship.AssociationForeignDBNames[idx]
|
|
|
- if f, ok := scope.FieldByName(associationForeignName); ok {
|
|
|
- scope.Err(newScope.SetColumn(fieldName, f.Field.Interface()))
|
|
|
+
|
|
|
+ if saveReference {
|
|
|
+ if len(relationship.ForeignFieldNames) != 0 {
|
|
|
+ for idx, fieldName := range relationship.ForeignFieldNames {
|
|
|
+ associationForeignName := relationship.AssociationForeignDBNames[idx]
|
|
|
+ if f, ok := scope.FieldByName(associationForeignName); ok {
|
|
|
+ scope.Err(newScope.SetColumn(fieldName, f.Field.Interface()))
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
- }
|
|
|
|
|
|
- if relationship.PolymorphicType != "" {
|
|
|
- scope.Err(newScope.SetColumn(relationship.PolymorphicType, relationship.PolymorphicValue))
|
|
|
+ if relationship.PolymorphicType != "" {
|
|
|
+ scope.Err(newScope.SetColumn(relationship.PolymorphicType, relationship.PolymorphicValue))
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
- if allowSaveAssociation {
|
|
|
+ if newScope.PrimaryKeyZero() {
|
|
|
+ if autoCreate {
|
|
|
+ scope.Err(scope.NewDB().Save(elem).Error)
|
|
|
+ }
|
|
|
+ } else if autoUpdate {
|
|
|
scope.Err(scope.NewDB().Save(elem).Error)
|
|
|
}
|
|
|
}
|