ruby - 覆盖方法然后恢复

Question

我正在尝试找到一种方法，可以覆盖方法，执行某些操作，然后在不留下任何工件的情况下进行恢复。

我已经使用 mocha 实现了这个，但显然这不会在生产应用程序中运行。请注意，新方法有参数，而旧方法没有。

示例如下

require 'rubygems'
require 'mocha'

class Example

  def to_something
    self.stubs(:attribs => other(1))
    r = attribs_caller
    self.unstub(:attribs)
    r
  end

  def other(int)
    {"other" => int }
  end

  def attribs_caller
    attribs
  end

  def attribs
    {"this" => 1 }
  end

end

a1 = Example.new

puts a1.attribs_caller  #=> this1
puts a1.to_something    #=> other1
puts a1.attribs_caller  #=> this1

Answer 1

class String
  alias orig_reverse reverse
  def reverse(n)
    'fooled you. '*n
  end
end

puts "ab".reverse(2)
#=> fooled you fooled you

# clean up:
class String
  alias reverse orig_reverse
  remove_method(:orig_reverse)
end

puts "ab".reverse #=> ba

Answer 2

另一种无需创建额外方法的方法是：

class Foo
  def bar
    :old_method
  end
end

Foo.new.bar # => :old_method

$old_method = Foo.new.method(:bar)

class Foo
  def bar
    :new_method
  end
end

Foo.new.bar # => :new_method

class Foo
  define_method($old_method.name, &$old_method)
end

Foo.new.bar # => :old_method

我认为这比使用别名方法更好。在 Ruby 中，方法也是对象。我只是在破坏对象（方法）与类的关联之前获取对象的引用。我添加相同的方法后。如果您使用 undef 关键字从类中删除该方法，它也有效。不好的一点是你必须有一个类的对象来获取方法的引用。