如何用Scala 3的Quotes.reflect API创建多态PolyFunction

要实现目标中的匿名PolyFunction类，核心是给apply方法添加类型参数X，并在参数和返回类型中引用它，同时处理实验性API的使用问题。以下是完整的修正实现：

import scala.quoted.*
import scala.quoted.experimental.*

inline def createIdentityPolyFunction: PolyFunction = ${ createIdentityPolyFunctionImpl }

def createIdentityPolyFunctionImpl(using Quotes): Expr[PolyFunction] = {
  import quotes.reflect.*

  val name: String = "$anon"
  val parents = List(TypeTree.of[Object], TypeTree.of[PolyFunction])

  def decls(cls: Symbol): List[Symbol] = {
    // 创建类型参数X的符号
    val xTypeParam = Symbol.newTypeParam(cls, "X", TypeBounds.empty)
    // 构建带类型参数的apply方法类型：[X](a: X): X
    val applyMethodType = PolyType(List(xTypeParam)) { _ =>
      MethodType(List("a")) { mt =>
        val xType = TypeRef(mt.owner, xTypeParam)
        (List(xType), xType) // 参数类型列表与返回类型均为X
      }
    }
    // 创建apply方法符号
    List(Symbol.newMethod(cls, "apply", applyMethodType))
  }

  val cls = Symbol.newClass(
    Symbol.spliceOwner,
    name,
    parents = parents.map(_.tpe),
    decls,
    selfType = None
  )
  val applySym = cls.declaredMethod("apply").head

  // 实现apply方法体：直接返回输入参数
  val applyDef = DefDef(applySym, argss => Some {
    Ident(argss.head.head.symbol)
  })

  val clsDef = ClassDef(cls, parents, body = List(applyDef))
  val closure = Block(
    List(clsDef),
    Apply(Select(New(TypeIdent(cls)), cls.primaryConstructor), Nil)
  )

  closure.asExprOf[PolyFunction]
}

关键修改说明 #

• 声明类型参数X：通过Symbol.newTypeParam创建类型参数符号，再用PolyType将其绑定到apply方法上，实现方法的多态类型参数声明。
• 引用类型参数：在MethodType定义中，使用TypeRef获取X的类型，将其设为参数a的类型和方法返回类型，确保类型一致性。
• 实现方法逻辑：DefDef中直接提取参数并返回，对应原代码的a => a逻辑。
• 实验性API处理：需要导入scala.quoted.experimental.*，编译时添加-Yexperimental选项，因为PolyType等反射API目前处于实验阶段。

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