将记录批量插入 Active Record 表

Question

当我一次添加大量记录时，我发现我的 Model.create! 语句需要很长时间才能运行。查看ActiveRecord-Import，但它不适用于哈希数组（这是我所拥有的，我认为这很常见）。如何提高性能？

Answer 1

感谢 Chris Heald @cheald 的 2009 article，向我展示了最好的方法是多行插入命令。

将以下代码添加到我的initializers/active_record.rb 文件中，将我的Model.create!(...) 调用更改为Model.import!(...)，然后它就消失了。几个警告：

1) 它不验证数据。
2) 它使用 SQL INSERT 命令的形式，读起来像 ...

INSERT INTO <table> (field-1, field-2, ...) 
       VALUES (value-1-1, value-1-2, ...), (value-2-1, value-2-2, ...), ...`

... 这可能不是所有数据库的正确语法，但它适用于 Postgres。为您的 SQL 版本更改适当语法的代码并不难。

在我的特定情况下，使用“model.create！”将 19K+ 记录插入我的开发机器（配备 8GB RAM、2.4GHz Intel Core i5 和 SSD 的 MacBook Pro）上的一个简单表中，从 223 秒开始使用 'model.import!' 到 7.2 秒。

class ActiveRecord::Base

  def self.import!(record_list)
    raise ArgumentError "record_list not an Array of Hashes" unless record_list.is_a?(Array) && record_list.all? {|rec| rec.is_a? Hash }
    key_list, value_list = convert_record_list(record_list)        
    sql = "INSERT INTO #{self.table_name} (#{key_list.join(", ")}) VALUES #{value_list.map {|rec| "(#{rec.join(", ")})" }.join(" ,")}"
    self.connection.insert_sql(sql)
  end

  def self.convert_record_list(record_list)
    key_list = record_list.map(&:keys).flatten.uniq.sort

    value_list = record_list.map do |rec|
      list = []
      key_list.each {|key| list <<  ActiveRecord::Base.connection.quote(rec[key]) }
      list
    end

    return [key_list, value_list]
  end
end

Answer 2

使用activerecord-import gem。假设您正在读取一个 CSV 文件并生成一个Product 目录，并且您希望以 1000 条为一组插入记录：

batch,batch_size = [], 1_000 
CSV.foreach("/data/new_products.csv", :headers => true) do |row|
  batch << Product.new(row)

  if batch.size >= batch_size
    Product.import batch
    batch = []
  end
end
Product.import batch

Answer 3

我开始遇到大量记录 (> 10000) 的问题，因此我修改了代码以一次以 1000 条记录为一组工作。这是新代码的链接：

https://gist.github.com/jackrg/76ade1724bd816292e4e

Answer 4

您也可以使用activerecord-insert_many gem。只需创建一个对象数组！

events = [{name: "Movie Night", time: "10:00"}, {name: "Tutoring", time: "7:00"}, ...]

Event.insert_many(events)

Answer 5

使用事务可以大大加快批量插入的速度！

Model.transaction do
    many.times{ Model.create! }
end

如果涉及到多个模型，为每个模型做一个Model.transaction，这会受到影响：

Model1.transaction do
    Model2.transaction do
        many.times do
            m1 = Model1.create!
            m1.add_model2
        end
    end
end

Answer 6

对于 Rails 6.x，请使用 insert_all。