覆盖 ActiveRecord 对模型和集合的查找的最简洁方法是什么？

Question

我有重写 Ar 的 find 方法的库代码。 我还包括所有 Association 类的模块，因此 MyModel.find 和 @parent.my_models.find 都可以工作并应用正确的范围。

我的代码基于 will_paginate 的：

a = ActiveRecord::Associations
returning([ a::AssociationCollection ]) { |classes|
  # detect http://dev.rubyonrails.org/changeset/9230
  unless a::HasManyThroughAssociation.superclass == a::HasManyAssociation
    classes << a::HasManyThroughAssociation
  end
}.each do |klass|
  klass.send :include, Finder::ClassMethods
  klass.class_eval { alias_method_chain :method_missing, :paginate }
end

我的问题是，我只想覆盖某些模型的查找器。 目前我需要扩展所有模型共享的所有关联集合类。 我知道我可以通过传递模块来扩展每个模型的关联：

has_many :things, :extend => SomeCustomMethods

但我的库基本上是 ActiveRecord 插件，所以我想要一个适用于模型和范围集合而不影响应用程序中的所有模型的可插入查找器扩展的干净约定。

Answer 1

您想要重写 find_every，这是最终将使用相应查询运行 find_by_sql 的 AR 方法。 重写 find 对于自定义查找器不起作用，而且它更脆弱。

但为了与其他插件兼容，你不能只重载这个方法。 相反，为其添加别名并在执行所需操作后调用原始实现：

module MyPlugin
  def self.included(base)
    class << base
      alias_method :find_every_without_my_plugin, :find_every
      def find_every(*args)
        # do whatever you need ...
        find_every_without_my_plugin(*args)
      end
    end
  end
end

ActiveRecord::Base.send :include, MyPlugin

这将为所有类启用您的插件。 您想如何控制启用哪些模型？ 也许是一个标准的插件访问器？

class User < ActiveRecord::Base
  my_plugin
end

为了支持这一点，您需要将 class << base 到类方法（因此 base 应该是 self）。 喜欢：

module MyPlugin
  def self.included(base)
    class << base
      base.extend ClassMethods
    end
  end

  module ClassMethods
    def my_plugin
      class << self
        alias_method :find_every_without_my_plugin, :find_every
        # ...
      end
    end
  end
end

Answer 2

首先，确保你熟悉Ruby的方法调用继承结构 ，因为如果没有这个，你最终可能会在黑暗中四处乱刺。

在 ActiveRecord 类中执行此操作的最直接方法是：

def self.find(*args)
  super
end

这也适用于关联，因为它们本身使用基查找器。 现在您只需要进行定制即可。 其复杂性可能差异很大，我不知道你在做什么，所以我无法提供任何建议。

此外，动态定义它本身就是一项练习，但这应该能让您指明正确的方向。

Answer 3

佩德罗的回答是正确的，但有一个小错误。

def self.included(base)
  class << base
    base.extend ClassMethods
  end
end

应该是

def self.included(base)
  base.extend ClassMethods
end

Using class << base ... end 具有在类方法范围内对“base”调用“extend”的效果，但 ActiveRecord::Base 中没有方法“base”，因此会引发错误。 单独使用base.extend将调用ActiveRecord::Base的'extend'方法。