看起来您正在尝试将您的函数应用于集合的最里面的项目,而不管您的值嵌套有多深(在这种情况下,List[T]
和List[List[T]]
)。
trait CanMapInner[WrappedV, WrappedB, V, B] {
def mapInner(in: WrappedV, f: V ⇒ B): WrappedB
}
// simple base case (no nesting involved).
implicit def simpleMapper[V, B] = new CanMapInner[V, B, V, B] {
def mapInner(in: V, f: (V) ⇒ B): B = f(in)
}
// drill down one level of "List".
implicit def wrappedMapper[V, B, InnerV, InnerB](implicit innerMapper: CanMapInner[InnerV, InnerB, V, B]) =
new CanMapInner[List[InnerV], List[InnerB], V, B] {
def mapInner(in: List[InnerV], f: (V) ⇒ B): List[InnerB] =
in.map(innerMapper.mapInner(_, f))
}
implicit class XXX[WrappedV](list: List[WrappedV]) {
def xxx[V, B, WrappedB](f: V ⇒ B)(implicit mapper: CanMapInner[WrappedV, WrappedB, V, B]) = {
list.map(inner ⇒ mapper.mapInner(inner, f))
}
}
改编自qmajor的解决方案Map
。
用法:
def f(i: Int): String = "Hello " + i.toString
val source1: List[List[Int]] = List(List(1, 2), List(3, 4))
val source2: List[Int] = List(1, 2)
val result1: List[List[String]] = source1.xxx(f)
val result2: List[String] = source2.xxx(f)
Console println source1
// > List(List(1, 2), List(3, 4))
Console println source2
// > List(1, 2)
Console println result1
// > List(List(Hello 1, Hello 2), List(Hello 3, Hello 4))
Console println result2
// > List(Hello 1, Hello 2)
f
我出于演示目的更改了您的功能。