Une méthode Ruby peut-elle produire un itérateur ou renvoyer un tableau en fonction du contexte?

Question

J'ai une méthode arbitraire dans Ruby qui donne des valeurs multiples de sorte qu'il peut être remis à un bloc:

def arbitrary 
    yield 1 
    yield 2 
    yield 3 
    yield 4 
end 

arbitrary { |x| puts x } 

J'aimerais modifier cette méthode de sorte que, s'il n'y a pas de bloc, il retourne juste la valeurs en tant que tableau. Donc, cette construction fonctionnerait également:

myarray = arbitrary 
p a -----> [1, 2, 3, 4, 5] 

Est-ce possible dans Ruby?

Answer 1

Il y a une syntaxe que:

def arbitrary(&block) 
    values = [1, 2, 3, 4] 
    if block 
    values.each do |v| 
     yield v 
    end 
    else 
    values 
    end 
end 

Note:

yield v 

peut être remplacé par:

block.call v 

Answer 2

def arbitrary 
    values = [1,2,3,4] 
    return values unless block_given? 
    values.each { |val| yield(val) } 
end 
arbitrary { |x| puts x } 
arbitrary 

Answer 3

En ruby ​​1.9+ vous pouvez utiliser Enumerator pour mettre en œuvre cela.

def arbitrary(&block) 
    Enumerator.new do |y| 
    values = [1,2,3,4] 
    values.each { |val| y.yield(val) } 
    end.each(&block) 
end 

Il a l'avantage que cela fonctionne pour les flux infinis aussi:

# block-only version 
# 
def natural_numbers 
    0.upto(1/0.0) { |x| yield x } 
end 

# returning an enumerator when no block is given 
# 
def natural_numbers(&block) 
    Enumerator.new do |y| 
    0.upto(1/0.0) { |x| y.yield(x) } 
    end.each(&block) 
end 

Mais la façon la plus idiomatiques de le faire est de garder votre méthode avec to_enum(your_method_name, your_args) comme ceci:

def arbitrary 
    return to_enum(:arbitrary) unless block_given? 

    yield 1 
    yield 2 
    yield 3 
    yield 4 
end 

Il s'agit d'un idiome que les bibliothèques de rubis utilisent elles-mêmes à plusieurs endroits.