scalaz,Disjunction.sequence返回左侧列表

时间:2016-09-13 13:57:38

标签: scala scalaz

在scalaz 7.2.6中,我想在sequence上实现Disjunction,这样如果有一个或多个左边,它会返回一个列表,而不是只取第一个(如Disjunction.sequenceU):

import scalaz._, Scalaz._

List(1.right, 2.right, 3.right).sequence
res1: \/-(List(1, 2, 3))

List(1.right, "error2".left, "error3".left).sequence
res2: -\/(List(error2, error3))

我已经按照以下方式实现了它并且它可以工作,但它看起来很难看。是否有getRight方法(例如scala Either类,Right[String, Int](3).right.get)?以及如何改进此代码?

implicit class RichSequence[L, R](val l: List[\/[L, R]]) {
  def getLeft(v: \/[L, R]):L = v match { case -\/(x) => x }
  def getRight(v: \/[L, R]):R = v match { case \/-(x) => x }

  def sequence: \/[List[L], List[R]] =
    if (l.forall(_.isRight)) {
      l.map(e => getRight(e)).right
    } else {
      l.filter(_.isLeft).map(e => getLeft(e)).left
    }
}

1 个答案:

答案 0 :(得分:1)

在我周围实现了一个递归函数,但最好的选择是使用separate

implicit class RichSequence[L, R](val l: List[\/[L, R]]) {
  def sequence: \/[List[L], List[R]] = {
    def seqLoop(left: List[L], right: List[R], list: List[\/[L, R]]): \/[List[L], List[R]] =
      list match {
        case (h :: t) =>
          h match {
            case -\/(e) => seqLoop(left :+ e, right, t)
            case \/-(s) => seqLoop(left, right :+ s, t)
          }
        case Nil =>
          if(left.isEmpty) \/-(right)
          else -\/(left)
    }
    seqLoop(List(), List(), l)
  }


  def sequenceSeparate: \/[List[L], List[R]] = {
    val (left, right) = l.separate[\/[L, R], L, R]
    if(left.isEmpty) \/-(right)
    else -\/(left)
  }
}

第一个只收集结果,最后决定如何处理这些,第二个基本相同,除了递归函数更简单,我没有考虑性能,我用过:+,如果你关心使用前置或其他集合。

您可能还想查看与Validation累积失败不同的ValidationNELDisjunction