为HABTM关联添加唯一性约束时遇异常延迟抛出问题求助

一、先确认异常后的关联状态 #

捕获ActiveRecord::RecordNotUnique后，先明确关联是否真的处于脏状态，在rescue块里加几行调试代码：

begin
  user.tokens << token
rescue ActiveRecord::RecordNotUnique
  puts "tokens关联是否已加载：#{user.tokens.loaded?}"
  puts "关联是否脏：#{user.tokens.changed?}"
end

通过输出能直观看到关联的加载状态和脏状态，确认问题根源。

二、重置关联脏状态的实用方法 #

1. 直接重置关联缓存 #

这是最轻便的方式，在rescue块里调用user.tokens.reset，它会清空关联的内存缓存，下次访问时重新从数据库拉取数据，直接重置脏状态：

begin
  user.tokens << token
rescue ActiveRecord::RecordNotUnique
  user.tokens.reset
end

之后再调用user.update_counters这类操作就不会再触发之前的异常了。

2. 重新加载整个用户实例 #

如果关联重置不管用，就直接刷新整个user对象：

begin
  user.tokens << token
rescue ActiveRecord::RecordNotUnique
  user.reload
end

不过这个方法会多一次数据库查询，适合关联状态比较复杂的场景。

三、从根源避免脏状态：提前检查关联 #

与其事后处理异常，不如在添加前先判断是否已经存在关联，从源头避免脏状态产生：

unless user.tokens.exists?(token.id)
  user.tokens << token
end

这里推荐用exists?而不是include?——include?会把所有关联的token加载到内存，数据量大的时候性能拉胯，exists?直接走数据库查询，高效得多。

四、封装安全添加方法（进阶） #

可以在User模型里写一个封装好的方法，把检查和异常处理都包进去，省得每次重复写：

class User < ApplicationRecord
  has_and_belongs_to_many :tokens

  def add_token_safely(token)
    return if tokens.exists?(token.id)
    tokens << token
  rescue ActiveRecord::RecordNotUnique
    tokens.reset
  end
end

之后直接调用user.add_token_safely(token)就行，省心又统一。

内容的提问来源于stack exchange，提问作者hbhanoo