diff options
author | Jason Zaugg <jzaugg@gmail.com> | 2012-11-26 14:39:22 -0800 |
---|---|---|
committer | Jason Zaugg <jzaugg@gmail.com> | 2012-11-26 14:39:22 -0800 |
commit | 72f72811582af9a9c41dbc82a74d771198024a57 (patch) | |
tree | 275eddcf4ab55fb04e9c5224c6c1da69de6c0449 /src/main/scala | |
parent | 3fd51865dfcb6121f84145f4504abd0f80bf6cca (diff) | |
parent | 4eb47ba34bf10bc483fa8f04a9f76708f9161317 (diff) | |
download | scala-async-72f72811582af9a9c41dbc82a74d771198024a57.tar.gz scala-async-72f72811582af9a9c41dbc82a74d771198024a57.tar.bz2 scala-async-72f72811582af9a9c41dbc82a74d771198024a57.zip |
Merge pull request #44 from phaller/topic/exception-handling
Topic/exception handling
Diffstat (limited to 'src/main/scala')
-rw-r--r-- | src/main/scala/scala/async/ExprBuilder.scala | 29 | ||||
-rw-r--r-- | src/main/scala/scala/async/TransformUtils.scala | 2 |
2 files changed, 26 insertions, 5 deletions
diff --git a/src/main/scala/scala/async/ExprBuilder.scala b/src/main/scala/scala/async/ExprBuilder.scala index 7b4ccb8..ef27672 100644 --- a/src/main/scala/scala/async/ExprBuilder.scala +++ b/src/main/scala/scala/async/ExprBuilder.scala @@ -30,7 +30,7 @@ private[async] final case class ExprBuilder[C <: Context, FS <: FutureSystem](c: def mkHandlerCaseForState: CaseDef - def mkOnCompleteHandler: Option[CaseDef] = None + def mkOnCompleteHandler[T: c.WeakTypeTag]: Option[CaseDef] = None def stats: List[Tree] @@ -75,13 +75,32 @@ private[async] final case class ExprBuilder[C <: Context, FS <: FutureSystem](c: mkHandlerCase(state, stats :+ callOnComplete) } - override def mkOnCompleteHandler: Option[CaseDef] = { + override def mkOnCompleteHandler[T: c.WeakTypeTag]: Option[CaseDef] = { val tryGetTree = Assign( Ident(awaitable.resultName), TypeApply(Select(Select(Ident(name.tr), Try_get), newTermName("asInstanceOf")), List(TypeTree(awaitable.resultType))) ) - Some(mkHandlerCase(state, List(tryGetTree, mkStateTree(nextState), mkResumeApply))) + + /* if (tr.isFailure) + * result$async.complete(tr.asInstanceOf[Try[T]]) + * else { + * <resultName> = tr.get.asInstanceOf[<resultType>] + * <nextState> + * <mkResumeApply> + * } + */ + val ifIsFailureTree = + If(Select(Ident(name.tr), Try_isFailure), + futureSystemOps.completeProm[T]( + c.Expr[futureSystem.Prom[T]](Ident(name.result)), + c.Expr[scala.util.Try[T]]( + TypeApply(Select(Ident(name.tr), newTermName("asInstanceOf")), + List(TypeTree(weakTypeOf[scala.util.Try[T]]))))).tree, + Block(List(tryGetTree, mkStateTree(nextState), mkResumeApply): _*) + ) + + Some(mkHandlerCase(state, List(ifIsFailureTree))) } override val toString: String = @@ -276,7 +295,7 @@ private[async] final case class ExprBuilder[C <: Context, FS <: FutureSystem](c: trait AsyncBlock { def asyncStates: List[AsyncState] - def onCompleteHandler: Tree + def onCompleteHandler[T: c.WeakTypeTag]: Tree def resumeFunTree[T]: Tree } @@ -320,7 +339,7 @@ private[async] final case class ExprBuilder[C <: Context, FS <: FutureSystem](c: * resume() * } */ - val onCompleteHandler: Tree = Match(Ident(name.state), initStates.flatMap(_.mkOnCompleteHandler).toList) + def onCompleteHandler[T: c.WeakTypeTag]: Tree = Match(Ident(name.state), initStates.flatMap(_.mkOnCompleteHandler[T]).toList) /** * def resume(): Unit = { diff --git a/src/main/scala/scala/async/TransformUtils.scala b/src/main/scala/scala/async/TransformUtils.scala index 5b1fcbe..7571f88 100644 --- a/src/main/scala/scala/async/TransformUtils.scala +++ b/src/main/scala/scala/async/TransformUtils.scala @@ -155,6 +155,8 @@ private[async] final case class TransformUtils[C <: Context](c: C) { } val Try_get = methodSym(reify((null: scala.util.Try[Any]).get)) + val Try_isFailure = methodSym(reify((null: scala.util.Try[Any]).isFailure)) + val TryClass = c.mirror.staticClass("scala.util.Try") val TryAnyType = appliedType(TryClass.toType, List(definitions.AnyTpe)) val NonFatalClass = c.mirror.staticModule("scala.util.control.NonFatal") |