How does the compose (*) function work in Ruby (from The Ruby Programming Language)?

Question

Excerpt The Ruby Programming Language:

module Functional
  def compose(f)
    if self.respond_to?(:arity) && self.arity == 1
      lambda {|*args| self[f[*args]] }
    else
      lambda {|*args| self[*f[*args]] }
    end
  end
  alias * compose
end

class Proc; include Functional; end
class Method; include Functional; end

f = lambda {|x| x * 2 }
g = lambda {|x, y| x * y}
(f*g)[2, 3] # => 12

What is the difference between f and *f in the if/else clause?

Answer 1

The * either collects all the items into an array, or explodes an array into individual elements--depending on the context.

If args = [1, 2, 3], then:

• f[args] is equivalent to f[ [1, 2, 3] ] #There is one argument: an array.
• f[*args] is equivalent to f[1, 2, 3] #There are three arguments.

If f[*args] returns [4, 5, 6], then:

• self[f[*args]] is equivalent to self[ [4, 5, 6] ] #self is called with 1 arg.
• self[*f[*args]] is equivalent to self[4, 5, 6] #self is called with 3 args.

An example of * being used to collect items into an Array is:

• lambda {|*args| ....}

You can call that function with any number of arguments, and all the arguments will be collected into an array and assigned to the parameter variable args.