Kann ich Arity für einen Block erzwingen, der an eine Methode übergeben wurde?

Question

Gibt es eine Möglichkeit, die strenge Arity-Erzwingung eines Proc instanziiert zu aktivieren, der Proc.new oder Kernel.proc instanziiert, so dass es sich wie ein Proc verhält, der mit lambda instanziiert wird?

Meine initialize Methode nimmt einen Block &action und weist es einer Instanzvariablen zu. Ich möchte action streng Arity erzwingen, also, wenn ich später Argumente darauf anwende, löst es eine ArgumentError, dass ich retten und eine sinnvollere Ausnahme auslösen kann. Grundsätzlich gilt:

class Command 
    attr_reader :name, :action 

    def initialize(name, &action) 
    @name = name 
    @action = action 
    end 

    def perform(*args) 
    begin 
     action.call(*args) 
    rescue ArgumentError 
     raise(WrongArity.new(args.size)) 
    end 
    end 
end 

class WrongArity < StandardError; end 

Leider hat action nicht arity standardmäßig erzwingen:

c = Command.new('second_argument') { |_, y| y } 
c.perform(1) # => nil 

action.to_proc nicht funktioniert, auch nicht lambda(&action).

Irgendwelche anderen Ideen? Oder bessere Ansätze für das Problem?

Danke!

Answer 1

Ihre @action wird eine Proc Instanz und Proc s haben eine arity Methode sein, so können Sie prüfen, wie viele Argumente der Block haben soll:

def perform(*args) 
    if args.size != @action.arity 
    raise WrongArity.new(args.size) 
    end 
    @action.call(*args) 
end 

, die Pflege von splatless Blöcke wie { |a| ... } nehmen und { |a,b| ... } aber die Dinge sind ein wenig komplizierter mit splats. Wenn Sie einen Block wie { |*a| ... } haben, dann wird @action.arity -1 sein und { |a,*b| ... } wird Ihnen eine arity von -2 geben. Ein Block mit arity -1 kann eine beliebige Anzahl von Argumenten annehmen (einschließlich keiner), ein Block mit arity -2 benötigt mindestens ein Argument, kann aber mehr als das nehmen, und so weiter. Eine einfache Modifikation von splatless Test sollte kümmern sich um die splatful Blöcke:

def perform(*args) 
    if @action.arity >= 0 && args.size != @action.arity 
    raise WrongArity.new(args.size) 
    elsif @action.arity < 0 && args.size < -(@action.arity + 1) 
    raise WrongArity.new(args.size) 
    end 
    @action.call(*args) 
end 

Answer 2

Ihre Lösung:

class Command 
    attr_reader :name, :action 

    def initialize(name) # The block argument is removed 
    @name = name 
    @action = lambda # We replace `action` with just `lambda` 
    end 

    def perform(*args) 
    begin 
     action.call(*args) 
    rescue ArgumentError 
     raise(WrongArity.new(args.size)) 
    end 
    end 
end 

class WrongArity < StandardError; end 

Einige Referenzen: „Wenn Proc.new aus innerhalb einer Methode ohne Argumente aufgerufen wird, es wird einen neuen Proc zurückgeben, der den Block enthält, der seiner umgebenden Methode gegeben wurde. " - http://mudge.name/2011/01/26/passing-blocks-in-ruby-without-block.html

Es stellt sich heraus, dass Lambda in der gleichen Weise funktioniert.

Answer 3

Nach this answer ist die einzige Möglichkeit, ein Proc in ein Lambda zu konvertieren, define_method und Freunde. Vom docs:

define_method definiert immer ein Verfahren ohne die Tricks [das heißt ein Lambda-Stil Proc], auch wenn ein nicht-Lambda-Proc-Objekt gegeben ist. Dies ist die einzige Ausnahme, bei der die Tricks nicht erhalten bleiben.

Spezifisch, ebenso wie tatsächlich eine Methode zu definieren, gibt define_method(:method_name, &block) ein Lambda zurück. Um dies zu verwenden, ohne unnötig viele Methoden auf einem schlechten Objekt zu definieren, könnten Sie define_singleton_method für ein temporäres Objekt verwenden.

So könnte man so etwas tun:

def initialize(name, &action) 
    @name = name 
    @action = to_lambda(&action) 
end 

def perform(*args) 
    action.call(*args) 
    # Could rescue ArgumentError and re-raise a WrongArity, but why bother? 
    # The default is "ArgumentError: wrong number of arguments (0 for 1)", 
    # doesn't that say it all? 
end 

private 

def to_lambda(&proc) 
    Object.new.define_singleton_method(:_, &proc) 
end