diff options
Diffstat (limited to 'src/main/scala/scala/async/Async.scala')
-rw-r--r-- | src/main/scala/scala/async/Async.scala | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/main/scala/scala/async/Async.scala b/src/main/scala/scala/async/Async.scala index 106bdf0..7d9ddf5 100644 --- a/src/main/scala/scala/async/Async.scala +++ b/src/main/scala/scala/async/Async.scala @@ -103,7 +103,10 @@ abstract class AsyncBase { import asyncBlockBuilder.asyncStates logDiagnostics(c)(anfTree, asyncStates.map(_.toString)) val initStates = asyncStates.init - val localVarTrees = asyncStates.flatMap(_.allVarDefs).toList + val localVarTrees = anfTree.collect { + case vd@ValDef(_, _, tpt, _) if renameMap contains vd.symbol => + builder.mkVarDefTree(tpt.tpe, renameMap(vd.symbol)) + } /* lazy val onCompleteHandler = (tr: Try[Any]) => state match { |