Welcome to mirror list, hosted at ThFree Co, Russian Federation.

20230711093010_drop_default_partition_id_value_for_ci_tables.rb « post_migrate « db - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 733cab057f6d1fe4e4b0d4c325dc1408c6831298 (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 DropDefaultPartitionIdValueForCiTables < Gitlab::Database::Migration[2.1]
  disable_ddl_transaction!

  TABLES = {
    ci_build_needs: [:partition_id],
    ci_build_pending_states: [:partition_id],
    ci_build_report_results: [:partition_id],
    ci_build_trace_chunks: [:partition_id],
    ci_builds_runner_session: [:partition_id],
    ci_job_variables: [:partition_id],
    ci_pending_builds: [:partition_id],
    ci_pipelines: [:partition_id],
    ci_running_builds: [:partition_id],
    ci_sources_pipelines: [:partition_id, :source_partition_id],
    ci_unit_test_failures: [:partition_id]
  }

  def up
    TABLES.each do |table_name, columns|
      with_lock_retries do
        columns.each do |column_name| # rubocop:disable Migration/WithLockRetriesDisallowedMethod
          change_column_default(table_name, column_name, from: 100, to: nil)
        end
      end
    end
  end

  def down
    TABLES.each do |table_name, columns|
      with_lock_retries do
        columns.each do |column_name| # rubocop:disable Migration/WithLockRetriesDisallowedMethod
          change_column_default(table_name, column_name, from: nil, to: 100)
        end
      end
    end
  end
end