如何避免Scala类型参数擦除？基于Breeze库的模式匹配优化

解决类型擦除问题：用ClassTag实现正确匹配 #

首先，要解决DenseVector类型参数擦除导致的模式匹配失效问题，我们可以利用Scala的ClassTag来保留运行时的类型信息。ClassTag是Scala反射库提供的工具，能让我们在运行时获取泛型类型的实际类对象。

下面是修改后的print2方法：

import scala.reflect.ClassTag
import breeze.linalg.DenseVector
type Real = Double

def print2[A](data: DenseVector[A])(implicit ct: ClassTag[A]): Unit = {
  ct.runtimeClass match {
    case cls if cls == classOf[Real] => println("Contains real scalars")
    case cls if cls == classOf[Int] => println("Contains integer scalars")
    case cls if cls == classOf[breeze.linalg.DenseMatrix[Real]] => println("Contains real matrices")
    case _ => println(s"Unsupported element type: ${ct.runtimeClass.getSimpleName}")
  }
}

测试你的示例代码：

val v0 = DenseVector(1.2, 1.5, 1.6)
val v1 = DenseVector(3, 4, 5)
val a = Array(v0, v1)
a.foreach(print2)

运行后会正确输出：

Contains real scalars
Contains integer scalars

更优的设计思路：类型类（Type Class）模式 #

虽然ClassTag能解决问题，但对于需要处理多种数据类型的场景，类型类是更符合Scala函数式风格的方案，它能在编译期保证类型安全，同时具备更好的扩展性。

我们可以定义一个类型类来抽象不同DenseVector元素类型的处理逻辑：

import breeze.linalg.{DenseVector, DenseMatrix}
type Real = Double

// 定义类型类：抽象DenseVector的处理行为
trait DenseVectorHandler[A] {
  def handle(vec: DenseVector[A]): Unit
}

// 为不同的元素类型提供具体实现
object DenseVectorHandler {
  // 处理Real类型元素的DenseVector
  implicit val realHandler: DenseVectorHandler[Real] = 
    (vec: DenseVector[Real]) => println("Contains real scalars")

  // 处理Int类型元素的DenseVector
  implicit val intHandler: DenseVectorHandler[Int] = 
    (vec: DenseVector[Int]) => println("Contains integer scalars")

  // 处理DenseMatrix[Real]类型元素的DenseVector
  implicit val matrixRealHandler: DenseVectorHandler[DenseMatrix[Real]] = 
    (vec: DenseVector[DenseMatrix[Real]]) => println("Contains real matrices")

  // 通用调用方法：隐式获取对应类型的处理逻辑
  def apply[A](vec: DenseVector[A])(implicit handler: DenseVectorHandler[A]): Unit = 
    handler.handle(vec)
}

使用方式非常简洁：

val v0 = DenseVector(1.2, 1.5, 1.6)
val v1 = DenseVector(3, 4, 5)
val v2 = DenseVector(DenseMatrix(1.0 -> 2.0, 3.0 -> 4.0))
val a = Array(v0, v1, v2)

a.foreach(DenseVectorHandler.apply)

类型类方案的优势：

• 编译期安全：如果传入了没有对应处理逻辑的DenseVector类型，编译器会直接报错，避免运行时遗漏case的问题。
• 扩展性强：新增类型时，只需要添加一个新的implicit实现，不需要修改原有代码，符合开闭原则。
• 无冗余包装：不需要像之前那样定义DenseVectorRoot特质和包装样例类，减少代码冗余。

为什么原模式匹配失效？ #

JVM的类型擦除机制会在编译后丢弃泛型类型参数的信息，所以DenseVector[Double]和DenseVector[Int]在运行时都会被视为DenseVector[_]，常规的模式匹配无法区分它们的元素类型。ClassTag通过在编译时嵌入类型信息，让我们能在运行时恢复泛型类型，而类型类则完全绕开了运行时类型检查，在编译期就绑定了对应的处理逻辑。

内容的提问来源于stack exchange，提问作者vkubicki