这是我的开发方式,用于在模型中的所有方法之前运行代码
电话" 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
答案 0 :(得分:8)
您可以使用prepend
执行此操作。 prepend
与include
类似,因为它向类的祖先添加了一个模块,但不是在类之后添加它,而是在之前添加它。
这意味着如果一个方法同时存在于前置模块和类中,则首先调用模块实现(如果它想调用基类,它可以选择调用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