Return in Scala

by

It’s not as simple as just omitting the return keyword. In Scala, if there is no return then the last expression is taken to be the return value. So, if the last expression is what you want to return, then you can omit the return keyword. But if what you want to return is not the last expression, then Scala will not know that you wanted to return it.

An example:

def f() = {
  if (something)
    "A"
  else
    "B"
}

Here the last expression of the function f is an if/else expression that evaluates to a String. Since there is no explicit return marked, Scala will infer that you wanted to return the result of this if/else expression: a String.

Now, if we add something after the if/else expression:

def f() = {
  if (something)
    "A"
  else
    "B"

  if (somethingElse)
    1
  else
    2
}

Now the last expression is an if/else expression that evaluates to an Int. So the return type of f will be Int. If we really wanted it to return the String, then we’re in trouble because Scala has no idea that that’s what we intended. Thus, we have to fix it by either storing the String to a variable and returning it after the second if/else expression, or by changing the order so that the String part happens last.

Finally, we can avoid the return keyword even with a nested if-else expression like yours:

def f() = {
  if(somethingFirst) {
    if (something)      // Last expression of `if` returns a String
     "A"
    else
     "B"
  }
  else {
    if (somethingElse)
      1
    else
      2

    "C"                // Last expression of `else` returns a String
  }

}

More Related Contents:

  1. error in if return in Scala
  2. Scala return statements in anonymous functions
  3. Combining two lists into one in Scala
  4. Why is Scala’s immutable Set not covariant in its type?
  5. Spark Scala: How to convert Dataframe[vector] to DataFrame[f1:Double, …, fn: Double)]
  6. How to override apply in a case class companion
  7. Getting the desugared part of a Scala for/comprehension expression?
  8. Scala how can I count the number of occurrences in a list
  9. Scala: How to define “generic” function parameters?
  10. Spark losing println() on stdout
  11. Spark DataFrame: does groupBy after orderBy maintain that order?
  12. Cartesian product of two lists
  13. Spark UDAF with ArrayType as bufferSchema performance issues
  14. I want to get the type of a variable at runtime
  15. Two ways of currying in Scala; what’s the use-case for each?
  16. When to use the equals sign in a Scala method declaration?
  17. How to get ID of a map task in Spark?
  18. “new” keyword in Scala
  19. Spark: produce RDD[(X, X)] of all possible combinations from RDD[X]
  20. Scala underscore minimal function
  21. Debugging Scala code with simple-build-tool (sbt) and IntelliJ
  22. Scala case class inheritance
  23. Use functional combinators on Scala Tuples?
  24. scala tuple unpacking
  25. Using varargs from Scala
  26. Can’t prove that singleton types are singleton types while generating type class instance
  27. Clarification needed about futures and promises in Scala
  28. Creating a method definition tree from a method symbol and a body
  29. How to sort an RDD in Scala Spark?
  30. Insert if not exists in Slick 3.0.0

Leave a Comment