Scala:如何以功能方式更改此代码?

时间:2015-01-31 08:55:01

标签: scala functional-programming

我有这个列表例如:

val list = List(
    Map("a" -> "John", "b" -> 20, "c" -> true), 
    Map("a" -> "Derp", "b" -> 10, "c" -> false), 
    Map("b" -> 8, "c" -> true),
    Map("a" -> "asdf", "b" -> 50, "c" -> true)
)

我有这个代码处理上面的列表,我想以最有效的方式将其更改为功能样式:

var result = ""
breakable {
    for (m <- list) {
        if (m.get("a").isEmpty) {
            result = "Error A"
            break()
        }
        if (m.get("b").isEmpty) {
            result = "Error B"
            break()
        }
    }
}

在上面的代码中,如果&#34; a&#34;不存在,if检查只在退出循环之前完成一次。如果&#34; b&#34;不存在,检查完成两次。

我尝试将其更改为功能代码:

val result = (for { 
    m <- list
    a = m.get("a").isEmpty
    b = m.get("b").isEmpty
    if a || b
} yield {
    if (a) 
        "Error A"
    else
        "Error B"
}).headOption.getOrElse("")

它看起来并不像是&#34; a&#34;不存在,if检查将完成两次,如果&#34; b&#34;不存在,它将完成三次。

还有.headOption.getOrElse("")开销。如果我能直接得到字符串结果会很好。

任何人都可以提供更好的解决方案吗?

2 个答案:

答案 0 :(得分:3)

list.view
    .map(m => if (m.get("a").isEmpty) "Error A" 
              else if (m.get("b").isEmpty) "Error B"
              else "")
    .find(_.nonEmpty)
    .getOrElse("")

注意视图使其不严格,因此它在第一个失败元素之后停止。

将if语句作为部分函数略微整理,如果地图中既没有“a”或“b”也没有定义,并且使用contains

list.collectFirst{case m if !(m contains "a") => "Error A"
                  case m if !(m contains "b") => "Error B"}
    .getOrElse("")

使用索引

list.view
    .zipWithIndex
    .map{case (m,i) => if (m.get("a").isEmpty) "Error A:"+ i
                       else if (m.get("b").isEmpty) "Error B:" + i
                       else ""}
    .find(_.nonEmpty)
    .getOrElse("")
    //> res1: String = Error A:2

并使用collectFirst

list.zipWithIndex
    .collectFirst{case (m, i) if !(m contains "a") => "Error A:" + i
                  case (m, i) if !(m contains "b") => "Error B:" + i}
    .getOrElse("")

答案 1 :(得分:0)

仅作为附录,另一个可行的解决方案是使用递归:

def getError(list: List[Map[String, Any]]): String = {
    if (list.nonEmpty) {
        val m = list.head
        if (m.get("a").isEmpty) 
            "Error A"
        else if (m.get("b").isEmpty)
            "Error B"
        else
            getError(list.tail)
    } else ""
}