Contents

# frozen_string_literal: true

module ActiveRecordDataLoader
  class BulkInsertStrategy
    def initialize(data_generator)
      @data_generator = data_generator
    end

    def load_batch(row_numbers, connection)
      connection.insert(<<~SQL)
        INSERT INTO #{quoted_table_name(connection)} (#{column_list(connection)})
        VALUES #{values(row_numbers, connection)}
      SQL
    end

    def table_name
      data_generator.table
    end

    def name
      "BULK INSERT"
    end

    private

    attr_reader :data_generator

    def quoted_table_name(connection)
      @quoted_table_name ||= connection.quote_table_name(data_generator.table)
    end

    def column_list(connection)
      @column_list ||= data_generator
                       .column_list
                       .map { |c| connection.quote_column_name(c) }
                       .join(",")
    end

    def values(row_numbers, connection)
      row_numbers
        .map { |i| "(#{row_values(i, connection)})" }
        .join(",")
    end

    def row_values(row_number, connection)
      data_generator
        .generate_row(row_number)
        .map { |v| connection.quote(v) }
        .join(",")
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
active_record_data_loader-1.1.0	lib/active_record_data_loader/bulk_insert_strategy.rb
active_record_data_loader-1.0.2	lib/active_record_data_loader/bulk_insert_strategy.rb
active_record_data_loader-1.0.1	lib/active_record_data_loader/bulk_insert_strategy.rb
active_record_data_loader-1.0.0	lib/active_record_data_loader/bulk_insert_strategy.rb
active_record_data_loader-0.1.2	lib/active_record_data_loader/bulk_insert_strategy.rb