我的代码中有一个场景,我需要Future
根据包含某个值的成功结果来失败。我可以通过 完成这项工作flatMap
,但我想知道是否有更清洁的方法来完成这项工作。首先,一个非常简化的例子:
import concurrent._
case class Result(successful:Boolean)
object FutureTest {
def main(args: Array[String]) {
import ExecutionContext.Implicits._
val f = Future{Result(false)}.flatMap{ result =>
result match{
case Result(false) => Promise.failed(new Exception("The call failed!!")).future
case _ => Promise.successful(result).future
}
}
f onFailure{
case x => println(x.getMessage())
}
}
}
因此,在我的示例中,如果返回的成功指示器的值为,我希望Future
失败。正如我所提到的,我可以使用 来完成这项工作,但我想删除的代码行是:Result
false
flatMap
case _ => Promise.successful(result).future
这种说法似乎没有必要。我想要的行为是能够定义条件,如果它评估为真,允许我返回一个不同Future
的,但如果它不是真的,就保持原样(有点像PartialFunction
语义。有没有办法做到这一点,我只是没有看到?我看过collect
并且transform
那些似乎也不合适。
编辑
在得到map
@Rex Kerr 和@senia 的建议后,我创建了PimpedFuture
一个隐式转换来美化代码,如下所示:
class PimpedFuture[T](f:Future[T])(implicit ex:ExecutionContext){
def failWhen(pf:PartialFunction[T,Throwable]):Future[T] = {
f map{
case x if (pf.isDefinedAt(x)) => throw pf(x)
case x => x
}
}
}
隐含的
implicit def futToPimpedFut[T](fut:Future[T])(implicit ec:ExecutionContext):PimpedFuture[T] = new PimpedFuture(fut)
以及新的处理代码:
val f = Future{ Result(false) } failWhen {
case Result(false) => new Exception("The call failed!!")
}
我认为这更清洁,同时仍然使用建议map
。