diff options
Diffstat (limited to 'db/post_migrate/20220920081631_prepare_ci_builds_metadata_for_partitioning_primary_key.rb')
-rw-r--r-- | db/post_migrate/20220920081631_prepare_ci_builds_metadata_for_partitioning_primary_key.rb | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/db/post_migrate/20220920081631_prepare_ci_builds_metadata_for_partitioning_primary_key.rb b/db/post_migrate/20220920081631_prepare_ci_builds_metadata_for_partitioning_primary_key.rb new file mode 100644 index 00000000000..90727178624 --- /dev/null +++ b/db/post_migrate/20220920081631_prepare_ci_builds_metadata_for_partitioning_primary_key.rb @@ -0,0 +1,36 @@ +# frozen_string_literal: true + +class PrepareCiBuildsMetadataForPartitioningPrimaryKey < Gitlab::Database::Migration[2.0] + disable_ddl_transaction! + + TABLE_NAME = 'ci_builds_metadata' + PRIMARY_KEY = 'ci_builds_metadata_pkey' + NEW_INDEX_NAME = 'index_ci_builds_metadata_on_id_partition_id_unique' + OLD_INDEX_NAME = 'index_ci_builds_metadata_on_id_unique' + + def up + with_lock_retries(raise_on_exhaustion: true) do + execute("ALTER TABLE #{TABLE_NAME} DROP CONSTRAINT #{PRIMARY_KEY} CASCADE") + + rename_index(TABLE_NAME, NEW_INDEX_NAME, PRIMARY_KEY) + + execute("ALTER TABLE #{TABLE_NAME} ADD CONSTRAINT #{PRIMARY_KEY} " \ + "PRIMARY KEY USING INDEX #{PRIMARY_KEY}") + end + end + + # rolling back this migration is time consuming with the creation of these two indexes + def down + add_concurrent_index(TABLE_NAME, :id, unique: true, name: OLD_INDEX_NAME) + add_concurrent_index(TABLE_NAME, [:id, :partition_id], unique: true, name: NEW_INDEX_NAME) + + with_lock_retries(raise_on_exhaustion: true) do + execute("ALTER TABLE #{TABLE_NAME} DROP CONSTRAINT #{PRIMARY_KEY} CASCADE") + + rename_index(TABLE_NAME, OLD_INDEX_NAME, PRIMARY_KEY) + + execute("ALTER TABLE #{TABLE_NAME} ADD CONSTRAINT #{PRIMARY_KEY} " \ + "PRIMARY KEY USING INDEX #{PRIMARY_KEY}") + end + end +end |