Scala AST中classOf表达式的表示及匹配模式咨询

嘿，刚好对Scala AST这块比较熟，来给你掰扯清楚！

首先得明确：Scala里的classOf[X]并不是一个特殊的语法糖节点，而是编译器把它解析成了对scala.Predef.classOf方法的调用——但因为这个方法接收的是类型参数而非普通的值参数，所以它在AST里对应的是TypeApply节点，而不是你之前尝试的Apply节点（Apply专门用来处理带值参数的方法调用）。

举个具体的例子，classOf[org.apache.commons.lang3.ArrayUtils]的AST结构大概是这样的：

TypeApply(
  // 指向Predef.classOf的引用，默认导入下会简化成Ident(TermName("classOf"))
  Select(Select(Ident(TermName("scala")), TermName("Predef")), TermName("classOf")),
  // 类型参数列表，这里就是你要的ArrayUtils类型节点
  List(Ident(TypeName("org.apache.commons.lang3.ArrayUtils")))
)

如果代码里已经默认导入了scala.Predef._（Scala默认会做这件事），那前面的Select链会简化成直接的Ident(TermName("classOf"))，不用写全限定名。

既然知道它是TypeApply节点，我们就可以针对性地写匹配逻辑，这里给你三种常用的写法：

1. 基础匹配（默认导入场景） #

如果你的代码都是默认导入Predef的情况，直接匹配TypeApply的主体是classOf的标识符就行：

import scala.reflect.internal.Trees.Tree

def matchClassOf(tree: Tree): Unit = tree match {
  case TypeApply(Ident(TermName("classOf")), List(targetType)) =>
    println(s"找到了classOf表达式，目标类型是: $targetType")
  case _ =>
    // 处理其他节点
}

2. 严谨全限定匹配 #

如果要覆盖有人写全限定名scala.Predef.classOf[X]的场景，可以同时匹配两种情况：

def matchClassOf(tree: Tree): Unit = tree match {
  case TypeApply(
    Select(Select(Ident(TermName("scala")), TermName("Predef")), TermName("classOf")),
    List(targetType)
  ) =>
    println(s"找到了全限定classOf表达式，目标类型是: $targetType")
  case TypeApply(Ident(TermName("classOf")), List(targetType)) =>
    println(s"找到了classOf表达式，目标类型是: $targetType")
  case _ =>
    // 处理其他节点
}

3. 符号精准匹配（最可靠） #

如果你想彻底避免名字冲突的问题（比如有人自己定义了叫classOf的方法），可以直接匹配方法的符号：

import scala.reflect.internal.Trees.Tree
import scala.reflect.internal.Symbols

def matchClassOf(tree: Tree): Unit = tree match {
  case ta: TypeApply =>
    val classOfSymbol = ta.fun.symbol.rootMirror.staticMethod("scala.Predef", "classOf")
    if (ta.fun.symbol == classOfSymbol) {
      val targetType = ta.args.head
      println(s"找到了classOf表达式，目标类型是: $targetType")
    }
  case _ =>
    // 处理其他节点
}

这种方式直接通过符号的唯一性来判断，是最可靠的。

最后再强调一遍：Apply和TypeApply的核心区别就是，前者传值参数，后者传类型参数——这就是你之前用case _:Apply匹配不到classOf的根本原因。

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