在 Rails 中更高效地查找或创建多个记录

Question

我有一个应用程序需要发送用户活动邀请。当用户邀请朋友（用户）参加活动时，如果尚不存在一条将用户连接到该活动的新记录，则会创建一条新记录。我的模型由用户、事件和 events_user 组成。

class Event
    def invite(user_id, *args)
        user_id.each do |u|
            e = EventsUser.find_or_create_by_event_id_and_user_id(self.id, u)
            e.save!
        end
    end
end

用法

Event.first.invite([1,2,3])

我认为上述方法不是完成我的任务的最有效方法。我设想了一种类似的方法

 Model.find_or_create_all_by_event_id_and_user_id

，但不存在。

未经验证的模型

class User 
  has_many :events_users 
  has_many :events 
end 
class EventsUser 
  belongs_to :events 
  belongs_to :users 
end 
class Event 
  has_many :events_users 
  has_many :users, :through => :events_users 
end

Answer 1

首先获取所有现有记录，然后创建所有缺失的记录可能会更快：

class Event
  def invite(user_ids, *args)
    existing_user_ids = event_users.where(user_id: user_ids).pluck(:user_id)
    (user_ids - existing_user_ids).each do |u|
      event_users.create(user_id: u)
     end
  end
end

这样，如果所有 event_users 均已存在，您只需进行 1 次查询。但是，如果不存在 event_users，则与每个 EventUser 创建所需的查询数量相比，此方法会执行额外的查询。

Answer 2

你所说的最有效率是什么意思？我假设高效意味着高性能，而不是优雅、干燥、可维护的代码等。

从数据库的角度来看，如果您想将 100 条记录插入数据库，这将转换为 100“INSERT INTO events_models VALUES (x, x)" sql 查询（如果您还有唯一性验证，则可能有 100 个“SELECT COUNT(*) ..”查询）。因此，即使您想要的方法在 AR 中实现，它仍然会在属性数组上有一个循环，并保存每个 event_id、user_id 对。

从 Ruby/Rails 的角度来看，如果你想在模型上进行验证/回调等，那么你必须在循环中一一创建一个 ActiveRecord 实例。现在，如果您想超级优化您的方法（删除 ActiveRecord 类的实例化），您可以手动编写 sql 查询（从而节省一些时间和内存）。然而，与风险相比，收益微乎其微。

顺便提一句，
e.save! 不是必需的，因为：

同样的动态取景器风格可以
用于创建对象，如果它
尚不存在。这个动态
查找器被调用
find_or_create_by_ 并将返回
对象（如果它已经存在）并且
否则创建它，然后返回它。

Answer 3

如果连接模型中没有任何验证或回调，最快的方法是原始 sql：

self.connection.execute(%Q{insert into events_users (event_id,user_id) 
                             select distinct events.id,users.id from events,users 
                             where events.id = #{self.id} 
                             and users.id in ( #{user_ids.join(",")} )
                           }
                        )

在进行上述调用之前，您可以编写一个查询来消除 user_ids 中的现有记录。我有一个关于此的上一个问题< /a>，特别是关于没有模型的连接表。

由于您确实有一个连接模型，因此您可以使用 ar-import gem：

books = []
10.times do |i| 
  books << Book.new(:name => "book #{i}")
end
Book.import books