diff options
Diffstat (limited to 'internal/db')
| -rw-r--r-- | internal/db/bundb/migrations/20240620074530_interaction_policy.go | 26 | ||||
| -rw-r--r-- | internal/db/bundb/migrations/20241011115713_pending_approval_fix.go | 80 |
2 files changed, 93 insertions, 13 deletions
diff --git a/internal/db/bundb/migrations/20240620074530_interaction_policy.go b/internal/db/bundb/migrations/20240620074530_interaction_policy.go index e45db9382..bbc75d9ec 100644 --- a/internal/db/bundb/migrations/20240620074530_interaction_policy.go +++ b/internal/db/bundb/migrations/20240620074530_interaction_policy.go @@ -40,7 +40,7 @@ func init() { table string column string columnType string - defaultVal string + extra string } for _, spec := range []spec{ // Statuses. @@ -48,19 +48,19 @@ func init() { table: "statuses", column: "interaction_policy", columnType: "JSONB", - defaultVal: "", + extra: "", }, { table: "statuses", column: "pending_approval", columnType: "BOOLEAN", - defaultVal: "DEFAULT false", + extra: "NOT NULL DEFAULT false", }, { table: "statuses", column: "approved_by_uri", columnType: "varchar", - defaultVal: "", + extra: "", }, // Status faves. @@ -68,13 +68,13 @@ func init() { table: "status_faves", column: "pending_approval", columnType: "BOOLEAN", - defaultVal: "DEFAULT false", + extra: "NOT NULL DEFAULT false", }, { table: "status_faves", column: "approved_by_uri", columnType: "varchar", - defaultVal: "", + extra: "", }, // Columns that must be added to the @@ -85,31 +85,31 @@ func init() { table: "account_settings", column: "interaction_policy_direct", columnType: "JSONB", - defaultVal: "", + extra: "", }, { table: "account_settings", column: "interaction_policy_mutuals_only", columnType: "JSONB", - defaultVal: "", + extra: "", }, { table: "account_settings", column: "interaction_policy_followers_only", columnType: "JSONB", - defaultVal: "", + extra: "", }, { table: "account_settings", column: "interaction_policy_unlocked", columnType: "JSONB", - defaultVal: "", + extra: "", }, { table: "account_settings", column: "interaction_policy_public", columnType: "JSONB", - defaultVal: "", + extra: "", }, } { exists, err := doesColumnExist(ctx, tx, @@ -130,9 +130,9 @@ func init() { } qStr := "ALTER TABLE ? ADD COLUMN ? ?" - if spec.defaultVal != "" { + if spec.extra != "" { qStr += " ?" - args = append(args, bun.Safe(spec.defaultVal)) + args = append(args, bun.Safe(spec.extra)) } log.Infof(ctx, "adding column '%s' to '%s'...", spec.column, spec.table) diff --git a/internal/db/bundb/migrations/20241011115713_pending_approval_fix.go b/internal/db/bundb/migrations/20241011115713_pending_approval_fix.go new file mode 100644 index 000000000..c6d3e288f --- /dev/null +++ b/internal/db/bundb/migrations/20241011115713_pending_approval_fix.go @@ -0,0 +1,80 @@ +// GoToSocial +// Copyright (C) GoToSocial Authors admin@gotosocial.org +// SPDX-License-Identifier: AGPL-3.0-or-later +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU Affero General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU Affero General Public License for more details. +// +// You should have received a copy of the GNU Affero General Public License +// along with this program. If not, see <http://www.gnu.org/licenses/>. + +package migrations + +import ( + "context" + + "github.com/superseriousbusiness/gotosocial/internal/log" + "github.com/uptrace/bun" +) + +func init() { + up := func(ctx context.Context, db *bun.DB) error { + return db.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error { + // Previous versions of 20240620074530_interaction_policy.go + // didn't set NOT NULL on gtsmodel.Status.PendingApproval and + // gtsmodel.StatusFave.PendingApproval, resulting in NULL being + // set for that column for some statuses. Correct for this. + + log.Info(ctx, "correcting pending_approval on statuses table...") + res, err := tx. + NewUpdate(). + Table("statuses"). + Set("? = ?", bun.Ident("pending_approval"), false). + Where("? IS NULL", bun.Ident("pending_approval")). + Exec(ctx) + if err != nil { + return err + } + + rows, err := res.RowsAffected() + if err == nil { + log.Infof(ctx, "corrected %d entries", rows) + } + + log.Info(ctx, "correcting pending_approval on status_faves table...") + res, err = tx. + NewUpdate(). + Table("status_faves"). + Set("? = ?", bun.Ident("pending_approval"), false). + Where("? IS NULL", bun.Ident("pending_approval")). + Exec(ctx) + if err != nil { + return err + } + + rows, err = res.RowsAffected() + if err == nil { + log.Infof(ctx, "corrected %d entries", rows) + } + + return nil + }) + } + + down := func(ctx context.Context, db *bun.DB) error { + return db.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error { + return nil + }) + } + + if err := Migrations.Register(up, down); err != nil { + panic(err) + } +} |