Ruby 《Ruby 元编程》读书笔记 (十四)

lazybios · 2015年09月21日 · 最后由 lazybios 回复于 2016年06月26日 · 2735 次阅读

传神的头图↓↓↓

Ruby元编程

前面已经对元编程的理论知识走了一遍，后面的内容侧重于实际操作和源码阅读，今天的 Topic 就是由书中第六章中的例子而来。

书中以迭代的形式，引导读者一步步实现一个名为 attr_checked 的类宏，正所谓上帝造人也需要 7 天，创造和生产这件事儿，过程和迭代都是少不了的，本文完全遵照 guide 而来，美中不足的地方是没有添加，测试用例，不过每个不走都给出了一些简单的实例演示代码。

任务描述

写一个操作方法类似attr_accessor的attr_checked的类宏，该类宏用来对属性值做检验，使用方法如下：

class Person
  include CheckedAttributes

  attr_checked :age do |v|
    v >= 18
  end
end

me = Person.new
me.age = 39  #ok
me.age = 12  #抛出异常

实施计划

使用 eval 方法编写一个名为add_checked_attribute的内核方法，为指定类添加经过简单校验的属性
重构 add_checked_attribute 方法，去掉 eval 方法，改用其它手段实现
添加代码块校验功能
修改 add_checked_attribute 为要求的 attr_checked，并使其对所有类都可用
通过引入模块的方式，只对引入该功能模块的类添加 attr_checked 方法

Step 1

def add_checked_attribute(klass, attribute)
  eval "
    class #{klass}
      def #{attribute}=(value)
        raise 'Invalid attribute' unless value
        @#{attribute} = value
      end
      def #{attribute}()
        @#{attribute}
      end
    end
  "
end

add_checked_attribute(String, :my_attr)
t = "hello,kitty"

t.my_attr = 100
puts t.my_attr

t.my_attr = false
puts t.my_attr

这一步使用eval方法，用class和def关键词分别打开类，且定义了指定的属性的 get 和 set 方法，其中的 set 方法会简单的判断值是否为空 (nil 或 false)，如果是则抛出Invalid attribute异常。

Setp 2

def add_checked_attribute(klass, attribute)
  klass.class_eval do
    define_method "#{attribute}=" do |value|
      raise "Invaild attribute" unless value
      instance_variable_set("@#{attribute}", value)
    end

    define_method attribute do
      instance_variable_get "@#{attribute}"
    end

  end
end

这一步更换掉了eval方法，同时也分别用class_eval和define_method方法替换了之前的class与def关键字，实例变量的设置和获取分别改用了instance_variable_set和instance_variable_get方法，使用上与第一步没有任何区别，只是一些内部实现的差异。

Step 3

def add_checked_attribute(klass, attribute, &validation)
  klass.class_eval do
    define_method "#{attribute}=" do |value|
      raise "Invaild attribute" unless validation.call(value)
      instance_variable_set("@#{attribute}", value)
    end

    define_method attribute do
      instance_variable_get "@#{attribute}"
    end

  end
end

add_checked_attribute(String, :my_attr){|v| v >= 180 }
t = "hello,kitty"

t.my_attr = 100  #Invaild attribute (RuntimeError)
puts t.my_attr

t.my_attr = 200
puts t.my_attr  #200

没有什么奇特的，只是加了通过代码块验证，增加了校验的灵活性，不再仅仅局限于 nil 和 false 之间了。

Step 4

class Class
  def attr_checked(attribute, &validation)
      define_method "#{attribute}=" do |value|
        raise "Invaild attribute" unless validation.call(value)
        instance_variable_set("@#{attribute}", value)
      end

      define_method attribute do
        instance_variable_get "@#{attribute}"
      end
  end
end

String.add_checked(:my_attr){|v| v >= 180 }
t = "hello,kitty"

t.my_attr = 100  #Invaild attribute (RuntimeError)
puts t.my_attr

t.my_attr = 200
puts t.my_attr  #200

这里我们把之前顶级作用域中方法名放到了Class中，由于所有对象都是Class的实例，所以这里定义的实例方法，也能被 Ruby 中的其它所有类访问到，同时在 class 定义中，self 就是当前类，所以也就省去了调用类这个参数和class_eval方法，并且我们把方法的名字也改成了attr_checked。

Step 5

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

module ClassMethods
  def attr_checked(attribute, &validation)
      define_method "#{attribute}=" do |value|
        raise "Invaild attribute" unless validation.call(value)
        instance_variable_set("@#{attribute}", value)
      end

      define_method attribute do
        instance_variable_get "@#{attribute}"
      end
  end
end

class Person
  include CheckedAttributes

  attr_checked :age do |v|
    v >= 18
  end
end