gtsocial-umbx

commit a0068e89158b67dfb790c77e875400a6772d1890
parent a684fc46288e38a7e1d3555cb9a5a48b8c65b634
Author: tobi <31960611+tsmethurst@users.noreply.github.com>
Date:   Sat, 18 Feb 2023 17:54:51 +0100

[bugfix] In Postgres, drop shortcodedomain constraint before creating new emoji table (#1528)


Diffstat:
M
internal/db/bundb/migrations/20220905150505_custom_emoji_updates.go
 | 
12
+++++++++++-

1 file changed, 11 insertions(+), 1 deletion(-)
diff --git a/internal/db/bundb/migrations/20220905150505_custom_emoji_updates.go b/internal/db/bundb/migrations/20220905150505_custom_emoji_updates.go
@@ -24,11 +24,21 @@ import (
 
 	gtsmodel "github.com/superseriousbusiness/gotosocial/internal/db/bundb/migrations/20211113114307_init"
 	"github.com/uptrace/bun"
+	"github.com/uptrace/bun/dialect"
 )
 
 func init() {
 	up := func(ctx context.Context, db *bun.DB) error {
 		return db.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error {
+			// SQLite doesn't mind creating multiple constraints with the same name,
+			// but Postgres balks at it, so remove the constraint before we go editing
+			// the emoji tables.
+			if tx.Dialect().Name() == dialect.PG {
+				if _, err := tx.ExecContext(ctx, "ALTER TABLE ? DROP CONSTRAINT ?", bun.Ident("emojis"), bun.Safe("shortcodedomain")); err != nil {
+					return err
+				}
+			}
+
 			// create the new emojis table
 			if _, err := tx.
 				NewCreateTable().
@@ -63,7 +73,7 @@ func init() {
 			}
 
 			// rename the new table to the same name as the old table was
-			if _, err := tx.ExecContext(ctx, "ALTER TABLE new_emojis RENAME TO emojis;"); err != nil {
+			if _, err := tx.ExecContext(ctx, "ALTER TABLE ? RENAME TO ?", bun.Ident("new_emojis"), bun.Ident("emojis")); err != nil {
 				return err
 			}