Ruby学习笔记之Ruby 模块

>> def obj.method_missing(m, *args)
>> puts "你不能在这个对象上调用 #{m}，试试别的吧。"
>> end
=> :method_missing
>> obj.blah
你不能在这个对象上调用 blah，试试别的吧。
=> nil
组合 method_missing 与 super

一般我们会拦截未知的信息，然后决定到底怎么去处理它，可以处理，也可以把它发送给原来的 method_missing 。使用 super 就很容易实现，看个例子：

class Student
  def method_missing(m, *args)
    if m.to_s.start_with?('grade_for_')
      # return the appropriate grade, based on parsing the method name
    else
      super
    end
  end
end
上面的代码，如果调用的方法是用 grade_for 开头的就会被处理，比如 grade_for_english 。如果不是，就会调用原始的 method_missing 。

再试一个复杂点的例子。比如我们要创建一个 Person 类，这个类可以这样用：

j = Person.new("John")
p = Person.new("Paul")
g = Person.new("George")
r = Person.new("Ringo")

j.has_friend(p)
j.has_friend(g)
g.has_friend(p)
r.has_hobby("rings")

Person.all_with_friends(p).each do |person|
  puts "#{person.name} is friends with #{p.name}"
end

Person.all_with_hobbies("rings").each do |person|
  puts "#{person.name} is into rings"
end
我们想要输出的东西像这样：

John is friends with Paul
George is friends with Paul
Ringo is into rings
一个人可以有朋友和爱好，Person 可以找出某个人的所有的朋友，或者拥有某个爱好的所有的人。这两个功能是用 all_with_friends 还有 all_with_hobbies 这两个方法实现的。

Person 类上的 all_with_* 方法可以使用 method_missing 改造一下，在类里定义一段代码：

class Person
  def self.method_missing(m, *args)
    # code here
  end
end
m 是方法的名字，它可以是用 all_with 开头的，也可以不是，如果是我们就去处理一下它，如果不是就交给原始的 method_missing 。再这样修改一下：

class Person
  def self.method_missing(m, *args)
    method = m.to_s
    if method.start_with?('all_with_')
      # 在这里处理请求
    else
      super
    end
  end
end
Person 对象要跟踪它所有的朋友与爱好
Person 类跟踪所有的人
每个人都有个名字
class Person
  PEOPLE = []
  attr_reader :name, :hobbies, :friends

  def initialize(name)
    @name = name
    @hobbies = []
    @friends = []
    PEOPLE << self
  end

  def has_hobby(hobby)
    @hobbies << hobby
  end

  def has_friend(friend)
    @friends << friend
  end
每次实例化一个新人都会把它放到 PEOPLE 这个数组里。还有几个读属性，name，hobbies，friends。

initialize 方法里有个 name 变量，把它放到了 @name 属性里，同时也会初始化 hobbies 和 friends ，这两个属性在 has_hobby 与 has_friend 方法里用到了。

再完成 Person.method_missing ：

def self.method_missing(m, *args)
  method = m.to_s
  if method.start_with?('all_with_')
    attr = method[9..-1]
    if self.public_method_defined?(attr)
      PEOPLE.find_all do |person|
        person.send(attr).include?(args[0])
      end
    else
      raise ArgumentError, "Can't find #{attr}"
    end
  else
    super
  end
end
全部代码如下：

class Person
  PEOPLE = []
  attr_reader :name, :hobbies, :friends

  def initialize(name)
    @name = name
    @hobbies = []
    @friends = []
    PEOPLE << self
  end

  def has_hobby(hobby)
    @hobbies << hobby
  end

  def has_friend(friend)
    @friends << friend
  end

  def self.method_missing(m, *args)
    method = m.to_s
    if method.start_with?('all_with_')
      attr = method[9..-1]
      if self.public_method_defined?(attr)
        PEOPLE.find_all do |person|
          person.send(attr).include?(args[0])
        end
      else
        raise ArgumentError, "Can't find #{attr}"
      end
    else
      super
    end
  end
end

j = Person.new("John")
p = Person.new("Paul")
g = Person.new("George")
r = Person.new("Ringo")

j.has_friend(p)
j.has_friend(g)
g.has_friend(p)
r.has_hobby("rings")

Person.all_with_friends(p).each do |person|
  puts "#{person.name} is friends with #{p.name}"
end

Person.all_with_hobbies("rings").each do |person|
  puts "#{person.name} is into rings"
end
执行的结果会是：

John is friends with Paul
George is friends with Paul
Ringo is into rings