收集单个值的多个部分函数的结果?

时间:2015-06-08 19:14:23

标签: scala pattern-matching partialfunction

假设我有一些可能具有重叠域的部分功能:

val funcs: Seq[PartialFunction[Any, Int]] = Vector(
{ case i: Int if i % 2 == 0 => i*2 }
,
{ case i: Int if i % 2 == 1 => i*3 }
,
{ case i: Int if i % 6 == 0 => i*5 }
)

我想将它们应用于某个值并收集该val定义的任何结果。我认为有一种优雅的方式可以使用collect,但我能做的最好的是

val num = 66
funcs.collect { case func if func.isDefinedAt(num) => func(num) }
// or 
funcs.filter(_.isDefinedAt(num)).map(_(num))

这似乎并不干净。有没有办法使用内置案例机制来测试定义和同时评估,可能是通过将它们与测试值或类似的东西进行对比?

2 个答案:

答案 0 :(得分:7)

如果定义了PF,则部分函数具有nil方法lift,如果未定义,则Some(res)方法。使用None时,我们基本上可以忽略flatMap并仅获取None值。

Some

如果您喜欢那种东西,请进行压缩:

funcs.flatMap { func => func.lift(num) }

答案 1 :(得分:1)

  

测试定义并同时评估

那是applyOrElsecollectlift使用它来避免双重评估。

这个递归版本说,应用下一个函数并使用它来构建结果,否则只需继续并构建结果的其余部分。

def f(ff: List[PartialFunction[Any, Int]]): List[Int] = ff match {
  case hd :: tail => hd.applyOrElse(num, (_: Any) => return f(tail)) :: f(tail)
  case _ => Nil
}

这是我第一次在Scala中使用return。我不得不在索引中查找。

此手动构建避免了LiftedOptionOption中的某些对象创建。

scala> :pa
// Entering paste mode (ctrl-D to finish)

val funcs: Seq[PartialFunction[Any, Int]] = Vector(
{ case i: Int if i % 2 == 0 => i*2 }
,
{ case i: Int if i % 2 == 1 => i*3 }
,
{ case i: Int if i % 6 == 0 => i*5 }
)
val num = 66

def f(ff: List[PartialFunction[Any, Int]]): List[Int] = ff match {
  case hd :: tail => hd.applyOrElse(num, (_: Any) => return f(tail)) :: f(tail)
  case _ => Nil
}

// Exiting paste mode, now interpreting.

funcs: Seq[PartialFunction[Any,Int]] = Vector(<function1>, <function1>, <function1>)
num: Int = 66
f: (ff: List[PartialFunction[Any,Int]])List[Int]

scala> f(funcs.toList)
res0: List[Int] = List(132, 330)

scala> funcs flatMap (_ lift num)
res1: Seq[Int] = Vector(132, 330)

第二个版本只使用一个标志来表达同样的想法。

scala> :pa
// Entering paste mode (ctrl-D to finish)

def f(ff: List[PartialFunction[Any, Int]]): List[Int] = ff match {
  case hd :: tail =>
    var ok = true
    val x  = hd.applyOrElse(num, (_: Any) => { ok = false ; 0 })
    if (ok) x :: f(tail) else f(tail)
  case _ => Nil
}

// Exiting paste mode, now interpreting.

f: (ff: List[PartialFunction[Any,Int]])List[Int]

scala> f(funcs.toList)
res2: List[Int] = List(132, 330)