forked from mirror/gorm
parent
3d87575e7e
commit
6095dbf939
|
@ -667,13 +667,15 @@ func (m Migrator) ReorderModels(values []interface{}, autoAdd bool) (results []i
|
||||||
}
|
}
|
||||||
orderedModelNamesMap[name] = true
|
orderedModelNamesMap[name] = true
|
||||||
|
|
||||||
dep := valuesMap[name]
|
if autoAdd {
|
||||||
for _, d := range dep.Depends {
|
dep := valuesMap[name]
|
||||||
if _, ok := valuesMap[d.Table]; ok {
|
for _, d := range dep.Depends {
|
||||||
insertIntoOrderedList(d.Table)
|
if _, ok := valuesMap[d.Table]; ok {
|
||||||
} else if autoAdd {
|
insertIntoOrderedList(d.Table)
|
||||||
parseDependence(reflect.New(d.ModelType).Interface(), autoAdd)
|
} else {
|
||||||
insertIntoOrderedList(d.Table)
|
parseDependence(reflect.New(d.ModelType).Interface(), autoAdd)
|
||||||
|
insertIntoOrderedList(d.Table)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -53,7 +53,7 @@ type Reference struct {
|
||||||
OwnPrimaryKey bool
|
OwnPrimaryKey bool
|
||||||
}
|
}
|
||||||
|
|
||||||
func (schema *Schema) parseRelation(field *Field) {
|
func (schema *Schema) parseRelation(field *Field) *Relationship {
|
||||||
var (
|
var (
|
||||||
err error
|
err error
|
||||||
fieldValue = reflect.New(field.IndirectFieldType).Interface()
|
fieldValue = reflect.New(field.IndirectFieldType).Interface()
|
||||||
|
@ -67,13 +67,10 @@ func (schema *Schema) parseRelation(field *Field) {
|
||||||
)
|
)
|
||||||
|
|
||||||
cacheStore := schema.cacheStore
|
cacheStore := schema.cacheStore
|
||||||
if field.OwnerSchema != nil {
|
|
||||||
cacheStore = field.OwnerSchema.cacheStore
|
|
||||||
}
|
|
||||||
|
|
||||||
if relation.FieldSchema, err = getOrParse(fieldValue, cacheStore, schema.namer); err != nil {
|
if relation.FieldSchema, err = getOrParse(fieldValue, cacheStore, schema.namer); err != nil {
|
||||||
schema.err = err
|
schema.err = err
|
||||||
return
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if polymorphic := field.TagSettings["POLYMORPHIC"]; polymorphic != "" {
|
if polymorphic := field.TagSettings["POLYMORPHIC"]; polymorphic != "" {
|
||||||
|
@ -92,7 +89,8 @@ func (schema *Schema) parseRelation(field *Field) {
|
||||||
}
|
}
|
||||||
|
|
||||||
if relation.Type == "has" {
|
if relation.Type == "has" {
|
||||||
if relation.FieldSchema != relation.Schema && relation.Polymorphic == nil {
|
// don't add relations to embeded schema, which might be shared
|
||||||
|
if relation.FieldSchema != relation.Schema && relation.Polymorphic == nil && field.OwnerSchema == nil {
|
||||||
relation.FieldSchema.Relationships.Relations["_"+relation.Schema.Name+"_"+relation.Name] = relation
|
relation.FieldSchema.Relationships.Relations["_"+relation.Schema.Name+"_"+relation.Name] = relation
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -117,6 +115,8 @@ func (schema *Schema) parseRelation(field *Field) {
|
||||||
schema.Relationships.Many2Many = append(schema.Relationships.Many2Many, relation)
|
schema.Relationships.Many2Many = append(schema.Relationships.Many2Many, relation)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return relation
|
||||||
}
|
}
|
||||||
|
|
||||||
// User has many Toys, its `Polymorphic` is `Owner`, Pet has one Toy, its `Polymorphic` is `Owner`
|
// User has many Toys, its `Polymorphic` is `Owner`, Pet has one Toy, its `Polymorphic` is `Owner`
|
||||||
|
|
|
@ -397,3 +397,39 @@ func TestMultipleMany2Many(t *testing.T) {
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type CreatedByModel struct {
|
||||||
|
CreatedByID uint
|
||||||
|
CreatedBy *CreatedUser
|
||||||
|
}
|
||||||
|
|
||||||
|
type CreatedUser struct {
|
||||||
|
gorm.Model
|
||||||
|
CreatedByModel
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEmbeddedRelation(t *testing.T) {
|
||||||
|
checkStructRelation(t, &CreatedUser{}, Relation{
|
||||||
|
Name: "CreatedBy", Type: schema.BelongsTo, Schema: "CreatedUser", FieldSchema: "CreatedUser",
|
||||||
|
References: []Reference{
|
||||||
|
{"ID", "CreatedUser", "CreatedByID", "CreatedUser", "", false},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
userSchema, err := schema.Parse(&CreatedUser{}, &sync.Map{}, schema.NamingStrategy{})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to parse schema, got error %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(userSchema.Relationships.Relations) != 1 {
|
||||||
|
t.Fatalf("expects 1 relations, but got %v", len(userSchema.Relationships.Relations))
|
||||||
|
}
|
||||||
|
|
||||||
|
if createdByRel, ok := userSchema.Relationships.Relations["CreatedBy"]; ok {
|
||||||
|
if createdByRel.FieldSchema != userSchema {
|
||||||
|
t.Fatalf("expects same field schema, but got new %p, old %p", createdByRel.FieldSchema, userSchema)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
t.Fatalf("expects created by relations, but not found")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue