描绘成功和失败的未来

Question

我有一个未来[T]，我想在成功和失败上映射结果。

比如说

val future = ... // Future[T]
val mapped = future.mapAll { 
  case Success(a) => "OK"
  case Failure(e) => "KO"
}

如果我使用< code>map或< code>flatmap，它将只映射成功的未来。如果我使用< code>recover，它将只映射失败的期货。< code>onComplete执行回调，但不返回修改后的未来值。< code>Transform可以工作，但是需要两个函数而不是一个部分函数，所以有点难看。

我知道我可以做出一个新的promise，并用onComplete或onSuccess/onFailure完成它，但我希望我缺少一些东西，可以让我用一个PF完成上述工作。

Answer 1

地图和平面地图变体：

implicit class FutureExtensions[T](f: Future[T]) {
  def mapAll[Target](m: Try[T] => Target)(implicit ec: ExecutionContext): Future[Target] = {
    val promise = Promise[Target]()
    f.onComplete { r => promise success m(r) }(ec)
    promise.future
  }

  def flatMapAll[Target](m: Try[T] => Future[Target])(implicit ec: ExecutionContext): Future[Target] = {
    val promise = Promise[Target]()
    f.onComplete { r => m(r).onComplete { z => promise complete z }(ec) }(ec)
    promise.future
  }
}

Answer 2

匿名用户

从Scala 2.12开始，您可以使用< code>transform来映射这两种情况:

future.transform {
      case Success(_) => Try("OK")
      case Failure(_) => Try("KO")
}

如果您喜欢使用< code>Future而不是< code>Try，也可以使用< code>transformWith。有关详细信息，请查看文档。

Answer 3

编辑2017-09-18：从Scala 2.12开始，有一个转换方法，它采用尝试[T]=

val future = ... // Future[T]
val mapped = future.transform {
  case Success(_) => Success("OK")
  case Failure(_) => Success("KO")
}

对于2.11.x，以下内容仍然适用：

好了，你不能直接用一个PF来做这件事。而变换变换是可丢弃的=

val mapped: Future[String] = future.map(_ => "OK").recover{case _ => "KO"}

也就是说，实现mapAll很简单:

implicit class RichFuture[T](f: Future[T]) {
  def mapAll[U](pf: PartialFunction[Try[T], U]): Future[U] = {
    val p = Promise[U]()
    f.onComplete(r => p.complete(Try(pf(r))))
    p.future
  }
}