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
// DisableAutomaticPing
DisableAutomaticPing bool
// DisableForeignKeyConstraintWhenMigrating
DisableForeignKeyConstraintWhenMigrating bool
// ClauseBuilders clause builder
ClauseBuilders map[string]clause.ClauseBuilder

4
migrator/migrator.go
View File

@ -97,6 +97,7 @@ func (m Migrator) AutoMigrate(values ...interface{}) error {
}
for _, rel := range stmt.Schema.Relationships.Relations {
if !m.DB.Config.DisableForeignKeyConstraintWhenMigrating {
if constraint := rel.ParseConstraint(); constraint != nil {
if constraint.Schema == stmt.Schema {
if !tx.Migrator().HasConstraint(value, constraint.Name) {
@ -106,6 +107,7 @@ func (m Migrator) AutoMigrate(values ...interface{}) error {
}
}
}
}
for _, chk := range stmt.Schema.ParseCheckConstraints() {
if !tx.Migrator().HasConstraint(value, chk.Name) {
@ -179,6 +181,7 @@ func (m Migrator) CreateTable(values ...interface{}) error {
}
for _, rel := range stmt.Schema.Relationships.Relations {
if !m.DB.DisableForeignKeyConstraintWhenMigrating {
if constraint := rel.ParseConstraint(); constraint != nil {
if constraint.Schema == stmt.Schema {
sql, vars := buildConstraint(constraint)
@ -186,6 +189,7 @@ func (m Migrator) CreateTable(values ...interface{}) error {
values = append(values, vars...)
}
}
}
// create join table
if rel.JoinTable != nil {