Question

是否有可能使这项工作成为可能，而不必在课程结束时包含该模块并将其包含在顶部？

module VerboseJob
  def self.included(job_class)
    class << job_class
      alias_method :original_perform, :perform
      def perform(*args)
        JobLogger.verbose { original_perform(*args) }
      end
    end
  end
end

class HelloJob
  include VerboseJob

  def self.perform(arg1, arg2)
    puts "Job invoked with #{arg1} and #{arg2}"
  end
end

我想要发生的是，HelloJob.perform实际调用VerboseJob.perform（然后调用块内的原始方法）。因为此处的模块包含在类的顶部，所以这不起作用，因为尚未定义perform。将include移动到最后确实有效，但是有一种方式更宽容吗？我喜欢将所有包含的模块放在我的类定义的顶部。

我正在寻找一些在Module或Class完全加载时调用的方法，而不是运行时解释的方法。

Answer 1

通过推迟包装器方法的定义直到定义了原始方法，我想出了一个相当迂回/ hackish的做法：

module A
  def self.included(base)
    base.class_eval do
      def self.singleton_method_added(name)
        @@ran||=false
        if name==:perform && !@@ran
          @@ran=true
          class<<self
            alias_method :original_perform, :perform
            def perform(*args)
              puts "Hello"
              original_perform(*args)
            end
          end
        end
      end
    end
  end
end

class B
  include A

  def self.perform
    puts "Foobar"
  end
end

B.perform

修改

d11wtq将此简化为更清洁：

module VerboseJob module ClassMethods def wrap_perform! class << self def perform_with_verbose(*args) JobLogger.verbose { perform_without_verbose(*args) } end alias_method_chain :perform, :verbose \ unless instance_method(:perform) == instance_method(:perform_with_verbose) end end def singleton_method_added(name) wrap_perform! if name == :perform end end def self.included(job_class) job_class.extend ClassMethods job_class.wrap_perform! if job_class.respond_to?(:perform) end end

Answer 2

假设您希望在perform运行之前定义所有类，则可能需要使用Kernel#at_exit：

将块转换为Proc对象（因此在该点绑定它调用）并在程序退出时将其注册执行。如果注册多个处理程序，它们以相反的顺序执行注册。

   def do_at_exit(str1)
     at_exit { print str1 }
   end
   at_exit { puts "cruel world" }
   do_at_exit("goodbye ")
   exit

产生
再见残酷的世界

您可能还想了解单元测试框架（如Test :: Unit或MiniTest）如何处理延迟任务的运行。

模块包装一个类方法？

2 个答案: