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

Add tag to support skip nested save for associations

This commit is contained in:
Jinzhu 2016-10-27 10:31:46 +08:00
parent 56a7d1b69e
commit 4a540f3ac8
3 changed files with 80 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
association_test.go
View File

@ -884,3 +884,24 @@ func TestHasManyChildrenWithOneStruct(t *testing.T) {
DB.Save(&category) DB.Save(&category)
} }
func TestSkipSaveAssociation(t *testing.T) {
type Company struct {
gorm.Model
Name string
}
type User struct {
gorm.Model
Name string
CompanyID uint
Company Company `gorm:"save_associations:false"`
}
DB.AutoMigrate(&Company{}, &User{})
DB.Save(&User{Name: "jinzhu", Company: Company{Name: "skip_save_association"}})
if !DB.Where("name = ?", "skip_save_association").First(&Company{}).RecordNotFound() {
t.Errorf("Company skip_save_association should not been saved")
}
}

19
callback_save.go
View File

@ -10,13 +10,23 @@ func commitOrRollbackTransactionCallback(scope *Scope) {
scope.CommitOrRollback() scope.CommitOrRollback()
} }
func saveFieldAsAssociation(scope *Scope, field *Field) (bool, *Relationship) {
if scope.changeableField(field) && !field.IsBlank && !field.IsIgnored {
if value, ok := field.TagSettings["SAVE_ASSOCIATIONS"]; !ok || (value != "false" && value != "skip") {
if relationship := field.Relationship; relationship != nil {
return true, relationship
}
}
}
return false, nil
}
func saveBeforeAssociationsCallback(scope *Scope) { func saveBeforeAssociationsCallback(scope *Scope) {
if !scope.shouldSaveAssociations() { if !scope.shouldSaveAssociations() {
return return
} }
for _, field := range scope.Fields() { for _, field := range scope.Fields() {
if scope.changeableField(field) && !field.IsBlank && !field.IsIgnored { if ok, relationship := saveFieldAsAssociation(scope, field); ok && relationship.Kind == "belongs_to" {
if relationship := field.Relationship; relationship != nil && relationship.Kind == "belongs_to" {
fieldValue := field.Field.Addr().Interface() 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 {
@ -30,7 +40,6 @@ func saveBeforeAssociationsCallback(scope *Scope) {
} }
} }
} }
}
} }
func saveAfterAssociationsCallback(scope *Scope) { func saveAfterAssociationsCallback(scope *Scope) {
@ -38,8 +47,7 @@ func saveAfterAssociationsCallback(scope *Scope) {
return return
} }
for _, field := range scope.Fields() { for _, field := range scope.Fields() {
if scope.changeableField(field) && !field.IsBlank && !field.IsIgnored { if ok, relationship := saveFieldAsAssociation(scope, field); ok &&
if relationship := field.Relationship; 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
@ -88,5 +96,4 @@ func saveAfterAssociationsCallback(scope *Scope) {
} }
} }
} }
}
} }

7
scope.go
View File

@ -964,9 +964,14 @@ func (scope *Scope) changeableField(field *Field) bool {
} }
func (scope *Scope) shouldSaveAssociations() bool { func (scope *Scope) shouldSaveAssociations() bool {
if saveAssociations, ok := scope.Get("gorm:save_associations"); ok && !saveAssociations.(bool) { if saveAssociations, ok := scope.Get("gorm:save_associations"); ok {
if v, ok := saveAssociations.(bool); ok && !v {
return false return false
} }
if v, ok := saveAssociations.(string); ok && (v != "skip") {
return false
}
}
return true && !scope.HasError() return true && !scope.HasError()
} }