mirrors/gotosocial
1
0
Fork 0
mirror of https://github.com/superseriousbusiness/gotosocial.git synced 2025-12-23 01:46:16 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions

use temporary index during migration

Browse source
This commit is contained in:
tobi 2025-03-03 17:56:53 +01:00
commit db09ad1569
1 changed files with 32 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

32
internal/db/bundb/migrations/20250224105654_token_app_client_refactor.go
View file

@ -23,11 +23,39 @@ import (
oldmodel "github.com/superseriousbusiness/gotosocial/internal/db/bundb/migrations/20211113114307_init" oldmodel "github.com/superseriousbusiness/gotosocial/internal/db/bundb/migrations/20211113114307_init"
newmodel "github.com/superseriousbusiness/gotosocial/internal/db/bundb/migrations/20250224105654_token_app_client_refactor" newmodel "github.com/superseriousbusiness/gotosocial/internal/db/bundb/migrations/20250224105654_token_app_client_refactor"
"github.com/superseriousbusiness/gotosocial/internal/id" "github.com/superseriousbusiness/gotosocial/internal/id"
"github.com/superseriousbusiness/gotosocial/internal/log"
"github.com/uptrace/bun" "github.com/uptrace/bun"
) )
func init() { func init() {
up := func(ctx context.Context, db *bun.DB) error { up := func(ctx context.Context, db *bun.DB) error {
log.Info(ctx, "migrating applications to new model")
// To make update queries faster,
// create an index on statuses
// that we'll drop when we're done.
log.Info(ctx, "creating temporary index statuses_local_created_with_application_id_idx, please wait...")
if _, err := db.
NewCreateIndex().
Table("statuses").
Index("statuses_local_created_with_application_id_idx").
Column("local", "created_with_application_id").
IfNotExists().
Exec(ctx); err != nil {
return err
}
defer func() {
log.Info(ctx, "cleaning up temporary index statuses_local_created_with_application_id_idx, please wait...")
if _, err := db.
NewDropIndex().
Index("statuses_local_created_with_application_id_idx").
IfExists().
Exec(ctx); err != nil {
panic(err)
}
}()
return db.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error { return db.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error {
// Drop unused clients table. // Drop unused clients table.
@ -89,17 +117,21 @@ func init() {
} }
if len(oldApps) != 0 { if len(oldApps) != 0 {
// Convert all the old model applications into new ones. // Convert all the old model applications into new ones.
newApps := make([]*newmodel.Application, 0, len(oldApps)) newApps := make([]*newmodel.Application, 0, len(oldApps))
for _, oldApp := range oldApps { for _, oldApp := range oldApps {
newAppID := id.NewULIDFromTime(oldApp.CreatedAt) newAppID := id.NewULIDFromTime(oldApp.CreatedAt)
log.Info(ctx, "migrating application %s (%s) to new model...", oldApp.Name, oldApp.ID)
// Update application ID on any // Update application ID on any
// statuses that used this app. // statuses that used this app.
if _, err := tx. if _, err := tx.
NewUpdate(). NewUpdate().
Table("statuses"). Table("statuses").
Set("? = ?", bun.Ident("created_with_application_id"), newAppID). Set("? = ?", bun.Ident("created_with_application_id"), newAppID).
Where("? = ?", bun.Ident("local"), true).
Where("? = ?", bun.Ident("created_with_application_id"), oldApp.ID). Where("? = ?", bun.Ident("created_with_application_id"), oldApp.ID).
Exec(ctx); err != nil { Exec(ctx); err != nil {
return err return err