Преобразовать список [Либо [A, B]] в Либо [Список [A], Список [B]] - PullRequest
Новая
       21

Преобразовать список [Либо [A, B]] в Либо [Список [A], Список [B]]

6 голосов
/ 07 июня 2019

Как преобразовать List[Either[String, Int]] в Either[List[String], List[Int]], используя метод, подобный кошкам sequence ? Например, xs.sequence в следующем коде 

import cats.implicits._
val xs: List[Either[String, Int]] = List(Left("error1"), Left("error2"))
xs.sequence

возвращает Left(error1) вместо обязательного Left(List(error1, error2)).

KevinWrights ' ответ предлагает 

val lefts = xs collect {case Left(x) => x }
def rights = xs collect {case Right(x) => x}
if(lefts.isEmpty) Right(rights) else Left(lefts)

, который возвращает Left(List(error1, error2)), однако предоставляет ли кошка готовую последовательность, которая бы собирала все левые?

Ответы [ 4 ]

5 голосов
/ 08 июня 2019

Еще одна вариация на ту же тему (аналогично этому ответу ), включая весь импорт: 

import scala.util.Either
import cats.data.Validated
import cats.syntax.traverse._
import cats.instances.list._

def collectErrors[A, B](xs: List[Either[A, B]]): Either[List[A], List[B]] = {
  xs.traverse(x => Validated.fromEither(x.left.map(List(_)))).toEither
}

Если дополнительно импортировать cats.syntax.either._, тогда становится доступным toValidated, так что вы также можете написать: 

xs.traverse(_.left.map(List(_)).toValidated).toEither

, и если вы дополнительно замените left.map на bimap(..., identity), вы получите чудесно лаконичное решение @ DmytroMitin.

4 голосов
/ 08 июня 2019

Это решение не использует cats, но из Scala 2.13 вы можете использовать partitionMap: 

    def convert[L,R](input: List[Either[L,R]]): Either[List[L], List[R]] = {
      val (left, right) = input.partitionMap(identity)
      if (left.isEmpty) Right(right) else Left(left)
    }

    println(convert(List(Left("error1"), Left("error2"))))
    // Left(List(error1, error2))
    println(convert(List(Right(1), Left("2"), Right(3), Left("4"))))
    // Left(List(2, 4))
    println(convert(List(Right(1), Right(2), Right(3), Right(4))))
    // Right(List(1, 2, 3, 4))
3 голосов
/ 08 июня 2019

Попробуйте 

xs.traverse(_.toValidated.bimap(List(_), identity)).toEither

// List(Left("error1"), Left("error2")) => Left(List("error1", "error2"))
// List(Right(10), Right(20))           => Right(List(10, 20))
// List(Right(10), Left("error2"))      => Left(List("error2"))
2 голосов
/ 08 июня 2019

Примечание: в этом ответе предполагается, что вы хотите получить Left, как только один элемент в списке равен единице.

Вы можете использовать .separate(), чтобы разбить список на кортеж (List[String], List[Int])и сопоставьте это: 

scala> xs.separate match {
          case (Nil, rights) => Right(rights)
          case (lefts, _)    => Left(lefts)
       }
res0: scala.util.Either[List[String],List[Int]] = Left(List(error1, error2))

Не «из коробки», поскольку вам все еще нужно совпадение, но все еще довольно короткое (и, надеюсь, простое для понимания).

...