使用 Scala 进行模式匹配

Question

我有一个必须使用 Double 和 Float 的类。由于性能要求，我对(Double, Float) 使用带有@specialized 注释的泛型。我需要调用两个第三方函数。 ffunc(x: Float) 接受 Float，dfunc(y: Double) 接受 Double。在某些时候，我必须致电ffunc 或dfunc。为此，我使用 scala 模式匹配。我的代码如下所示：

class BoxingTest[@specialized(Double, Float) T] {
  def foo(p: T): Unit = {
    p match {
      case dp: Double => dfunc(dp)
      case df: Float => ffunc(df)
    }
  }
}

但是，scala 编译器为专用版本提供了未优化的字节码。当我查看专用类的字节码时，我看到了将我的专用类型转换为对象的非优化匹配代码。还有额外的装箱/拆箱如下：

41: invokestatic  #21                 // Method scala/runtime/BoxesRunTime.boxToDouble:(D)Ljava/lang/Double;
44: invokestatic  #39                 // Method scala/runtime/BoxesRunTime.unboxToDouble:(Ljava/lang/Object;)D

您能否建议将匹配代码替换为将被优化并避免装箱/拆箱的代码？

Answer 1

这是以前出现的。我认为最好的办法是覆盖专门的方法：

scala> class X[@specialized(Int, Double) A] { def f(a: A): String = ??? }
defined class X

scala> trait T { def f$mcI$sp(i: Int): String = "my int" }
defined trait T

scala> val x = new X[Int] with T
x: X[Int] with T = $anon$1@6137cf6e

scala> x.f(42)
res0: String = my int

那可能是在 SO 上。