tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow DisableForeignKeyConstraintWhenMigrating

Browse Source
This commit is contained in:
Jinzhu 2020-06-22 11:04:44 +08:00
parent 7851faa094
commit 5d044642d1
2 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gorm.go
View File

@ -30,6 +30,8 @@ type Config struct {
PrepareStmt bool PrepareStmt bool
// DisableAutomaticPing // DisableAutomaticPing
DisableAutomaticPing bool DisableAutomaticPing bool
// DisableForeignKeyConstraintWhenMigrating
DisableForeignKeyConstraintWhenMigrating bool
// ClauseBuilders clause builder // ClauseBuilders clause builder
ClauseBuilders map[string]clause.ClauseBuilder ClauseBuilders map[string]clause.ClauseBuilder

24
migrator/migrator.go
View File

@ -97,11 +97,13 @@ func (m Migrator) AutoMigrate(values ...interface{}) error {
} }
for _, rel := range stmt.Schema.Relationships.Relations { for _, rel := range stmt.Schema.Relationships.Relations {
if constraint := rel.ParseConstraint(); constraint != nil { if !m.DB.Config.DisableForeignKeyConstraintWhenMigrating {
if constraint.Schema == stmt.Schema { if constraint := rel.ParseConstraint(); constraint != nil {
if !tx.Migrator().HasConstraint(value, constraint.Name) { if constraint.Schema == stmt.Schema {
if err := tx.Migrator().CreateConstraint(value, constraint.Name); err != nil { if !tx.Migrator().HasConstraint(value, constraint.Name) {
return err if err := tx.Migrator().CreateConstraint(value, constraint.Name); err != nil {
return err
}
} }
} }
} }
@ -179,11 +181,13 @@ func (m Migrator) CreateTable(values ...interface{}) error {
} }
for _, rel := range stmt.Schema.Relationships.Relations { for _, rel := range stmt.Schema.Relationships.Relations {
if constraint := rel.ParseConstraint(); constraint != nil { if !m.DB.DisableForeignKeyConstraintWhenMigrating {
if constraint.Schema == stmt.Schema { if constraint := rel.ParseConstraint(); constraint != nil {
sql, vars := buildConstraint(constraint) if constraint.Schema == stmt.Schema {
createTableSQL += sql + "," sql, vars := buildConstraint(constraint)
values = append(values, vars...) createTableSQL += sql + ","
values = append(values, vars...)
}
} }
} }