Rails中如何在非多态关联下验证两个belongs_to字段二选一存在

我有一个名为Log的模型，关联了ButtonPress和LinkClick两个模型。出于分析需求，我需要验证这两个关联字段中必须存在其一，且不能使用多态关联。

我的测试代码 #

RSpec.describe RedirectLog, type: :model do
  it { is_expected.to belong_to(:button_press).optional }
  it { is_expected.to belong_to(:link_click).optional }

  it "has either a button_press or a link_click" do
    log = build(:log, button_press: nil, link_click: nil)
    expect(log).not_to be_valid
  end
end

我的模型代码 #

class Log < ApplicationRecord
  belongs_to :button_press, optional: true
  belongs_to :link_click, optional: true

  validates :button_press, presence: true, unless: :link_click
end

运行测试时的错误 #

Failure/Error: it { is_expected.to belong_to(:button_press).optional }
       Expected Log to have a belongs_to association called button_press (and for the record not to fail validation if :button_press is unset; i.e., either the association should have been defined with `optional: true`, or there should not be a presence validation on :button_press)

简言之，即使是条件验证，我也无法同时保留belongs_to关联和存在验证。请问在Rails中，不使用多态关联时，如何实现两个belongs_to字段二选一的存在验证？

问题出在你给button_press加的条件存在验证，会和belongs_to的optional: true检测冲突——Rails的关联匹配器会检查：当关联未设置时，模型不会因该关联的验证失败。而你的条件验证在link_click不存在时，会强制要求button_press存在，导致匹配器判定button_press关联并非可选。

正确的做法是用自定义验证方法替代单独的条件validates声明，这样既不会干扰关联的optional设置，又能实现二选一的逻辑：

修改后的模型代码 #

class Log < ApplicationRecord
  belongs_to :button_press, optional: true
  belongs_to :link_click, optional: true

  validate :at_least_one_association_present

  private

  def at_least_one_association_present
    if button_press.nil? && link_click.nil?
      errors.add(:base, "必须关联button_press或link_click中的一个")
    end
    # 可选：如果需要限制不能同时关联两个，取消下面注释
    # elsif button_press.present? && link_click.present?
    #   errors.add(:base, "不能同时关联button_press和link_click")
    # end
  end
end

调整后的测试代码 #

可以补充用例让测试更全面：

RSpec.describe RedirectLog, type: :model do
  it { is_expected.to belong_to(:button_press).optional }
  it { is_expected.to belong_to(:link_click).optional }

  it "无效当两个关联都为空时" do
    log = build(:log, button_press: nil, link_click: nil)
    expect(log).not_to be_valid
    expect(log.errors[:base]).to include("必须关联button_press或link_click中的一个")
  end

  it "有效当仅button_press存在时" do
    log = build(:log, button_press: create(:button_press), link_click: nil)
    expect(log).to be_valid
  end

  it "有效当仅link_click存在时" do
    log = build(:log, button_press: nil, link_click: create(:link_click))
    expect(log).to be_valid
  end

  # 可选：如果限制不能同时存在两个关联，添加此测试
  # it "无效当两个关联都存在时" do
  #   log = build(:log, button_press: create(:button_press), link_click: create(:link_click))
  #   expect(log).not_to be_valid
  #   expect(log.errors[:base]).to include("不能同时关联button_press和link_click")
  # end
end

原理说明 #

• 自定义validate方法直接操作errors[:base]，不会触发Rails关联匹配器对单个关联验证的检测，因此belongs_to ... optional: true的断言可以正常通过。
• 自定义方法可以灵活适配业务逻辑：既可以要求至少存在一个关联，也可以限制只能存在一个，按需调整即可。

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