Rails、理论和生产中的 find_or_create 和race-condition

Question

你好，我有这段代码

class Place < ActiveRecord::Base
  def self.find_or_create_by_latlon(lat, lon)
    place_id = call_external_webapi
    result = Place.where(:place_id => place_id).limit(1)
    result = Place.create(:place_id => place_id, ... ) if result.empty? #!
    result
  end
end

然后我想在另一个模型或控制器中执行

p = Post.new
p.place = Place.find_or_create_by_latlon(XXXXX, YYYYY) # race-condition
p.save

但是如果执行的操作是创建，Place.find_or_create_by_latlon需要太多时间来获取数据 em> 有时在生产中 p.place 为零。

如何在执行 p.save 之前强制等待响应？ 谢谢你的建议

Answer 1

你是对的，这是一个竞争条件，它通常可以由双击表单上的提交按钮的人触发。如果遇到错误，您可能会做的就是循环返回。

result = Place.find_by_place_id(...) ||
  Place.create(...) ||
  Place.find_by_place_id(...)

有更优雅的方法可以做到这一点，但基本方法就在这里。

Answer 2

我不得不处理类似的问题。在我们的后端，如果用户不存在，则会根据令牌创建用户。创建用户记录后，将发送一个缓慢的 API 调用来更新用户信息。

def self.find_or_create_by_facebook_id(facebook_id)
  User.find_by_facebook_id(facebook_id) || User.create(facebook_id: facebook_id)
rescue ActiveRecord::RecordNotUnique => e
  User.find_by_facebook_id(facebook_id)
end

def self.find_by_token(token)
  facebook_id = get_facebook_id_from_token(token)

  user = User.find_or_create_by_facebook_id(facebook_id)

  if user.unregistered?
    user.update_profile_from_facebook
    user.mark_as_registered
    user.save
  end

  return user
end

该策略的步骤是首先从 create 方法中删除缓慢的 API 调用（在我的例子中为 update_profile_from_facebook）。由于该操作需要很长的时间，因此当您将该操作作为 create 调用的一部分时，会显着增加重复插入操作的机会。

第二步是向数据库列添加唯一约束，以确保不会创建重复项。

最后一步是创建一个函数，在重复插入操作发送到数据库的罕见情况下捕获 RecordNotUnique 异常。

这可能不是最优雅的解决方案，但它对我们有用。

Answer 3

我在一个伙伴作业中遇到了这个问题，该作业会重试并反复出现错误，并最终自行清除。我找到的最好的解释是在博客文章此处。要点是 postgres 保留一个内部存储的值，用于增加主键，但主键会以某种方式弄乱。这对我来说是正确的，因为我正在设置主键，而不仅仅是使用递增的值，所以这很可能是这样出现的。上面链接中评论中的解决方案似乎是调用 ActiveRecord::Base.connection.reset_pk_sequence!(table_name) 这为我解决了这个问题。

begin
   result = Place.where(:place_id => place_id).limit(1)
   result = Place.create(:place_id => place_id, ... ) if result.empty? #!
rescue ActiveRecord::StatementInvalid => error
   @save_retry_count =  (@save_retry_count || 1)
   ActiveRecord::Base.connection.reset_pk_sequence!(:place)
   retry if( (@save_retry_count -= 1) >= 0 )
   raise error
end