Rails STI场景下子类B1如何配置仅关联C1子类的has_many关系

该需求完全可以实现，核心通过重写子类关联、指定关联类型约束 + 模型校验的组合方案实现，也可额外补充数据库层约束增强安全性。

1. 基础STI结构准备 #

首先确认你的STI父类已经配置好必要的字段：

• B表包含type字段（STI标识字段）、a_id外键
• C表包含type字段（STI标识字段）、b_id外键

基础父类模型代码示例：

# app/models/b.rb
class B < ApplicationRecord
  belongs_to :a
  has_many :cs
end

# app/models/c.rb
class C < ApplicationRecord
  belongs_to :b
end

# B的子类B1
class B1 < B
end

# C的子类C1
class C1 < C
end

2. 子类关联限制配置 #

方案一：重写父类关联，强制限定关联类型 #

在B1子类中重写继承自B的cs关联，自动过滤非C1类型的记录，赋值时也会自动设置C的类型为C1：

class B1 < B
  has_many :cs, -> { where(type: 'C1') }, class_name: 'C1', inverse_of: :b
end

方案二：补充模型校验兜底，避免非法赋值 #

为了防止绕过关联直接赋值非C1类型的记录，新增自定义校验规则：

class B1 < B
  has_many :cs, -> { where(type: 'C1') }, class_name: 'C1', inverse_of: :b

  validate :all_cs_are_c1

  private

  def all_cs_are_c1
    return if cs.all? { |c| c.is_a? C1 }
    errors.add(:cs, "仅允许关联C1子类的实例，不允许关联C的其他子类")
  end
end

可选方案：不覆盖父类通用关联，单独定义子类专属关联 #

如果需要保留B1访问所有C类型记录的能力，可以单独定义C1专属关联，不重写父类的cs关联：

class B1 < B
  # 专属关联仅用于操作C1类型的记录
  has_many :c1s, class_name: 'C1', foreign_key: 'b_id'
  # 可选：增加校验限制通用cs关联不能传入非C1类型
  validate :no_non_c1_in_cs
end

3. 数据库层约束增强（可选） #

如果使用支持Check约束的数据库（如PostgreSQL），可以新增数据库级约束从底层避免非法数据写入：

ALTER TABLE cs ADD CONSTRAINT check_c_type_match_b_type
CHECK (
  (SELECT type FROM bs WHERE bs.id = cs.b_id) != 'B1' OR cs.type = 'C1'
);

内容的提问来源于stack exchange，提问作者Kévin-DDR