diff options
Diffstat (limited to 'db/post_migrate/20230711093010_drop_default_partition_id_value_for_ci_tables.rb')
-rw-r--r-- | db/post_migrate/20230711093010_drop_default_partition_id_value_for_ci_tables.rb | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/db/post_migrate/20230711093010_drop_default_partition_id_value_for_ci_tables.rb b/db/post_migrate/20230711093010_drop_default_partition_id_value_for_ci_tables.rb new file mode 100644 index 00000000000..733cab057f6 --- /dev/null +++ b/db/post_migrate/20230711093010_drop_default_partition_id_value_for_ci_tables.rb @@ -0,0 +1,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 |