diff options
author | Pavlo Strokov <pstrokov@gitlab.com> | 2020-08-10 09:43:30 +0300 |
---|---|---|
committer | Pavlo Strokov <pstrokov@gitlab.com> | 2020-09-02 17:54:39 +0300 |
commit | f2402e2772c5067a46ffc70ea137404d89795bb1 (patch) | |
tree | 124da71f751be22623575055fd4884116400a130 /internal/praefect | |
parent | 8f8ae3025cb6ee9800182f41de86208a608958d2 (diff) |
Replication job acknowledge removes 'completed' and 'dead' events.
Migration script to remove all redundant rows from replication_queue.
It is safe to re-run migration if it fails. It is designed to remove
rows in batches by 100000 each iteration. Each removal happens in
separate transaction. The script removes only 'dead' and 'completed'
replication events.
Closes: https://gitlab.com/gitlab-org/gitaly/-/issues/2941
Diffstat (limited to 'internal/praefect')
-rw-r--r-- | internal/praefect/datastore/migrations/20200810055650_replication_queue_cleanup.go | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/internal/praefect/datastore/migrations/20200810055650_replication_queue_cleanup.go b/internal/praefect/datastore/migrations/20200810055650_replication_queue_cleanup.go new file mode 100644 index 000000000..4dcf2fed4 --- /dev/null +++ b/internal/praefect/datastore/migrations/20200810055650_replication_queue_cleanup.go @@ -0,0 +1,39 @@ +package migrations + +import migrate "github.com/rubenv/sql-migrate" + +func init() { + m := &migrate.Migration{ + Id: "20200810055650_replication_queue_cleanup", + Up: []string{ + `CREATE TEMP TABLE ids(id BIGINT)`, + ` + -- +migrate StatementBegin + DO $$DECLARE + found_amount BIGINT; + BEGIN + LOOP + TRUNCATE TABLE ids; + INSERT INTO ids + SELECT id + FROM replication_queue + WHERE state = ANY (ARRAY['dead'::REPLICATION_JOB_STATE, 'completed'::REPLICATION_JOB_STATE]) + LIMIT 100000; + SELECT COUNT(*) INTO found_amount FROM ids; + IF found_amount > 0 THEN + DELETE FROM replication_queue WHERE id IN (SELECT id FROM ids); + COMMIT; + ELSE + RETURN; + END IF; + END LOOP; + END$$; + -- +migrate StatementEnd`, + `DROP TABLE ids`, + }, + Down: []string{}, + DisableTransactionUp: true, + } + + allMigrations = append(allMigrations, m) +} |