假设我有一些可能具有重叠域的部分功能:
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))
这似乎并不干净。有没有办法使用内置案例机制来测试定义和同时评估,可能是通过将它们与测试值或类似的东西进行对比?
答案 0 :(得分:7)
如果定义了PF,则部分函数具有nil
方法lift
,如果未定义,则Some(res)
方法。使用None
时,我们基本上可以忽略flatMap
并仅获取None
值。
Some
如果您喜欢那种东西,请进行压缩:
funcs.flatMap { func => func.lift(num) }
答案 1 :(得分:1)
测试定义并同时评估
那是applyOrElse
。 collect
和lift
使用它来避免双重评估。
这个递归版本说,应用下一个函数并使用它来构建结果,否则只需继续并构建结果的其余部分。
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
。我不得不在索引中查找。
此手动构建避免了Lifted
,Option
和Option
中的某些对象创建。
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)