如果 after_initialize 回调返回 false，则返回 nil 对象

Question

我的 Rails 应用程序有许多形成层次结构的模型。例如：零售商 > 部门 > 产品类别 > 产品 > 评论。

业务需求是高权限用户可以与新的或现有的“普通”用户“共享”层次结构中的任何单个元素。在没有与他们共享对象的情况下，普通用户无权查看（或执行其他任何操作）层次结构中任何级别的任何对象。

共享过程包括选择共享是否授予对目标对象的只读、读取更新或完整 CRUD 权限。

共享任何对象会授予该对象和层次结构中所有较低级别对象的 R/O、R/W 或 CRUD 权限，以及授予对象的所有直接祖先的 R/O 权限。对象集合有机地增长，因此权限系统仅通过记录 user_id、共享的 object_id 和共享的性质（R/O、CRUD 等）来工作。由于此层次结构中的对象数量一直在增长，因此在数据库中为每个用户/对象组合创建明确的权限记录是不切实际的。

相反，在用户请求周期开始时，ApplicationController 收集所有权限记录（用户 X 对部门 #5 具有 CRUD 权限）并将它们保存在内存中的哈希中。 Permissions 模型知道在将任何对象传递给它时如何评估散列 - Permission.allow?(:show, Department#5) 将根据用户权限散列的内容返回 true 或 false。

我们以部门模型为例：

# app/models/department.rb
class Department < ActiveRecord::Base

  after_initialize :check_permission

  private

  def check_permission
    # some code that returns true or false 
  end

end

当check_permission方法返回true时，我希望Department.first正常带回数据库中的第一条记录，但是，如果check_permission返回false，我想返回nil。

现在，我有一个解决方案，默认范围触发权限检查，但这会导致查询数量增加 2 倍，并且对于具有大量对象的类，内存问题和时间/性能问题肯定会出现地平线。

我的目标是使用after_initialize 回调来预授权对象。

然而，after_initialize 似乎无法阻止返回原始对象。它确实允许我重置对象属性的值，但不能放弃它。

有人知道如何实现吗？

编辑：

非常感谢到目前为止提供的所有答案和 cmets；希望这个问题的扩展版本可以澄清问题。

Answer 1

如果检查权限为假，您可以使用 before_create 回调来停止创建记录。只需在 check_permission 过滤器中返回 false 并不会创建记录。

 class Department < ActiveRecord::Base

  before_create :check_permission

  private

  def check_permission
    # return false if permission is not allowed 
  end

end

Answer 2

基本上，您需要在返回数据库查询结果之前检查访问权限（或权限）。而您正在尝试将此逻辑集成到您的模型中。

这是可能的，但不是您在问题中描述的设计。在 ActiveRecord 适配器方法中直接实现这一点并不干净（例如first、all、last 等...）。您需要重新考虑您的设计。

（如果阅读过多，请跳至点“D”）

您有多种选择，它们都取决于您的权限定义方式。我们来看几个案例：

A.一个用户有一个他拥有部门的列表，只有他可以访问它们

您可以简单地将其实现为 has_many/belongs_to 与 Active Record Associations 的关联

B. 用户和部门是独立的（换句话说：没有前面案例中描述的所有权）并且可以为每个用户和每个用户单独设置权限部门。

同样，您可以实现has_and_belongs_to_many 与Active Record Associations 的关联。您将需要创建 Web 逻辑，以便您的应用程序的管理员可以添加/编辑/删除访问权限。

C.更复杂的案例：现有的授权库

大多数人会转向授权解决方案，例如cancan、pundit 或other

D.当这些授权库对于您的需求来说过大时（实际上，我的大多数项目都是这样），我发现通过 rails scoping 实现授权可以满足我的所有需求。

让我们通过一个简单的例子来看看。我希望管理员能够访问整个数据库记录；普通用户只能访问 status = open 的部门，并且只能在营业时间（比如上午 8 点到下午 6 点）访问。我编写了一个实现我的权限逻辑的范围

# Class definition
class Department
  scope :accessible_by -> (user) do
    # admin user have all access, always
    if user.is_admin?
      all
    # Regular user can access only 'open' departments, and only
    # if their request is done between 8am and 6pm
    elsif Time.now.hour >= 8 and Time.now.hour <= 18
      where status: 'open'
    # Fallback to return ActiveRecord empty result set
    else
      none
    end
  end
end

# Fetching without association
Department.accessible_by(current_user)

# Fetching through association
Building.find(5).departments.accessible_by(current_user)

定义范围迫使我们在代码中的任何地方都使用它。您可以考虑“忘记”通过范围并直接访问模型的风险（即编写 Department.all 而不是 Department.accessible_by(current_user)）。因此，您必须在规范中（在控制器或功能级别）严格测试您的权限。

注意在此示例中，当权限失败时（如您在问题中提到的），我们不返回 nil，而是返回一个空的结果集。它通常会更好，因此您可以保留 ActiveRecord 方法链接功能。但是您也可以引发异常并将其从控制器中拯救出来，然后重定向到“未授权”页面。

Answer 3

这不是 after_initialize 回调的用途。相反，您可以只定义一个做同样事情的方法。例如，把它放在你的 Department 模型中，它应该会达到你想要的结果：

def self.get_first
  check_permission ? first : nil
end

更新

我不确定这样的事情有多安全，但您可以重写 all 方法，因为其他查询方法都基于它。

class Department < ActiveRecord::Base
  def self.all
    check_permission ? super : super.none
  end

  private

  def self.check_permission
    # some code that returns true or false 
  end
end

不过，你最好还是使用一些授权框架。

更新 2

再考虑一下，我强烈建议使用不同的方法。你真的不应该覆盖像all 这样的方法，因为肯定会有意想不到的副作用。

一种实用的替代方法是在Department 和User 之间创建has_and_belongs_to_many 关系。以下是您的设置方式：

user.rb

class User < ActiveRecord::Base
  has_and_belongs_to_many :departments
  ...
end

department.rb

class Department < ActiveRecord::Base
  has_and_belongs_to_many :users
  ...
end

然后在终端中运行这些命令：

rails g migration CreateJoinTableDepartmentsUsers departments users
rake db:migrate

现在您可以使用@department.users << @user 将用户添加到部门，或使用@user.departments << @department 将部门添加到用户。这应该可以实现您正在寻找的功能。

@user.departments 将只返回该用户的部门，@user.departments.first 将返回该用户的第一个部门或nil，如果它没有任何部门，@user.departments.find(1) 将只返回相应的部门，如果它属于给用户，否则抛出异常。