Add Delete support for has many associations

This commit is contained in:
Jinzhu 2015-12-25 18:42:06 +08:00
parent a00cb7432c
commit a00fb4db04
2 changed files with 75 additions and 1 deletions

View File

@ -136,6 +136,10 @@ func (association *Association) Delete(values ...interface{}) *Association {
query := scope.NewDB() query := scope.NewDB()
relationship := association.Field.Relationship relationship := association.Field.Relationship
if len(values) == 0 {
return association
}
// many to many // many to many
if relationship.Kind == "many_to_many" { if relationship.Kind == "many_to_many" {
// current value's foreign keys // current value's foreign keys
@ -178,8 +182,27 @@ func (association *Association) Delete(values ...interface{}) *Association {
primaryKeys := association.getPrimaryKeys(relationship.AssociationForeignFieldNames, values...) primaryKeys := association.getPrimaryKeys(relationship.AssociationForeignFieldNames, values...)
sql := fmt.Sprintf("%v IN (%v)", toQueryCondition(scope, relationship.ForeignDBNames), toQueryMarks(primaryKeys)) sql := fmt.Sprintf("%v IN (%v)", toQueryCondition(scope, relationship.ForeignDBNames), toQueryMarks(primaryKeys))
association.setErr(query.Model(scope.Value).Where(sql, toQueryValues(primaryKeys)...).UpdateColumn(foreignKeyMap).Error) association.setErr(query.Model(scope.Value).Where(sql, toQueryValues(primaryKeys)...).UpdateColumn(foreignKeyMap).Error)
} else if relationship.Kind == "has_one" || relationship.Kind == "has_many" { } else if relationship.Kind == "has_one" || relationship.Kind == "has_many" {
// query.Model(association.Field).UpdateColumn(foreignKeyMap) var foreignKeyMap = map[string]interface{}{}
for _, foreignKey := range relationship.ForeignDBNames {
foreignKeyMap[foreignKey] = nil
}
primaryKeys := association.getPrimaryKeys(relationship.AssociationForeignFieldNames, scope.Value)
sql := fmt.Sprintf("%v IN (%v)", toQueryCondition(scope, relationship.ForeignDBNames), toQueryMarks(primaryKeys))
var primaryFieldNames, primaryFieldDBNames []string
for _, field := range scope.New(values[0]).Fields() {
if field.IsPrimaryKey {
primaryFieldNames = append(primaryFieldNames, field.Name)
primaryFieldDBNames = append(primaryFieldDBNames, field.DBName)
}
}
relationsPrimaryKeys := association.getPrimaryKeys(primaryFieldNames, values...)
sql += fmt.Sprintf(" AND %v IN (%v)", toQueryCondition(scope, primaryFieldDBNames), toQueryMarks(relationsPrimaryKeys))
query.Model(association.Field.Field.Interface()).Where(sql, append(toQueryValues(primaryKeys), toQueryValues(relationsPrimaryKeys)...)...).UpdateColumn(foreignKeyMap)
} }
} }
return association return association

View File

@ -2,6 +2,8 @@ package gorm_test
import ( import (
"fmt" "fmt"
"reflect"
"sort"
"testing" "testing"
) )
@ -125,9 +127,58 @@ func TestHasMany(t *testing.T) {
t.Errorf("Got errors when save post", err.Error()) t.Errorf("Got errors when save post", err.Error())
} }
for _, comment := range post.Comments {
if comment.PostId == 0 {
t.Errorf("comment's PostID should be updated")
}
}
var compareComments = func(comments []Comment, contents []string) bool {
var commentContents []string
for _, comment := range comments {
commentContents = append(commentContents, comment.Content)
}
sort.Strings(commentContents)
sort.Strings(contents)
return reflect.DeepEqual(commentContents, contents)
}
// Query // Query
if DB.First(&Comment{}, "content = ?", "Comment 1").Error != nil {
t.Errorf("Comment 1 should be saved")
}
var comments1 []Comment
DB.Model(&post).Association("Comments").Find(&comments1)
if !compareComments(comments1, []string{"Comment 1", "Comment 2"}) {
t.Errorf("Query has many relations with Association")
}
var comments11 []Comment
DB.Model(&post).Related(&comments11)
if !compareComments(comments11, []string{"Comment 1", "Comment 2"}) {
t.Errorf("Query has many relations with Related")
}
// Append // Append
DB.Model(&post).Association("Comments").Append(&Comment{Content: "Comment 3"})
var comments2 []Comment
DB.Model(&post).Related(&comments2)
if !compareComments(comments2, []string{"Comment 1", "Comment 2", "Comment 3"}) {
t.Errorf("Append new record to has many relations")
}
// Delete // Delete
DB.Model(&post).Association("Comments").Delete(comments11)
var comments3 []Comment
DB.Model(&post).Related(&comments3)
if !compareComments(comments3, []string{"Comment 3"}) {
fmt.Println(comments3)
t.Errorf("Delete an existing resource for has many relations")
}
// Replace // Replace
// Clear // Clear
} }