Better fors
The betterFors
language extension improves the usability of for
-comprehensions.
The extension is enabled by the language import import scala.language.experimental.betterFors
or by setting the command line option -language:experimental.betterFors
.
The biggest user facing change is the new ability to start for
-comprehensions with with aliases. This means that the following previously invalid code is now valid:
for
as = List(1, 2, 3)
bs = List(4, 5, 6)
a <- as
b <- bs
yield a + b
The desugaring of this code is the same as if the aliases were introduced with val
:
val as = List(1, 2, 3)
val bs = List(4, 5, 6)
for
a <- as
b <- bs
yield a + b
Additionally this extension changes the way for
-comprehensions are desugared. The desugaring is now done in a more intuitive way and the desugared code can be more efficient, because it avoids some unnecessary method calls. There are two main changes in the desugaring:
-
Simpler Desugaring for Pure Aliases: When an alias is not followed by a guard, the desugaring is simplified. The last generator and the aliases don't have to be wrapped in a tuple, and instead the aliases are simply introduced as local variables in a block with the next generator. Current Desugaring:
for { a <- doSth(arg) b = a } yield a + b
Desugars to:
doSth(arg).map { a => val b = a (a, b) }.map { case (a, b) => a + b }
New Desugaring:
doSth(arg).map { a => val b = a a + b }
This change makes the desugaring more intuitive and avoids unnecessary
map
calls, when an alias is not followed by a guard. -
Avoiding Redundant
map
Calls: When the result of thefor
-comprehension is the same expression as the last generator pattern, the desugaring avoids an unnecessarymap
call. but th eequality of the last pattern and the result has to be able to be checked syntactically, so it is either a variable or a tuple of variables. Current Desugaring:for { a <- List(1, 2, 3) } yield a
Desugars to:
List(1, 2, 3).map(a => a)
New Desugaring:
List(1, 2, 3)
For more details on the desugaring scheme see the comment in Desugar.scala#makeFor
.