scala - Scala 中 List[Try[String]] 的分区

Question

我有一个 Try 的 List，我需要按 Failure 和 Success 进行分区，如下:

import scala.util.{Try, Success, Failure}

val list = List(Success(10), Success(20), Failure(new Exception("error1")), Success(30))

val (listTryFailures: List[Try[Int]], listTrySuccesses: List[Try[Int]]) = list.partition(_.isFailure)
listTryFailures: List[Try[Int]] = List(Failure(java.lang.Exception: error1))
listTrySuccesses: List[Try[Int]] = List(Success(10), Success(20), Success(30))

val (listFailures: List[Throwable], listSuccesses: List[Int]) = (listTryFailures.map(_.failed.get), listTrySuccesses.map(_.get))
listFailures: List[Throwable] = List(java.lang.Exception: error1)
listSuccesses: List[Int] = List(10, 20, 30)

是否已经有一个函数可以实现这一点？(而不是调用丑陋的 partition(_.isFailure) + map(_.failed.get) + map(_.get)) ？

ps:另一种选择如下:

val listFailures: List[Throwable] = list.collect { case Failure(f) => f }
val listSuccesses: List[String] = list.collect { case Success(entry) => entry }

// or 
val listFailures: List[Throwable] = list.flatMap(_.failed.toOption)
val listSuccesses: List[String] = list.flatMap(_.toOption)

但我问标准库中是否已存在 partitionSuccess (或类似名称)函数:

val (listFailures: List[Throwable], listSuccesses: List[Int]) = list.partitionSuccess

Answer 1

从 Scala 2.13 开始，大多数集合现在都提供 partitionMap方法，根据返回 Right 或 Left 的函数对元素进行分区。

在我们的例子中，我们可以使用一个函数来调用 partitionMap ，该函数将 Try 转换为 Either 以便分区Successes 为 Right，Failure 为 Left:

// val list = List(Success(10), Success(20), Failure(new Exception("error1")))
val (failures, successes) = list.partitionMap(_.toEither)
// failures: List[Throwable] = List(java.lang.Exception: error1)
// successes: List[Int] = List(10, 20)