blob: 4104493cb362b3053578ce6cb00def143c0db6da (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
# frozen_string_literal: true
class AddAsyncIndexToVsaMrs < Gitlab::Database::Migration[2.1]
include Gitlab::Database::PartitioningMigrationHelpers
TABLE_NAME = :analytics_cycle_analytics_merge_request_stage_events
COLUMN_NAMES = %I[stage_event_hash_id group_id end_event_timestamp merge_request_id]
INDEX_NAME = 'index_mr_stage_events_for_consistency_check'
disable_ddl_transaction!
def up
# The table is hash partitioned
each_partition(TABLE_NAME) do |partition, partition_index_name|
prepare_async_index(
partition.identifier,
COLUMN_NAMES,
name: partition_index_name
)
end
end
def down
each_partition(TABLE_NAME) do |partition, partition_index_name|
unprepare_async_index_by_name(partition.identifier, partition_index_name)
end
end
private
def each_partition(table_name)
partitioned_table = find_partitioned_table(table_name)
partitioned_table.postgres_partitions.order(:name).each do |partition|
partition_index_name = generated_index_name(partition.identifier, INDEX_NAME)
yield partition, partition_index_name
end
end
end
|
