Use fold instead of map+getOrElse

aztek · Jun 24, 2014 · 53954b2 · 53954b2
1 parent e4ed918
commit 53954b2
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/main/scala/scala/workflow/package.scala b/src/main/scala/scala/workflow/package.scala
@@ -201,20 +201,24 @@ package object workflow extends Instances {
           val (newscope, newexpr) = rewrite(scope.enter)(expr)
           val frame = newscope.materialized.last
           if (frame.isEmpty) {
-            val name = signature map { case (_, name, _) ⇒ name } getOrElse TermName("_")
+            val name = signature.fold(TermName("_")) {
+              case (_, name, _) ⇒ name
+            }
             val tpe = typeCheck(newexpr, newscope).get.tpe
             val bind = Bind(name, TypeTree(tpe), newexpr)
-            val cont = (_: Boolean) ⇒ signature map {
-                                        case (mods, _, tpt) ⇒ block(ValDef(mods, name, tpt, newexpr))
-                                      } getOrElse block(q"$newexpr")
+            val cont = (_: Boolean) ⇒ signature.fold(block(q"$newexpr")) {
+              case (mods, _, tpt) ⇒ block(ValDef(mods, name, tpt, newexpr))
+            }
 
             val newerscope = if (signature.isDefined) scope :+ bind else scope
 
             (newerscope ++ newscope.frames.head, cont)
           } else {
             val value = apply(frame)(newexpr)
             val tpe = typeCheck(newexpr, newscope).get.tpe
-            val name = signature map { case (_, term, _) ⇒ term } getOrElse TermName("_")
+            val name = signature.fold(TermName("_")) {
+              case (_, name, _) ⇒ name
+            }
             val bind = Bind(name, TypeTree(tpe), value)
             val cont = (x: Boolean) ⇒ if (x) >>=(bind) compose lambda(bind) // Especially dirty hack!
                                       else   map(bind) compose lambda(bind) // TODO: figure out a better way