为Ruby on Rails模型添加校验：限制任务指派人所属群组

我有一个Ruby on Rails项目，包含User、Group和Task三个模型。用户可加入不同群组，群组可拥有不同用户，通过member表实现多对多关联。

执行种子数据时：

puts "Creating Users"
admin_one = User.create!(email: "jose@test.io", password: "1234567")
admin_two = User.create!(email: "joe@test.io", password: "1234567")

puts "Creating Groups"
group_one = Group.create!(name: "Test Group One", description: "blablabla", admin: admin_one)
group_two = Group.create!(name: "Test Group Two", description: "blablabla", admin: admin_two)

puts "Creating Tasks"
task_one = Task.create!(name: "Task 1", user: user_one, group: group_one, assignee: admin_one)
task_two = Task.create!(name: "Task 2", user: admin_one, group: group_one, assignee: admin_one)
# Test if we can create tasks for users not part of the group
task_two = Task.create!(name: "Task 3", user: admin_one, group: group_one, assignee: admin_two)

我发现即使admin_one属于group_one，仍可创建指派人是admin_two（与该群组无关联）的Task。请问如何设置规则，确保任务指派人所属群组与任务所在群组一致？除了callback外，还有其他实现逻辑的方式吗？

1. 模型自定义验证（非Callback） #

在Task模型中添加自定义验证规则，直接校验指派人是否属于任务所属群组：

class Task < ApplicationRecord
  belongs_to :group
  belongs_to :assignee, class_name: 'User'

  # 自定义验证规则
  validate :assignee_must_be_group_member

  private

  def assignee_must_be_group_member
    # 检查assignee是否在当前group的用户列表中
    unless group.users.exists?(id: assignee.id)
      errors.add(:assignee, "必须是当前群组的成员")
    end
  end
end

这种方式属于Active Record的验证机制，区别于before_save这类回调，更专注于数据合法性校验，符合Rails的最佳实践。

2. 数据库级约束 #

为了彻底避免绕过模型验证的场景（比如直接执行SQL语句），可以在数据库层面添加约束，从根源上保证数据一致性。

PostgreSQL 实现 #

创建迁移文件添加CHECK约束：

class AddAssigneeGroupConstraintToTasks < ActiveRecord::Migration[7.0]
  def change
    execute <<-SQL
      ALTER TABLE tasks
      ADD CONSTRAINT assignee_belongs_to_group
      CHECK (EXISTS (
        SELECT 1 FROM members
        WHERE members.user_id = tasks.assignee_id
        AND members.group_id = tasks.group_id
      ))
    SQL
  end
end

MySQL 实现（8.0.16+ 支持CHECK约束） #

迁移文件写法类似：

class AddAssigneeGroupConstraintToTasks < ActiveRecord::Migration[7.0]
  def change
    execute <<-SQL
      ALTER TABLE tasks
      ADD CONSTRAINT assignee_belongs_to_group
      CHECK (EXISTS (
        SELECT 1 FROM members
        WHERE members.user_id = tasks.assignee_id
        AND members.group_id = tasks.group_id
      ))
    SQL
  end
end

如果是MySQL 8.0.16之前的版本，可以通过触发器实现类似效果，或者考虑调整表结构关联逻辑。

3. 服务层封装业务逻辑 #

把任务创建的逻辑封装到独立的服务类中，集中处理业务规则，让模型和控制器职责更清晰：

class TaskCreationService
  def initialize(task_params)
    @task_params = task_params
  end

  def call
    task = Task.new(@task_params)
    unless task.group.users.exists?(id: task.assignee_id)
      task.errors.add(:assignee, "必须是当前群组的成员")
      raise ActiveRecord::RecordInvalid, task
    end
    task.save!
    task
  end
end

使用时直接调用服务类：

# 正常创建任务
TaskCreationService.new(name: "Task 2", user: admin_one, group: group_one, assignee: admin_one).call

# 非法创建会抛出异常
TaskCreationService.new(name: "Task 3", user: admin_one, group: group_one, assignee: admin_two).call

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