sigb.us/dendrite
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases 1 Wiki Activity

Drop constraint if exists

Browse source
This commit is contained in:
Neil Alexander 2022-11-10 10:54:49 +00:00
parent 792b4142df
commit 8cac948d7e
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 14 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
userapi/storage/postgres/deltas/2022110411000000_server_names.go
View file

@ -45,19 +45,25 @@ func UpServerNames(ctx context.Context, tx *sql.Tx, serverName gomatrixserverlib
return fmt.Errorf("add server name to %q error: %w", table, err)
}
}
// Commit at this point, because it's possible that altering indices will
// fail and force us to roll back.
if err := tx.Commit(); err != nil {
return fmt.Errorf("can't commit after column addition: %w", err)
}
for newTable, oldTable := range serverNamesDropPK {
q := fmt.Sprintf(
"ALTER TABLE IF EXISTS %s DROP CONSTRAINT %s;",
"ALTER TABLE IF EXISTS %s DROP CONSTRAINT IF EXISTS %s;",
pq.QuoteIdentifier(newTable), pq.QuoteIdentifier(newTable+"_pkey"),
)
if _, err := tx.ExecContext(ctx, q); err != nil {
q = fmt.Sprintf(
"ALTER TABLE IF EXISTS %s DROP CONSTRAINT %s;",
pq.QuoteIdentifier(oldTable), pq.QuoteIdentifier(oldTable+"_pkey"),
)
if _, err := tx.ExecContext(ctx, q); err != nil {
return fmt.Errorf("drop PK from %q / %q error: %w", newTable, oldTable, err)
}
return fmt.Errorf("drop PK from %q error: %w", newTable, err)
}
q = fmt.Sprintf(
"ALTER TABLE IF EXISTS %s DROP CONSTRAINT IF EXISTS %s;",
pq.QuoteIdentifier(oldTable), pq.QuoteIdentifier(oldTable+"_pkey"),
)
if _, err := tx.ExecContext(ctx, q); err != nil {
return fmt.Errorf("drop PK from %q error: %w", oldTable, err)
}
}
return nil