gitlab-org--gitlab-foss/lib/gitlab/github_import/bulk_importing.rb

# frozen_string_literal: true

module Gitlab
  module GithubImport
    module BulkImporting
      # Builds and returns an Array of objects to bulk insert into the
      # database.
      #
      # enum - An Enumerable that returns the objects to turn into database
      #        rows.
      def build_database_rows(enum)
        enum.each_with_object([]) do |(object, _), rows|
          rows << build(object) unless already_imported?(object)
        end
      end

      # Bulk inserts the given rows into the database.
      def bulk_insert(model, rows, batch_size: 100, pre_hook: nil)
        rows.each_slice(batch_size) do |slice|
          pre_hook.call(slice) if pre_hook
          Gitlab::Database.bulk_insert(model.table_name, slice)
        end
        rows
      end
    end
  end
end