从flatMap返回多个集合

时间:2012-05-30 15:03:36

标签: scala collections iteration

我正在研究一种方法,它对多个项目有3种可能的结果:错误,无效和成功。对于其中的每一个,我需要返回一个json列表,标识哪些项目出错,无效且成功。

我目前的尝试如下。我使用Object来表示我的对象所在的类,因为完全解释会花费太长时间。 Object类有一个方法process,它返回一个布尔值来表示成功或错误,并在对象无效时抛出异常:

def process(list: List[Objects]) = {
    val successIds = new ListBuffer[Int]();
    val errorIds = new ListBuffer[Int]();
    val invalidIds = new ListBuffer[Int]();

    list.foreach( item => {
        try {
            if (item.process) {
                successIds ++ item.id
            } else {
                errorIds ++ item.id
            }
        } catch {
            case e: Exception => invalidIds ++ item.id
        }
    })

    JsonResult(
        Map("success" -> successIds, 
            "failed" -> errorIds, 
            "invalid" -> invalidIds)
    ) 
}

问题是使用Mutable数据结构不是很“Scala-y”。我更喜欢以更实用的方式构建这些列表,但我对scala很新。关于如何做到这一点的任何想法或提示?

我虽然使用类似flatMap方法的东西,它采用一组集合并以与flatMap方法对单个集合相同的方式对它们进行整理:

def process(list: List[Objects]) = {

    val (success, error, invalid) = list.flatMap( item => {
        try {
            if (item.process) {
                (List(item.id), List.empty, List.empty)
            } else {
                (List.empty, List(item.id), List.empty)
            }
        } catch {
            case e: Exception => 
                (List.empty, List.empty, List(item.id))
        }
    })

    JsonResult(
        Map("success" -> success, 
            "failed" -> error, 
            "invalid" -> invalid)
    ) 
}

3 个答案:

答案 0 :(得分:8)

flatMap不是您需要的 - 您需要groupBy

def process(list: List[Objects]) = {

  def result(x: Objects) =  
    try if (x.process) "success" else "failed"
    catch {case _ => "invalid"}     

  JsonResult(list groupBy result mapValues (_ map (_.id)))
}

答案 1 :(得分:1)

总有递归:

class Ob(val id: Int) { def okay: Boolean = id < 5 }


@annotation.tailrec def process(
  xs: List[Ob], 
  succ: List[Int] = Nil,
  fail: List[Int] = Nil,
  invalid: List[Int] = Nil
): (List[Int], List[Int], List[Int]) = xs match {
  case Nil => (succ.reverse, fail.reverse, invalid.reverse)
  case x :: more =>
    val maybeOkay = try { Some(x.okay) } catch { case e: Exception => None }
    if (!maybeOkay.isDefined) process(more, succ, fail, x.id :: invalid)
    else if (maybeOkay.get)   process(more, x.id :: succ, fail, invalid)
    else                      process(more, succ, x.id :: fail, invalid)
}

这是有希望的(如果您不关心订单,请跳过反转):

scala> process(List(new Ob(1), new Ob(7), new Ob(2), 
   new Ob(4) { override def okay = throw new Exception("Broken") }))

res2: (List[Int], List[Int], List[Int]) = (List(1,2),List(7),List(4))

答案 2 :(得分:0)

适合于在没有“Objects”的情况下进行编译

def procex (item: String): Boolean = ((9 / item.toInt) < 1)

def process (list: List[String]) = {
    val li: List[(Option[String], Option[String], Option[String])] = list.map (item => {
        try {
            if (procex (item)) {
                (Some (item), None, None)
            } else {
                (None, Some (item), None)
            }
        } catch {
            case e: Exception => 
                (None, None, Some (item))
        }
    })
    li
}
// below 10 => failure
val in = (5 to 15).map (""+_).toList
// 0 to throw a little exception
val ps = process ("0" :: in)

val succeeders = ps.filter (p=> p._1 != None).map (p=>p._1)
val errors     = ps.filter (p=> p._2 != None).map (p=>p._2)
val invalides  = ps.filter (p=> p._3 != None).map (p=>p._3)

什么行不通:

(1 to 3).map (i=> ps.filter (p=> p._i != None).map (p=>p._i))

_i不起作用。