我有一个简单的函数:
implicit class ArrayView[A](self: Array[A]) {
def filterByType[B <: A: ClassTag]: Array[B] = {
val ctg = implicitly[ClassTag[B]]
val runtimeClazz = ctg.runtimeClass
self.flatMap{
v =>
if (v.getClass.isAssignableFrom(runtimeClazz)) Some(v.asInstanceOf[B])
else None
}.toArray
}
}
和一个简单的测试用例:
assert(Array(1, 2.2, "a").filterByType[Int].toSeq == Seq(1))
assert(Array(1, 2.2, "a").filterByType[java.lang.Integer].toSeq == Seq(1: java.lang.Integer))
assert(Array(1, 2.2, "a").filterByType[Double].toSeq == Seq(2.2))
assert(Array(1, 2.2, "a").filterByType[java.lang.Double].toSeq == Seq(2.2: java.lang.Double))
assert(Array(1, 2.2, "a").filterByType[String].toSeq == Seq("a"))
它只适用于Scala>2.11.3(因为https://issues.scala-lang.org/browse/SI-6967,这已经造成了一段时间),但许多库(例如spark-hive_fructiveserver)在移植到2.11时仍然存在问题,而2.10在大多数其他情况下基本上是稳定的,有没有针对这个错误的快速补丁?
好吧,这真的很时髦,但不管怎样:
implicit class ArrayView[A](self: Array[A]) {
def filterByType[B <: A: ClassTag]: Array[B] = {
self.flatMap{
v =>
try {
Array[B](v.asInstanceOf[B])
}
catch {
case e: Throwable =>
Array[B]()
}
}.toArray
}
}
通过所有测试