Scala LanguageFunctions

Remarks

Scala has first-class functions.

Difference between functions and methods:

A function is not a method in Scala: functions are a value, and may be assigned as such. Methods (created using def), on the other hand, must belong to a class, trait or object.

  • Functions are compiled to a class extending a trait (such as Function1) at compile-time, and are instantiated to a value at runtime. Methods, on the other hand, are members of their class, trait or object, and do not exist outside of that.
  • A method may be converted to a function, but a function cannot be converted to a method.
  • Methods can have type parameterization, whereas functions do not.
  • Methods can have parameter default values, whereas functions can not.

Anonymous Functions

Anonymous functions are functions that are defined but not assigned a name.

The following is an anonymous function that takes in two integers and returns the sum.

(x: Int, y: Int) => x + y

The resultant expression can be assigned to a val:

val sum = (x: Int, y: Int) => x + y

Anonymous functions are primarily used as arguments to other functions. For instance, the map function on a collection expects another function as its argument:

// Returns Seq("FOO", "BAR", "QUX")
Seq("Foo", "Bar", "Qux").map((x: String) => x.toUpperCase)

The types of the arguments of the anonymous function can be omitted: the types are inferred automatically:

Seq("Foo", "Bar", "Qux").map((x) => x.toUpperCase)

If there is just one argument, the parentheses around that argument can be omitted:

Seq("Foo", "Bar", "Qux").map(x => x.toUpperCase)

Underscores shorthand

There is an even shorter syntax that doesn't require names for the arguments. The above snippet can be written:

Seq("Foo", "Bar", "Qux").map(_.toUpperCase)

_ represents the anonymous function arguments positionally. With an anonymous function that has multiple parameters, each occurrence of _ will refer to a different argument. For instance, the two following expressions are equivalent:

// Returns "FooBarQux" in both cases
Seq("Foo", "Bar", "Qux").reduce((s1, s2) => s1 + s2)
Seq("Foo", "Bar", "Qux").reduce(_ + _)

When using this shorthand, any argument represented by the positional _ can only be referenced a single time and in the same order.

Anonymous Functions with No Parameters

To create a value for an anonymous function that does not take parameters, leave the parameter list blank:

val sayHello = () => println("hello")

Composition

Function composition allows for two functions to operate and be viewed as a single function. Expressed in mathematical terms, given a function f(x) and a function g(x), the function h(x) = f(g(x)).

When a function is compiled, it is compiled to a type related to Function1. Scala provides two methods in the Function1 implementation related to composition: andThen and compose. The compose method fits with the above mathematical definition like so:

val f: B => C = ...
val g: A => B = ...

val h: A => C = f compose g

The andThen (think h(x) = g(f(x))) has a more 'DSL-like' feeling:

val f: A => B = ...
val g: B => C = ...

val h: A => C = f andThen g

A new anonymous function is allocated with that is closed over f and g. This function is bound to the new function h in both cases.

def andThen(g: B => C): A => C = new (A => C){
  def apply(x: A) = g(self(x))
}

If either f or g works via a side-effect, then calling h will cause all side-effects of f and g to happen in the order. The same is true of any mutable state changes.

Relationship to PartialFunctions

trait PartialFunction[-A, +B] extends (A => B)

Every single-argument PartialFunction is also a Function1. This is counter-intuitive in a formal mathematical sense, but better fits object oriented design. For this reason Function1 does not have to provide a constant true isDefinedAt method.

To define a partial function (which is also a function), use the following syntax:

{ case i: Int => i + 1 } // or equivalently { case i: Int ⇒ i + 1 }

For further details, take a look at PartialFunctions.