在Ruby中動態定義方法時，如何處理參數和塊？

Question

以下是可以動態地將日誌記錄添加到方法的類的代碼。這適用於方法不帶參數且不接受任何塊的情況。

如何重寫代碼，以便我可以添加方法的日誌記錄，即使它接受參數並接受一個塊？

class X 
    def greet 
     puts "hi" 
    end 

    def self.add_logging(method_name) 
     alias_method("original_#{method_name}".to_sym, method_name) 

     #How do i rewrie this to account for method_name's arguments and ability to accept a block? 
     define_method(method_name) do 
      puts "calling #{method_name}" 
      send "original_#{method_name}".to_sym 
      puts "done #{method_name}" 
     end 
    end 
end 

x = X.new 
x.greet 
X.add_logging(:greet) 
x.greet 

輸出

>hi 
>calling greet 
>hi 
>done greet 

Answer 1

class X 
    def self.add_logging method 
    alias_method "original_#{method}", method 
    define_method(method) do |*args, &pr| 
     puts "calling #{method}" 
     send "original_#{method}", *args, &pr 
     puts "done #{method}" 
    end 
    end 
    def greet person 
    puts "hi #{person}" 
    end 
end 

x = X.new 
x.greet("Joe") 
X.add_logging(:greet) 
x.greet("Jane") 

會給：

hi Joe 
calling greet 
hi Jane 
done greet 

Answer 2

據我知道define_method - 方法塊接受參數，它代表該方法的參數，因此我們建議如下：

define_method(method_name) do |*arguments| 
     puts "calling #{method_name}" 
     send "original_#{method_name}".to_sym, *arguments 
     puts "done #{method_name}" 
end 

*arguments在塊中的do部分中包含所有在調用方法method_name時發送的參數/參數到一個數組中，*arguments在send-method調用中再次展開它們以便在原始方法中使用。