重新打开由gem提供的ActiveRecord模型

Question

我正在尝试扩展Vote一个gem(https://github.com/peteonrails/vote_fu)为我的应用程序提供的ActiveRecord模型().(即,不存在vote.rb于app/models)

我的第一个方法是创建一个lib/extend_vote.rb包含代码的文件:

Vote.class_eval do
  after_create :create_activity_stream_event
  has_one :activity_stream_event

  def create_activity_stream_event
    # something..
  end
end

这在创建第一个投票时有效,但是当我尝试创建每个后续投票时,我得到了错误TypeError (can't dup NilClass).

我认为这个错误是由于Vote每次请求后自动重新加载类这一事实引起的,但是lib/extend_vote.rb当服务器启动时,代码只加载一次,这会导致has_one :activity_stream_event关联表现得很奇怪.(另外,问题消失,如果我设置config.cache_classes = true的development.rb)

为了解决这个问题,我尝试通过向to_prepare我的块添加块来对每个请求重新加载投票扩展development.rb:

config.to_prepare do
  load 'extend_vote.rb'
end

这解决了这个(can't dup NilClass)问题,但现在每当我创建一个新的投票时,create_activity_stream_event回调都会被称为额外的时间.即,第一次投票调用一次,第二次调用两次,等等.所以看起来这个to_prepare块正在积极地重新加载扩展TOO并添加重复的回调.

向此Vote模型添加方法和回调的最佳方法是什么？

Answer 1

[更新:应该是防止模块在同一类中多次包含的正确解决方案]

我相信你可以使用ActiveSupport :: Concern防止模块被多次包含,这是由几次调用回调造成的.请参阅以下示例:

module VotePatch
  extend ActiveSupport::Concern

  included do
    after_create :create_activity_stream_event
    has_one :activity_stream_event
  end

  module InstanceMethods
    def create_activity_stream_event
      #your code here
    end  
  end

end

Vote.send(:include, VotePatch)

Answer 2

我会尝试 agmcleod 在评论中建议的内容，但不要将其放入 lib 中，而是将其放入config/initializers/vote.rb中：

 class Vote
   after_create :create_activity_stream_event
   has_one :activity_stream_event

   def create_activity_stream_event
   # something..
   end
 end

当然，您可以分叉 gem，进行修改并链接到 Gemfile 中的分叉版本（这是我的偏好）。