我有一个PartialFunction[String,String]
和Map[String,String]
。我想在映射值上应用分部函数,并收集应用它的条目。即给定:
val m = Map( "a"->"1", "b"->"2" )
val pf : PartialFunction[String,String] = {
case "1" => "11"
}
我想以某种方式将_._2
和pf
结合起来,并能够做到这一点:
val composedPf : PartialFunction[(String,String),(String,String)] = /*someMagicalOperator(_._2,pf)*/
val collected : Map[String,String] = m.collect( composedPf )
// collected should be Map( "a"->"11" )
到目前为止,我得到的最好的是:
val composedPf = new PartialFunction[(String,String),(String,String)]{
override def isDefinedAt(x: (String, String)): Boolean = pf.isDefinedAt(x._2)
override def apply(v1: (String, String)): (String,String) = v1._1 -> pf(v1._2)
}
有更好的方法吗?
这里是神奇的运算符:
val composedPf: PartialFunction[(String, String), (String, String)] =
{case (k, v) if pf.isDefinedAt(v) => (k, pf(v))}
另一个不创建组合函数的选项是:
m.filter(e => pf.isDefinedAt(e._2)).mapValues(pf)
Scalaz中有一个函数,它正是这样做的:second
scala> m collect pf.second
res0: scala.collection.immutable.Map[String,String] = Map(a -> 11)
这是有效的,因为PartialFunction
是Arrow(一个广义函数)类型类的实例,而second
是为Arrow定义的常见操作之一。