Question

我有一个“字符串”列表（字符串类的包装，名为Str），其中一些具有混合特征。在某个时间点，我需要区分mixin特性以提供其他功能。

我的代码可以恢复为此并且可以正常工作：

case class Str(s: String)
trait A
trait B

object GenericsPatternMatch extends {
  def main(args: Array[String]): Unit = {

    val listOfStr: Seq[Str] =
      Seq(
        Str("String"),
        new Str("String A") with A, // Some trait mixins
        new Str("String B") with B
      )

    println("A: " + selectStrA(listOfStr))
    println("B: " + selectStrB(listOfStr))
  }

  val selectStrA: Seq[Str] => Seq[Str with A] = (strList: Seq[Str]) => strList.collect { case s: A => s }
  val selectStrB: Seq[Str] => Seq[Str with B] = (strList: Seq[Str]) => strList.collect { case s: B => s }
}

为了保持代码符合DRY原则，我想泛化selectStr函数。我的第一次尝试是：

def selectStrType[T](strList: Seq[Str]): Seq[Str with T] =
    strList.collect { case f: Str with T => f }

但是，由于JVM运行时类型擦除功能（限制？），编译器会发出警告，但它不起作用，很可能是因为它将与Object匹配所有内容：

Warning:(31, 31) abstract type pattern T is unchecked since it is eliminated by erasure
    strList.collect { case f: Str with T => f }

经过几个小时的搜索和学习，我想到了：

def selectStrType[T: ClassTag](strList: Seq[Str]): Seq[Str with T] =
    strList.collect {
      case f: Str if classTag[T].runtimeClass.isInstance(f) => f.asInstanceOf[Str with T]
    }

使用这种方法，我现在可以选择如下特定特征：

val selectStrA: Seq[Str] => Seq[Str with A] = (strList: Seq[Str]) => selectStrType[A](strList: Seq[Str])
val selectStrB: Seq[Str] => Seq[Str with B] = (strList: Seq[Str]) => selectStrType[B](strList: Seq[Str])

我相信可能有一种方法可以改善selectStrType函数，即：

简化if条件
删除显式强制转换“ .asInstanceOf [Str with T]”，但仍返回Seq [Str with T]

你能帮我吗？

Answer 1

您可以按以下方法定义方法，它将起作用。

def selectStrType[T: ClassTag](strList: Seq[Str]): Seq[Str with T] =
  strList.collect { case f: T => f }

由于受ClassTag上下文的约束，仅在T上进行类型匹配将起作用（理想情况下Str with T也应起作用，但这似乎是一个限制）。现在，编译器知道f的类型为Str，并且类型也为T，或者换句话说Str with T，因此可以进行编译。它将做正确的事情：

scala> selectStrType[A](listOfStr)
res3: Seq[Str with A] = List(Str(String A))

scala> selectStrType[B](listOfStr)
res4: Seq[Str with B] = List(Str(String B))

编辑：更正，看来这将在 Scala 2.13版本中起作用。在2.12中，您需要稍微帮助编译器：

def selectStrType[T: ClassTag](strList: Seq[Str]): Seq[Str with T] =
  strList.collect { case f: T => f: Str with T }

泛型的Scala模式匹配

1 个答案: