修改 ActiveRecord 模块以适用于任何模型

Question

因此，不久前，我创建了一个小模块来充当可投票多态关联所需的方法，虽然它最初仅用于 ActiveRecord，但我现在想将它与 mongo 一起使用，因为我使用的是 mongoid我需要在这个实例中创建的关联的所有方法都具有相同的名称，这里的所有内容都可以看一下我以前的代码：

# config/initializers/acts_as_votable.rb

module ActsAsVotable

end

module ActiveRecord
  class Base
    class << self
      cattr_accessor :votable

      def acts_as_votable
        has_many :votes, :as => :voteable
      end

      def acts_as_voter
        has_many :votes, :as => :voter
      end

      def votable?
        method_defined? :votes
      end
    end

    def votable?
      self.class.send(:method_defined?, :votes)
    end
  end
end

这是它的使用方式：

class Recipe < ActiveRecord::Base
  acts_as_votable
  # more stuff...
end

所以你会注意到这里有两个问题，首先，我'米延伸ActiveRecord::Base，我如何才能使其适用于任何模型，而不仅仅是继承自 ActiveRecord 的模型？，其次，我是否真的需要一个像这样的空模块 ActsAsVotable 之一？我在这里做错了什么？

如果我只是将所有代码放入模块 ActsAsVotable 中，我是否应该能够从我的模型中调用 includes ActsAsVotable ？

Answer 1

首先，将其放入初始值设定项或 /lib 中，但确保该路径已加载到您的应用程序中。

module ActsAsVotable

  extend ActiveSupport::Concern

  module InstanceMethods
    def votable?
      self.class.send(:method_defined?, :votes)
    end
  end

  module ClassMethods
    cattr_accessor :votable

    def acts_as_votable
      has_many :votes, :as => :voteable
    end

    def acts_as_voter
      has_many :votes, :as => :voter
    end

    def votable?
      method_defined? :votes
    end

  end
end

然后在任何模型中：

class User < ActiveRecord::Base

  include ActsAsVotable

end

最后：

User.methods.include?("acts_as_votable") # => true