summaryrefslogtreecommitdiff
path: root/lib/gitlab/import/database_helpers.rb
blob: f8ea7a7adcda69be9eaa4fd11b24d6d72c32de49 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# frozen_string_literal: true

module Gitlab
  module Import
    module DatabaseHelpers
      # Inserts a raw row and returns the ID of the inserted row.
      #
      # attributes - The attributes/columns to set.
      # relation - An ActiveRecord::Relation to use for finding the table name
      def insert_and_return_id(attributes, relation)
        # We use bulk_insert here so we can bypass any queries executed by
        # callbacks or validation rules, as doing this wouldn't scale when
        # importing very large projects.
        result = Gitlab::Database # rubocop:disable Gitlab/BulkInsert
                 .bulk_insert(relation.table_name, [attributes], return_ids: true)

        result.first
      end
    end
  end
end