mirror of
https://github.com/matrix-org/dendrite
synced 2024-09-18 08:58:56 +02:00
081f5e7226
* Add new db migration * Update migrations Remove goose * Add possibility to test direct upgrades * Try to fix WASM test * Add checks for specific migrations * Remove AddMigration Use WithTransaction Add Dendrite version to table * Fix linter issues * Update tests * Update comments, outdent if * Namespace migrations * Add direct upgrade tests, skipping over one version * Split migrations * Update go version in CI * Fix copy&paste mistake * Use contexts in migrations Co-authored-by: kegsay <kegan@matrix.org> Co-authored-by: Neil Alexander <neilalexander@users.noreply.github.com>
29 lines
964 B
Go
29 lines
964 B
Go
package deltas
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
)
|
|
|
|
func UpAddAccountType(ctx context.Context, tx *sql.Tx) error {
|
|
// initially set every account to useraccount, change appservice and guest accounts afterwards
|
|
// (user = 1, guest = 2, admin = 3, appservice = 4)
|
|
_, err := tx.ExecContext(ctx, `ALTER TABLE account_accounts ADD COLUMN IF NOT EXISTS account_type SMALLINT NOT NULL DEFAULT 1;
|
|
UPDATE account_accounts SET account_type = 4 WHERE appservice_id <> '';
|
|
UPDATE account_accounts SET account_type = 2 WHERE localpart ~ '^[0-9]+$';
|
|
ALTER TABLE account_accounts ALTER COLUMN account_type DROP DEFAULT;`,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to execute upgrade: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func DownAddAccountType(ctx context.Context, tx *sql.Tx) error {
|
|
_, err := tx.ExecContext(ctx, "ALTER TABLE account_accounts DROP COLUMN account_type;")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to execute downgrade: %w", err)
|
|
}
|
|
return nil
|
|
}
|