[ACCEPTED]-Can a Ruby method yield as an iterator or return an array depending on context?-return-value

Accepted answer

Score: 19

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

Score: 14

There is a syntax for that:

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

Note:

yield v

Can be replaced 1 with:

block.call v

Score: 14

In ruby 1.9+ you can use Enumerator to implement that.

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

It 5 has the advantage that it works for infinite 4 streams too:

# 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

But the most idiomatic way to 3 do it is to guard your method with to_enum(your_method_name, your_args) like 2 so:

def arbitrary
  return to_enum(:arbitrary) unless block_given?

  yield 1
  yield 2
  yield 3
  yield 4
end

This is an idiom that ruby core libraries 1 themselves use in multiple places.

Source: stackoverflow.com

When does a 'void' method affect the parameter, and when does it affect the original object?

Return value of initialize method

VBA - Returning array from Property Get

How to drop the end of an array in Ruby

Get return code from std::thread?

gometalinter / errcheck returns a warning on deferring a func which returns a variable

Scope and return values in C++

How to elegantly ignore some return values of a MATLAB function

Is it a good practice to write methods that return void?

PL/SQL How to call a function without getting returned object

Cookie Warning

We use cookies to improve the performance of the site. By staying on our site, you agree to the terms of use of cookies.