scala - 将一系列 ZIO 的所有成功和失败集中在一起？

Question

我有一堆 IO，所有这些都可能成功或失败:

val result: Seq[IO[MyFailure, MySuccess]] = ...

我需要总结结果，以便我可以一起检查所有失败和所有成功:

case class MySummary(failures: Seq[MyFailure], successes: Seq[MySuccess])
def makeSummary(results: Seq[IO[MyFailure, MySuccess]]): UIO[MySummary] = ???

起初这看起来与 foreach 非常相似，但仔细检查后我发现这对这里没有帮助。对于 Future，我通常使用 sequence，对于 Seq[Either[A,B]]，我可以使用 partitionMap。 ZIO 中的等效方法是什么？

Answer 1

您无法使用 Future.sequence 检查所有失败，如果有任何失败，您将只能返回第一个失败。

Future.sequence 的 ZIO 等效项是 ZIO.collectAll，它几乎等同于 ZIO.foreach(result)(identity).

如果您真的需要累积所有错误，那么您应该使用 ZIO.validate 并且对于分区还有 ZIO.partition。

一些比较代码:

  val result: Seq[IO[MyFailure, MySuccess]] = ???

  val withCollectAll: IO[MyFailure, Seq[MySuccess]] = ZIO.collectAll(result)
  val withForeach: IO[MyFailure, Seq[MySuccess]] = ZIO.foreach(result)(identity)
  val withValidate: IO[::[MyFailure], Seq[MySuccess]] = ZIO.validate(result)(identity) //`::` just means that the list cannot be empty
  val withPartition: UIO[(Iterable[MyFailure], Iterable[MySuccess])] = ZIO.partition(result)(identity)

使用 ZIO.partition 的 makeSummary 代码:

  def makeSummary(results: Seq[IO[MyFailure, MySuccess]]): UIO[MySummary] =
    ZIO.partition(results)(identity).map { case (f, s) => MySummary(f.toSeq, s.toSeq) }