Вызовите метод перед запуском другого метода в Rails - PullRequest
Новая
       24

Вызовите метод перед запуском другого метода в Rails

4 голосов
/ 20 февраля 2012

У меня есть Model, который имеет method_1 до method_10.У меня также есть ModelObserver.Я хотел бы уведомить ModelObserver перед вызовом method1 на method_9, но не method_10.

Есть ли СУХОЙ способ написать это вместо повторения notify_observers (: after_something) во всех 9 методах?

Ответы [ 2 ]

8 голосов
/ 20 февраля 2012

Добавьте файл с именем monkey_patches.rb в каталог config/initializers. 

class Object
  def self.method_hook(*args)
    options = args.extract_options!
    return unless (options[:before].present? or options[:after].present?)
    args.each do |method_name|      
      old_method = instance_method(method_name) rescue next

      define_method(method_name) do |*args|
        # invoke before callback
        if options[:before].present?
          options[:before].is_a?(Proc) ? options[:before].call(method_name, self):
            send(options[:before], method_name) 
        end

        # you can modify the code to call after callback 
        # only when the old method returns true etc..
        old_method.bind(self).call(*args)

        # invoke after callback
        if options[:after].present?
          options[:after].is_a?(Proc) ? options[:after].call(method_name, self): 
            send(options[:after], method_name) 
        end
      end
    end
  end
end

Патч позволяет добавлять обратные вызовы before и after в метод экземпляра класса. Крюк может быть:

  • Имя метода экземпляра, который принимает один параметр
  • Лямбда, принимающая два параметра

Несколько хуков могут быть зарегистрированы одним и тем же способом. Подцепляемый метод должен предшествовать зацепке.

например: 

class Model < ActiveRecord::Base

  def method1
  end

  def method2
  end

  def method3
  end

  def method4
  end

  def update_cache
  end

  # instance method name as `after` callback parameter
  method_hook :method1, :method2, :after => :update_cache

  # lambda as `before` callback parameter
  method_hook :method1, :method2, 
    :before => lambda{|name, record| p name;p record}

  # lambda as `after` callback parameter
  method_hook :method3, :method4, 
    :after => lambda{|name, record| 
       Model2.increment_counter(:post_count, record.model2_id)}

end
1 голос
/ 20 февраля 2012

Как насчет этого? 

def notify; puts "Was notified."; end
def method1; end
def method2; end
def method3; end

def original
  notify
  method1
  notify
  method2
  method3
end

def dry
  [:method1, :method2].each do |m|
    notify
    send(m)
  end
  method3
 end

original
dry
...