Add missing recursion to branchWithAssign

retronym · retronym · commit ad1905a99ddd · 2017-10-16T07:41:41.000-05:00
diff --git a/src/main/scala/scala/async/internal/AnfTransform.scala b/src/main/scala/scala/async/internal/AnfTransform.scala
@@ -118,7 +118,7 @@ private[async] trait AnfTransform {
                     case MatchEnd(ld) =>
                       deriveLabelDef(ld, branchWithAssign)
                     case blk @ Block(thenStats, thenExpr) =>
-                      treeCopy.Block(blk, thenStats, typedAssign(thenExpr)).setType(definitions.UnitTpe)
+                      treeCopy.Block(blk, thenStats, branchWithAssign(thenExpr)).setType(definitions.UnitTpe)
                     case _ =>
                       typedAssign(t)
                   }

Original file line number	Diff line number	Diff line change
`@@ -118,7 +118,7 @@ private[async] trait AnfTransform {`
`118`	`118`	`case MatchEnd(ld) =>`
`119`	`119`	`deriveLabelDef(ld, branchWithAssign)`
`120`	`120`	`case blk @ Block(thenStats, thenExpr) =>`
`121`		`- treeCopy.Block(blk, thenStats, typedAssign(thenExpr)).setType(definitions.UnitTpe)`
	`121`	`+ treeCopy.Block(blk, thenStats, branchWithAssign(thenExpr)).setType(definitions.UnitTpe)`
`122`	`122`	`case _ =>`
`123`	`123`	`typedAssign(t)`
`124`	`124`	`}`