如何检索创建的 ID 列表以在 Active Record 中批量插入？

Question

我有三个模型：

class Coupon < ActiveRecord::Base
  belongs_to :event
  has_many :coupon_events, :dependent => :destroy
  has_many :events, :through => :coupon_events
end 

class Event < ActiveRecord::Base
  belongs_to :event
  has_many :coupon_events, :dependent => :destroy
  has_many :coupons, :through => :coupon_events
end

class CouponEvent < ActiveRecord::Base
  belongs_to :coupon
  belongs_to :event
end

我通读了一个 CSV 文件以创建优惠券和优惠券_事件。这是非常低效的，因为一次创建一个记录并导致多个查询，每个查询都包括两个插入语句。

我想使用这样的单个插入查询：

coupon_string = " ('abc','AAA'), ('123','BBB')"
Coupon.connection.insert("INSERT INTO coupons (code, name) VALUES"+coupon_string)

然后我需要为 CouponEvent 模型创建第二个插入查询，但我需要返回的 coupon_id 的列表。是否有在插入时检索 ID 的内置方法？

Answer 1

目前，最好的（但不是理想的）解决方案是使用"activerecord-import" 批量导入。不幸的是，该 gem 不会返回插入的 id，因此您必须转身查询才能获取 id。也就是说，您将批量插入事件模型，查询数据库以将它们全部恢复到内存中。现在您有了事件 ID，因此您可以创建优惠券并批量插入它们。为 CouponEvents 重复冲洗泡沫。

与每个 Event、Coupon 和 CouponEvent 一次往返相比 - 对于一个包含数千行的文件可能有数千次往返 - 每个模型只执行 2 次往返 - 一次用于插入事件，一次用于取回事件使用 ids、同上 Coupon 和 CouponEvent - 总共 6 次往返。

Answer 2

实际上我不确定这个 colud 是否有效（如果它创建一个插入查询），但您可以尝试使用带有参数数组的 #create 方法：

new_coupons = Coupon.create([
  { :code => "abc", :name => "AAA" },
  { :code => "123", :name => "BBB" }
])

CouponEvent.create([
  { :enevt_id => ..., coupon_id: ...},
  ...
])

要为 CouponEvent 创建参数列表，您需要将返回的 new_coupons 集合映射到 id，并根据优惠券代码/名称添加 event_id（取决于它存储在 CVS 文件中的方式）。

更新：

我自己检查了，如果第一个解决方案不起作用（我的代码中没有简单的模型没有唯一性约束，所以我没有检查），并且你使用 PostgreSQL，你总是可以做类似的事情这个：

res = Coupon.connection.execute(<<-EOSQL)
  INSERT INTO coupons (code, name)
  VALUES #{values}
  RETURNING id, code
EOSQL

您需要最后一个“Returning”子句，因此您可以获取插入的 id 以及插入行的代码。您需要映射结果集：

res.map {|row|
  { :coupon_id => row["id"],
    :event_id => events.find { |e| e.coupon_code == row["code"] }
  }
}

SQL中没有标准的方法来返回插入行的列，“RETURNING”子句只在PostgreSQL中有效，所以如果你使用不同的数据库，你需要检查文档或一一插入行。

您也不能使用connection.insert，因为在 ActiveRecord 中它只返回一个插入行的 id，而不是所有行。

Answer 3

这样做的方法是插入具有唯一import_id 值的记录。步骤是：

1. 将import_id 列添加到表中。可能是 INT 或 VARCHAR，具体取决于您生成随机 ID 的方式。

2. 在第一个INSERT之前，生成一个随机ID。

3. 执行第一个多值INSERT，对每一行使用相同的import_id。

4. SELECT id FROM first_table WHERE import_id=<the random import ID>

5. 使用返回的 ID 生成第二个多值 INSERT。

Answer 4

connection.insert 只返回一个 id，公用表表达式适用于我

    insert_sql = <<-SQL
      WITH inserted_ids AS (
        INSERT INTO clients (email, name) VALUES #{array.join(', ')}
        RETURNING id
      )
      SELECT * FROM inserted_ids
    SQL
    result = ActiveRecord::Base.connection.execute(insert_sql)

Answer 5

如果您使用 mysql 并且没有在另一个脚本/进程中插入更多行，则可以使用 last_insert_id() 获取插入的第一行的 id

    first_id = ActiveRecord::Base.connection.execute("select last_insert_id()").first[0]

然后依次生成其他记录的id。

即

    data = %w(one two three)
    to_insert = "('" + data.join("'), ('") + "')"
    Model.connection.insert("INSERT INTO models(name) VALUES #{to_insert}")
    first_id = ActiveRecord::Base.connection.execute("select last_insert_id()").first[0].to_i
    hash = {}
    data.each_with_index {|d, i| hash[first_id + i] = d}