在 scala 2.10 中绕过 [SI-6976] 进行基元集合模式匹配的最简单方法是什么?



我有一个简单的函数:

  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
    }
  }

通过所有测试

最新更新