Rails中无法捕获PG::UniqueViolation异常的原因排查

核心问题原因 #

1. 异常层级不匹配：PostgreSQL驱动抛出的PG::UniqueViolation会被ActiveRecord自动包装为ActiveRecord::RecordNotUnique，直接捕获底层驱动异常无法命中目标。
2. 代码结构缺陷：外层rescue包裹整个循环，一旦异常抛出会直接终止迭代流程；同时response_objs.shift会破坏数组的迭代索引逻辑，导致重复处理或漏处理元素。
3. 保存方法的异常行为：若使用save方法，数据库唯一约束冲突时默认仅返回false、不会抛出异常，外层rescue根本不会触发；只有save!或开启raise_on_save_failure配置时才会抛出RecordNotUnique异常。

修正后的代码 #

def create_x_from_y_query
  response_objs = @response.body["data"]["objects"]

  response_objs.each do |obj|
    begin
      # 使用save!主动抛出异常，确保异常能被捕获
      Obj.create!(k_id: obj["k_id"], etc: obj["etc"])
      # ... 其他后续业务逻辑 ...
    rescue ActiveRecord::RecordNotUnique
      logger.warn "发现重复k_id: #{obj["k_id"]}, 跳过该条目"
      # 直接跳过当前元素，继续处理下一个
      next
    end
  end
end

额外优化建议 #

• 避免修改原数组：直接迭代原数组，用next跳过异常元素，比shift更安全，不会破坏迭代顺序。
• 批量处理优化：如果API返回条目较多，可使用find_or_create_by替代异常捕获，减少数据库异常抛出的开销：

  def create_x_from_y_query
    response_objs = @response.body["data"]["objects"]
  
    response_objs.each do |obj|
      # 先查询是否存在，不存在则创建，避免触发唯一约束异常
      Obj.find_or_create_by(k_id: obj["k_id"]) do |record|
        record.etc = obj["etc"]
        # ... 其他属性赋值 ...
      end
    end
  end
  

  注：find_or_create_by在高并发场景下仍可能出现竞态条件，此时数据库唯一索引仍会触发异常，需结合rescue处理极端情况。
• Sidekiq任务幂等性：作为Sidekiq任务，需确保任务可重复执行，避免因重试导致重复创建或遗漏数据。

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