summaryrefslogtreecommitdiff
path: root/lib/gitlab/background_migration/populate_merge_request_assignees_table.rb
blob: a4c6540c61badc64f9394da8182bbfdeb2285bf9 (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
34
35
36
# frozen_string_literal: true

module Gitlab
  module BackgroundMigration
    # This background migration creates records on merge_request_assignees according
    # to the given merge request IDs range. A _single_ INSERT is issued for the given range.
    # This is required for supporting multiple assignees on merge requests.
    class PopulateMergeRequestAssigneesTable
      def perform(from_id, to_id)
        select_sql =
          MergeRequest
            .where(merge_request_assignees_not_exists_clause)
            .where(id: from_id..to_id)
            .where('assignee_id IS NOT NULL')
            .select(:id, :assignee_id)
            .to_sql

        execute("INSERT INTO merge_request_assignees (merge_request_id, user_id) #{select_sql}")
      end

      private

      def merge_request_assignees_not_exists_clause
        <<~SQL
            NOT EXISTS (SELECT 1 FROM merge_request_assignees
                        WHERE merge_request_assignees.merge_request_id = merge_requests.id)
        SQL
      end

      def execute(sql)
        @connection ||= ActiveRecord::Base.connection
        @connection.execute(sql)
      end
    end
  end
end