如何在使用<<操作符更新has-many-through集合时实现验证

在Rails 3.2.1的遗留项目中，通过collection.platforms << new_platform的方式更新关联集合时，添加到PlatformCollection模型的自定义验证each_platform_branch_id并未触发。查文档得知，<<操作符会直接执行SQL更新，不会触发父模型的验证，事后手动调用collection.save时数据库已被修改，无法回滚。

核心需求：如何在使用<<操作符更新has-many-through关联集合时，正确执行验证逻辑，避免非法数据写入数据库？

相关模型代码 #

PlatformCollection 模型

class PlatformCollection < ActiveRecord::Base
  belongs_to :tezt
  has_many :platform_collection_platforms, :order => "platform_collection_platforms.default DESC"
  has_many :platforms, :through => :platform_collection_platforms, :order => "platform_collection_platforms.default DESC"

  def each_platform_branch_id
    platforms.each do |p|
      if p.branch_id != @branch.id
        err_msg = "Invalid Platform: #{p.to_json} for PlatformCollection: #{self.to_json}. " \
                  "Expected Branch: #{@branch.id}, Actual Branch: #{p.branch_id}"
        errors.add(:platform, err_msg)
      end
    end
  end
end

Platform 模型

class Platform < ActiveRecord::Base
  attr_accessible :name
  belongs_to :branch
  has_many :platform_collection_platforms
  has_many :platform_collections, :through => :platform_collection_platforms
end

方法1：在中间模型添加验证 #

has-many-through的关联依赖中间表platform_collection_platforms，直接在中间模型中添加验证，拦截非法关联的创建：

1. 生成中间模型（若未创建）：

rails generate model PlatformCollectionPlatform platform_collection:references platform:references

2. 编写中间模型的验证逻辑：

class PlatformCollectionPlatform < ActiveRecord::Base
  belongs_to :platform_collection
  belongs_to :platform

  validate :check_branch_matching

  private

  def check_branch_matching
    # 这里根据实际业务调整获取期望分支ID的逻辑，示例假设集合关联的tezt有branch_id
    expected_branch_id = platform_collection.tezt.branch_id
    if platform.branch_id != expected_branch_id
      errors.add(:base, "平台#{platform.name}的分支ID与集合所属分支不匹配")
    end
  end
end

使用<<添加关联时，会自动触发中间模型的验证，验证不通过则不会写入数据库。

方法2：重写PlatformCollection的<<方法 #

在PlatformCollection模型中重写platforms关联的<<方法，先执行验证再决定是否添加：

class PlatformCollection < ActiveRecord::Base
  belongs_to :tezt
  has_many :platform_collection_platforms, :order => "platform_collection_platforms.default DESC"
  has_many :platforms, :through => :platform_collection_platforms, :order => "platform_collection_platforms.default DESC"

  def platforms<<(new_platform)
    # 执行分支匹配验证
    expected_branch_id = tezt.branch_id
    if new_platform.branch_id != expected_branch_id
      err_msg = "Invalid Platform: #{new_platform.to_json} for PlatformCollection: #{self.to_json}. " \
                "Expected Branch: #{expected_branch_id}, Actual Branch: #{new_platform.branch_id}"
      errors.add(:platform, err_msg)
      raise ActiveRecord::RecordInvalid.new(self) # 抛出异常阻止关联创建
    else
      super # 验证通过，调用原生<<方法执行添加
    end
  end
end

注意：调用方需要捕获异常或处理验证失败的场景。

方法3：改用build构建关联并手动验证保存 #

放弃直接使用<<，通过中间模型构建关联，触发父模型验证后再保存：

# 业务代码中替换原<<操作
pcp = collection.platform_collection_platforms.build(platform: new_platform)
if collection.valid?
  pcp.save
else
  # 处理验证失败逻辑，比如返回错误信息
end

同时需要将原自定义方法改为标准验证方法：

class PlatformCollection < ActiveRecord::Base
  # ... 其他代码 ...
  validate :each_platform_branch_id

  def each_platform_branch_id
    platforms.each do |p|
      expected_branch_id = tezt.branch_id
      if p.branch_id != expected_branch_id
        err_msg = "Invalid Platform: #{p.to_json} for PlatformCollection: #{self.to_json}. " \
                  "Expected Branch: #{expected_branch_id}, Actual Branch: #{p.branch_id}"
        errors.add(:platform, err_msg)
      end
    end
  end
end

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