为Ruby模块中的每个方法调用执行代码

时间:2011-04-01 12:46:24

标签: ruby methods module metaprogramming

我正在Ruby 1.9.2中编写一个定义了几个方法的模块。当调用这些方法中的任何一个时,我希望它们中的每一个首先执行某个语句。

module MyModule
  def go_forth
    a re-used statement
    # code particular to this method follows ...
  end

  def and_multiply
    a re-used statement
    # then something completely different ...
  end
end

但我想避免在每个方法中明确地放置a re-used statement代码。有办法吗?

(如果重要的话,a re-used statement将使每个方法在被调用时打印自己的名称。它将通过puts __method__的某种变体来实现。)

6 个答案:

答案 0 :(得分:64)

像这样:

module M
  def self.before(*names)
    names.each do |name|
      m = instance_method(name)
      define_method(name) do |*args, &block|  
        yield
        m.bind(self).(*args, &block)
      end
    end
  end
end

module M
  def hello
    puts "yo"
  end

  def bye
    puts "bum"
  end

  before(*instance_methods) { puts "start" }
end

class C
  include M
end

C.new.bye #=> "start" "bum"
C.new.hello #=> "start" "yo"

答案 1 :(得分:11)

这正是为 aspector 创建的。

使用aspector,您无需编写样板元编程代码。您甚至可以更进一步将公共逻辑提取到单独的方面类中并单独测试它。

require 'aspector'

module MyModule
  aspector do
    before :go_forth, :add_multiply do
      ...
    end
  end

  def go_forth
    # code particular to this method follows ...
  end

  def and_multiply
    # then something completely different ...
  end
end

答案 2 :(得分:4)

您可以通过代理模块使用method_missing实现它,如下所示:

module MyModule

  module MyRealModule
    def self.go_forth
      puts "it works!"
      # code particular to this method follows ...
    end

    def self.and_multiply
      puts "it works!"
      # then something completely different ...
    end
  end

  def self.method_missing(m, *args, &block)
    reused_statement
    if MyModule::MyRealModule.methods.include?( m.to_s )
      MyModule::MyRealModule.send(m)
    else
      super
    end
  end

  def self.reused_statement
    puts "reused statement"
  end
end

MyModule.go_forth
#=> it works!
MyModule.stop_forth
#=> NoMethodError...

答案 3 :(得分:3)

我不知道,为什么我被贬低了 - 但是正确的AOP框架比元编程hackery更好。这就是OP试图实现的目标。

http://debasishg.blogspot.com/2006/06/does-ruby-need-aop.html

另一种解决方案可能是:

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

  module ClassMethods
    def before_filter(method_name, options = {})
      aop_methods = Array(options[:only]).compact
      return if aop_methods.empty?
      aop_methods.each do |m|
        alias_method "#{m}_old", m
        class_eval <<-RUBY,__FILE__,__LINE__ + 1
          def #{m}
            #{method_name}
            #{m}_old
          end
        RUBY
      end
    end
  end
end

module Bar
  def hello
    puts "Running hello world"
  end
end

class Foo
  include Bar
  def find_hello
    puts "Running find hello"
  end
  include Aop
  before_filter :find_hello, :only => :hello
end

a = Foo.new()
a.hello()

答案 4 :(得分:2)

你可以通过元编程技术来做到这一点,这是一个例子:

module YourModule
  def included(mod)
    def mod.method_added(name)
      return if @added 
      @added = true
      original_method = "original #{name}"
      alias_method original_method, name
      define_method(name) do |*args|
        reused_statement
        result = send original_method, *args
        puts "The method #{name} called!"
        result
      end
      @added = false
    end
  end

  def reused_statement
  end
end

module MyModule
  include YourModule

  def go_forth
  end

  def and_multiply
  end
end

仅适用于ruby 1.9及更高版本

UPDATE:也不能使用block,即实例方法中没有产量

答案 5 :(得分:0)

元编程可以实现。

另一种选择是Aquarium。 Aquarium是一个为Ruby实现面向方面编程(AOP)的框架。 AOP允许您跨正常对象和方法边界实现功能。您的用例,对每种方法应用预处理,是AOP的基本任务。