使用附加功能包装 ActiveRecord 方法

Question

我想增强 Rails 中的 ActiveRecord setter 以确保仅保存有效值。需要此功能的地方之一就是电话号码。用户可以以各种格式输入电话号码，例如，

(123) 456-7890
+1 123-456-7890

但我只想存储数字并在进入数据库时​​丢弃其余部分。我现在使用的方法是使用 alias_method 覆盖 setter 方法。另外，我试图将其放入一个模块中，以便任何包含电话号码的模型类都可以包含该模块，并定义应清理的字段。我希望使用的接口类型是，

# Person has a "phone" attribute to store phone numbers
class Person < ActiveRecord::Base
  # first include this module
  include PhoneSanitizer

  # then call the class method and tell it which 
  # fields need need to be sanitized
  sanitize_phone_field :phone
end

我在模型类中所做的唯一一件事就是包含 PhoneSanitizer 模块（它添加了一个类方法 - sanitize_phone_field在 Person 类中）。该方法现在负责重写 setter phone= 方法。这是我还没有开始工作的部分。

module PhoneSanitizer

  module ClassMethods
    # wrap each of the passed-in fields with setters that
    # clean up the phone number value of non-digits.
    def sanitize_phone(*fields)
      fields.each do |field|
        new_method = "original_#{field}=".to_sym
        original_method = "#{field}=".to_sym
        alias_method new_method, original_method
        define_method(original_method) do |value|
          self.send(new_method, phone_to_number(value))
        end
      end
    end
  end

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

  def phone_to_number(number)
    number.gsub(/[^\d]/, '')
  end

end

当调用 sanitize_phone 时，它会抛出一个错误，指出 :phone= 没有为 Person 类定义，这是有道理的。我该如何为 Person 实例的方法添加别名呢？

Answer 1

我认为你的错误不是未定义的方法alias_method，它是不同的东西，你误解了它（？）

真正的问题是ActiveRecord上的getter和setter方法是动态。 getter 和 setter 方法（即 phone 和 phone=）在从数据库加载实际 AR 对象之前不会创建。此时，AR 会枚举数据库字段并创建相应的字段方法。

在源中定义类时，这些字段方法不可用，因此您不能 alias_method 不存在的方法。然而，你可以做这样的事情（未经测试）：

module PhoneSanitizer
  module ClassMethods
    def sanitize_phone(*fields)
      fields.each do |field|
        original_method = "#{field}=".to_sym
        define_method(original_method) do |value|
          self.write_attribute(field, phone_to_number(value))
        end
      end
    end
  end

  ...
end

这应该完成与你最初想要的几乎相同的事情：
http://apidock.com/rails/ActiveRecord/AttributeMethods/Write/write_attribute