diff options
author | Michael Kozono <mkozono@gmail.com> | 2018-02-08 11:10:57 -0800 |
---|---|---|
committer | Michael Kozono <mkozono@gmail.com> | 2018-02-14 13:07:04 -0800 |
commit | f44e93e2de5b2b57602aeee21a6cb837d3938905 (patch) | |
tree | 773fd645739e0c11b6e7379e89e531c7b7455264 | |
parent | 5199dcd7a15ba8483438f6784191370edcae1ebb (diff) | |
download | gitlab-ce-f44e93e2de5b2b57602aeee21a6cb837d3938905.tar.gz |
Schedule PopulateUntrackedUploads if needed
To finish migrating untracked files to uploads for installations that were affected by https://gitlab.com/gitlab-org/gitlab-ce/issues/42881
Or just to delete the temp table if it is empty and left behind.
-rw-r--r-- | db/migrate/20180208183958_schedule_populate_untracked_uploads_if_needed.rb | 47 | ||||
-rw-r--r-- | db/schema.rb | 2 |
2 files changed, 48 insertions, 1 deletions
diff --git a/db/migrate/20180208183958_schedule_populate_untracked_uploads_if_needed.rb b/db/migrate/20180208183958_schedule_populate_untracked_uploads_if_needed.rb new file mode 100644 index 00000000000..e46e793d9d2 --- /dev/null +++ b/db/migrate/20180208183958_schedule_populate_untracked_uploads_if_needed.rb @@ -0,0 +1,47 @@ +# See http://doc.gitlab.com/ce/development/migration_style_guide.html +# for more information on how to write migrations for GitLab. + +class SchedulePopulateUntrackedUploadsIfNeeded < ActiveRecord::Migration + include Gitlab::Database::MigrationHelpers + + DOWNTIME = false + + FOLLOW_UP_MIGRATION = 'PopulateUntrackedUploads'.freeze + + class UntrackedFile < ActiveRecord::Base + include EachBatch + + self.table_name = 'untracked_files_for_uploads' + end + + def up + if table_exists?(:untracked_files_for_uploads) + process_or_remove_table + end + end + + def down + # nothing + end + + private + + def process_or_remove_table + if UntrackedFile.all.empty? + drop_temp_table + else + schedule_populate_untracked_uploads_jobs + end + end + + def drop_temp_table + drop_table(:untracked_files_for_uploads, if_exists: true) + end + + def schedule_populate_untracked_uploads_jobs + say "Scheduling #{FOLLOW_UP_MIGRATION} background migration jobs since there are rows in untracked_files_for_uploads." + + bulk_queue_background_migration_jobs_by_range( + UntrackedFile, FOLLOW_UP_MIGRATION) + end +end diff --git a/db/schema.rb b/db/schema.rb index f0d94974f41..ac59992fba1 100644 --- a/db/schema.rb +++ b/db/schema.rb @@ -11,7 +11,7 @@ # # It's strongly recommended that you check this file into your version control system. -ActiveRecord::Schema.define(version: 20180202111106) do +ActiveRecord::Schema.define(version: 20180208183958) do # These are extensions that must be enabled in order to support this database enable_extension "plpgsql" |