add fk constraint param

Toscale-platform · Aug 16, 2024 · 43d1448 · 43d1448
1 parent 2adbc97
commit 43d1448
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/utils/gorm.go b/utils/gorm.go
@@ -9,7 +9,7 @@ import (
 	"gorm.io/gorm/schema"
 )
 
-func NewGormConn(db, dbschema, user, pswd, host string, port int) (conn *gorm.DB, err error) {
+func NewGormConn(db, dbschema, user, pswd, host string, port int, disableForeignKeyConstraintWhenMigrating bool) (conn *gorm.DB, err error) {
 
 	if len(dbschema) == 0 {
 		dbschema = "public"
@@ -22,6 +22,7 @@ func NewGormConn(db, dbschema, user, pswd, host string, port int) (conn *gorm.DB
 			TablePrefix:   fmt.Sprintf("%s.", dbschema),
 			SingularTable: false,
 		},
+		DisableForeignKeyConstraintWhenMigrating: disableForeignKeyConstraintWhenMigrating,
 	})
 
 	if err != nil {
@@ -43,4 +44,4 @@ func CloseGormConn(conn *gorm.DB) error {
 	}
 
 	return nil
-}
+}