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

20201026182253_schedule_populate_vulnerability_feedback_pipeline_id.rb « post_migrate « db - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 2df475ab2a87ba1ab7f73c81a975531bd2963f17 (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
# frozen_string_literal: true

class SchedulePopulateVulnerabilityFeedbackPipelineId < ActiveRecord::Migration[6.0]
  include Gitlab::Database::MigrationHelpers

  DOWNTIME = false
  INTERVAL = 2.minutes.to_i
  BATCH_SIZE = 100
  MIGRATION = 'PopulateVulnerabilityFeedbackPipelineId'

  disable_ddl_transaction!

  def up
    return unless Gitlab.ee?

    vulnerability_feedback = exec_query <<~SQL
      SELECT DISTINCT "vulnerability_feedback"."project_id"
      FROM "vulnerability_feedback"
      WHERE "vulnerability_feedback"."pipeline_id" IS NULL
      ORDER BY "vulnerability_feedback"."project_id" ASC
    SQL

    return if vulnerability_feedback.rows.blank?

    vulnerability_feedback.rows.flatten.in_groups_of(BATCH_SIZE, false).each_with_index do |project_ids, index|
      migrate_in(index * INTERVAL, MIGRATION, [project_ids])
    end
  end

  def down
    # no-op
  end
end