Иногда можно засунуть Руби в чашку для эспрессо. Посмотрим как.
Вот модуль FunNotFun ...
module FunNotFun
def fun
@method_type = 'fun'
end
def notfun
@method_type = 'not fun'
end
def method_added(id)
return unless @method_type
return if @bypass_method_added_hook
orig_method = instance_method(id)
@bypass_method_added_hook = true
method_type = @method_type
define_method(id) do |*args|
orig_method.bind(self).call(*args).tap do
puts "That was #{method_type}"
end
end
@bypass_method_added_hook = false
end
end
... что вы можете использовать для расширения класса ...
class Thing
extend FunNotFun
fun
def f1
puts "hey"
end
notfun
def f2
puts "hey"
end
end
... с таким результатом:
Thing.new.f1
# => hey
# => That was fun
Thing.new.f2
# => hey
# => That was not fun
Но см. Ниже строки для лучшего способа.
Аннотации (см. Ответ normalocity) представляют меньшую проблему и, будучи распространенной идиомой Ruby, будут легче сообщать о намерениях вашего кода. Вот как это сделать с аннотациями:
module FunNotFun
def fun(method_id)
wrap_method(method_id, "fun")
end
def notfun(method_id)
wrap_method(method_id, "not fun")
end
def wrap_method(method_id, type_of_method)
orig_method = instance_method(method_id)
define_method(method_id) do |*args|
orig_method.bind(self).call(*args).tap do
puts "That was #{type_of_method}"
end
end
end
end
При использовании аннотация появляется после определения метода, а не до:
class Thing
extend FunNotFun
def f1
puts "hey"
end
fun :f1
def f2
puts "hey"
end
notfun :f2
end
Результат тот же:
Thing.new.f1
# => hey
# => That was fun
Thing.new.f2
# => hey
# => That was not fun