summaryrefslogtreecommitdiff
path: root/lib/gitlab/background_migration/cleanup_concurrent_type_change.rb
diff options
context:
space:
mode:
authorSean McGivern <sean@mcgivern.me.uk>2018-01-04 10:14:06 +0000
committerSean McGivern <sean@mcgivern.me.uk>2018-01-04 10:14:06 +0000
commitac409fb44402622dfd6abb076f7a85df4b27d39d (patch)
treee32e266d1ab620e70486926f31bafcc4e3e7ae88 /lib/gitlab/background_migration/cleanup_concurrent_type_change.rb
parentb1e1990ee263bcae73f0e55526a55cff66103220 (diff)
parent78d22fb20db14c90861318b9f316466fbf002114 (diff)
downloadgitlab-ce-ac409fb44402622dfd6abb076f7a85df4b27d39d.tar.gz
Merge branch 'change-issues-closed-at-background-migration' into 'master'
Use a background migration for migrating issues.closed_at See merge request gitlab-org/gitlab-ce!16083
Diffstat (limited to 'lib/gitlab/background_migration/cleanup_concurrent_type_change.rb')
-rw-r--r--lib/gitlab/background_migration/cleanup_concurrent_type_change.rb54
1 files changed, 54 insertions, 0 deletions
diff --git a/lib/gitlab/background_migration/cleanup_concurrent_type_change.rb b/lib/gitlab/background_migration/cleanup_concurrent_type_change.rb
new file mode 100644
index 00000000000..de622f657b2
--- /dev/null
+++ b/lib/gitlab/background_migration/cleanup_concurrent_type_change.rb
@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+module Gitlab
+ module BackgroundMigration
+ # Background migration for cleaning up a concurrent column rename.
+ class CleanupConcurrentTypeChange
+ include Database::MigrationHelpers
+
+ RESCHEDULE_DELAY = 10.minutes
+
+ # table - The name of the table the migration is performed for.
+ # old_column - The name of the old (to drop) column.
+ # new_column - The name of the new column.
+ def perform(table, old_column, new_column)
+ return unless column_exists?(:issues, new_column)
+
+ rows_to_migrate = define_model_for(table)
+ .where(new_column => nil)
+ .where
+ .not(old_column => nil)
+
+ if rows_to_migrate.any?
+ BackgroundMigrationWorker.perform_in(
+ RESCHEDULE_DELAY,
+ 'CleanupConcurrentTypeChange',
+ [table, old_column, new_column]
+ )
+ else
+ cleanup_concurrent_column_type_change(table, old_column)
+ end
+ end
+
+ # These methods are necessary so we can re-use the migration helpers in
+ # this class.
+ def connection
+ ActiveRecord::Base.connection
+ end
+
+ def method_missing(name, *args, &block)
+ connection.__send__(name, *args, &block) # rubocop: disable GitlabSecurity/PublicSend
+ end
+
+ def respond_to_missing?(*args)
+ connection.respond_to?(*args) || super
+ end
+
+ def define_model_for(table)
+ Class.new(ActiveRecord::Base) do
+ self.table_name = table
+ end
+ end
+ end
+ end
+end