diff options
Diffstat (limited to 'db/migrate/20201217132603_create_elastic_reindexing_subtasks.rb')
-rw-r--r-- | db/migrate/20201217132603_create_elastic_reindexing_subtasks.rb | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/db/migrate/20201217132603_create_elastic_reindexing_subtasks.rb b/db/migrate/20201217132603_create_elastic_reindexing_subtasks.rb new file mode 100644 index 00000000000..db084b885c2 --- /dev/null +++ b/db/migrate/20201217132603_create_elastic_reindexing_subtasks.rb @@ -0,0 +1,56 @@ +# frozen_string_literal: true + +class CreateElasticReindexingSubtasks < ActiveRecord::Migration[6.0] + include Gitlab::Database::MigrationHelpers + + DOWNTIME = false + + disable_ddl_transaction! + + class ReindexingTask < ActiveRecord::Base + self.table_name = 'elastic_reindexing_tasks' + end + + class ReindexingSubtask < ActiveRecord::Base + self.table_name = 'elastic_reindexing_subtasks' + end + + def up + unless table_exists?(:elastic_reindexing_subtasks) + create_table :elastic_reindexing_subtasks do |t| + t.references :elastic_reindexing_task, foreign_key: { on_delete: :cascade }, null: false + t.text :alias_name, null: false + t.text :index_name_from, null: false + t.text :index_name_to, null: false + t.text :elastic_task, null: false + t.integer :documents_count_target + t.integer :documents_count + t.timestamps_with_timezone null: false + end + end + + add_text_limit :elastic_reindexing_subtasks, :index_name_from, 255 + add_text_limit :elastic_reindexing_subtasks, :index_name_to, 255 + add_text_limit :elastic_reindexing_subtasks, :elastic_task, 255 + add_text_limit :elastic_reindexing_subtasks, :alias_name, 255 + + ReindexingTask.find_each do |task| + next if task.index_name_from.blank? || task.index_name_to.blank? || task.elastic_task.blank? + next if ReindexingSubtask.where(elastic_reindexing_task_id: task.id).exists? + + ReindexingSubtask.create( + elastic_reindexing_task_id: task.id, + documents_count_target: task.documents_count_target, + documents_count: task.documents_count, + alias_name: 'gitlab-production', + index_name_from: task.index_name_from, + index_name_to: task.index_name_to, + elastic_task: task.elastic_task + ) + end + end + + def down + drop_table :elastic_reindexing_subtasks + end +end |