在ruby中调用模型中的方法之前

时间:2015-04-22 00:16:48

标签: ruby model metaprogramming

这是我的开发方式,用于在模型中的所有方法之前运行代码

电话" before_hook:months_used"方法需要在类的底部到ExecutionHooks才能获得在模块中加载的instance_method。我想在顶部加载实例方法

class BalanceChart < BalanceFind
 include ExecutionHooks

 attr_reader :options

 def initialize(options = {})
  @options = options
  @begin_at = @options[:begin_at]
 end

 def months_used
  range.map{|date| I18n.l date, format: :month_year}.uniq!
 end

 before_hook :months_used
end

module ExecutionHooks

def self.included(base)
 base.send :extend, ClassMethods
end

module ClassMethods
  def before
   @hooks.each do |name|
    m = instance_method(name)
    define_method(name) do |*args, &block|  

      return if @begin_at.blank? ## the code you can execute before methods

      m.bind(self).(*args, &block) ## your old code in the method of the class
    end
   end
  end

  def before_hook(*method_name)
   @hooks = method_name
   before
  end

  def hooks
   @hooks ||= []
  end
 end
end

3 个答案:

答案 0 :(得分:8)

您可以使用prepend执行此操作。 prependinclude类似,因为它向类的祖先添加了一个模块,但不是在类之后添加它,而是在之前添加它。

这意味着如果一个方法同时存在于前置模块和类中,则首先调用模块实现(如果它想调用基类,它可以选择调用super。)

这允许你编写一个钩子模块,如下所示:

module Hooks
  def before(*method_names)
    to_prepend = Module.new do
      method_names.each do |name| 
        define_method(name) do |*args, &block|
          puts "before #{name}"
          super(*args,&block)
        end
      end
    end
    prepend to_prepend
  end
end


class Example
  extend Hooks
  before :foo, :bar

  def foo
    puts "in foo"
  end
  def bar
    puts "in bar"
  end
end

在实际使用中,您可能希望将该模块存储在某处,以便每次调用before都不会创建新模块,但这只是一个实现细节

答案 1 :(得分:1)

不是在调用before_hook时重新定义方法,而是可以覆盖method_added挂钩,以便在定义之后立即将前挂钩添加到方法之前。这样,before_hook调用可以(实际上,必须)放在类定义的顶部。

答案 2 :(得分:1)

@rathrio这是我使用你谈过的method_added的实现。感谢

module ExecutionHooks

def validation
  p "works1"
end

def self.included(base)
  base.send :extend, ClassMethods
end

module ClassMethods

  attr_writer :hooked

  def hooked
    @hooked ||= []
  end

  def method_added(method)
    return if @hooks.nil? 
    return unless @hooks.include?(method)
    m = self.instance_method(method)
    unless hooked.include?(method)
      hooked << method
      define_method(method) do |*args, &block|  
        validation      
        m.bind(self).(*args, &block) ## your old code in the method of the class
      end
    end
  end

  def before_hook(*method_name)
    @hooks = method_name
  end

  def hooks
    @hooks ||= []
  end
 end
end

class BalanceChart < BalanceFind
 include ExecutionHooks
 before_hook :months_data, :months_used, :debits_amount, :test

  def test
    "test"
  end
end