Drop FK constraints if they already exist before creating (#1534)

google · Jan 8, 2021 · 1b0a580 · 1b0a580
1 parent f2b0194
commit 1b0a580
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/database/migrations.go b/pkg/database/migrations.go
@@ -1764,9 +1764,15 @@ func (db *Database) Migrations(ctx context.Context) []*gormigrate.Migration {
 
 				// Convert realm and user into real foreign keys. If the parent record
 				// is deleted, also delete the membership record.
+				if err := tx.Exec(`ALTER TABLE user_realms DROP CONSTRAINT IF EXISTS fk_realm`).Error; err != nil {
+					return err
+				}
 				if err := tx.Exec(`ALTER TABLE user_realms ADD CONSTRAINT fk_realm FOREIGN KEY (realm_id) REFERENCES realms(id) ON DELETE CASCADE`).Error; err != nil {
 					return err
 				}
+				if err := tx.Exec(`ALTER TABLE user_realms DROP CONSTRAINT IF EXISTS fk_user`).Error; err != nil {
+					return err
+				}
 				if err := tx.Exec(`ALTER TABLE user_realms ADD CONSTRAINT fk_user FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE`).Error; err != nil {
 					return err
 				}