扩展ActiveRecord :: Base

Question

我正在尝试向ActiveRecord添加一些自定义方法.我想为模型的每个字段添加一个*_after和*_before范围,date以便我可以这样做:

User.created_at_after(DateTime.now - 3.days).created_at_before(DateTime.now)

我按照这里解释的解决方案Rails扩展了ActiveRecord :: Base但是当我执行rails控制台并尝试调用方法时出现undefined method错误.

这是我的代码:

# config/initializers/active_record_date_extension.rb
require "active_record_date_extension"

# lib/active_record_date_extension.rb
module ActiveRecordDateExtension
  extend ActiveSupport::Concern

  included do |base|
    base.columns_hash.each do |column_name,column|
      if ["datetime","date"].include? column.type
        base.define_method("#{column_name}_after") do |date|
          where("#{column_name} > ?", date)
        end
        base.define_method("#{column_name}_before") do |date|
          where("#{column_name} < ?", date)
        end
      end
    end
  end
end
Rails.application.eager_load!

ActiveRecord::Base.descendants.each do |model|
  model.send(:include, ActiveRecordDateExtension)
end

我究竟做错了什么？

Answer 1

感谢之前的回答，我意识到了部分问题。以下是我经过研究后得出的所有问题和解决方案：

1. column.type是一个符号，我将它与字符串进行比较。
2. base.define_method是一个私有方法
3. 我必须在 中定义方法singleton_class，而不是在base类中或class.
4. Rails.application.eager_load!即使不需要，也会导致急切加载。这不会影响功能，但首先，急切加载不应该由这个“扩展”负责，其次它依赖于 Rails，使得“扩展”仅与 Rails 兼容。

考虑到这些问题，我决定使用 method_missingruby​​ 的功能来实现它，并编写了这个 gem ( https://github.com/simon0191/date_supercharger )。这是这个问题的相关部分：

module DateSupercharger
  extend ActiveSupport::Concern

  included do
    def self.method_missing(method_sym, *arguments, &block)
      return super unless descends_from_active_record? 
      matcher = Matcher.new(self,method_sym)
      # Inside matcher
      # method_sym.to_s =~ /^(.+)_(before|after)$/

      if matcher.match?
        method_definer = MethodDefiner.new(self) # self will be klass inside Matcher
        method_definer.define(attribute: matcher.attribute, suffix: matcher.suffix)
        # Inside MethodDefiner
        # new_method = "#{attribute}_#{suffix}"
        # operators = { after: ">", before: "<" }
        # klass.singleton_class.class_eval do
        #   define_method(new_method) do |date|
        #     where("#{attribute} #{operators[suffix]} ?", date)
        #   end
        # end
        send(method_sym, *arguments)
      else
        super
      end
    end

    def self.respond_to?(method_sym, include_private = false)
      return super unless descends_from_active_record?
      if Matcher.new(self,method_sym).match?
        true
      else
        super
      end
    end
  end
end
ActiveRecord::Base.send :include, DateSupercharger