Scala 3宏中如何匹配Lambda函数字面量表达式

问题重现 #

你定义的宏代码如下，调用r((arg: Int) => arg == 0)时始终进入第二个匹配分支，无法捕获Lambda的函数体：

def rCode(formal: Expr[Int => Boolean])(using Quotes): Expr[Int => Boolean] =
  formal match {
    case '{ (arg: Int) => ${ body }: Boolean } => println(body)
    case f => println(f.show)
  }
  formal

inline def r(inline formal: Int => Boolean): Int => Boolean = ${rCode('formal)}

问题原因 #

1. 参数名严格匹配：宏模式中的arg是固定名称，但编译器在处理Lambda时可能会对参数进行重命名（生成唯一标识符避免冲突），导致模式无法匹配。
2. 返回值类型标注冗余：你在模式中指定了: Boolean，但实际生成的AST里可能不会保留这个显式的返回值类型标注，进一步导致匹配失败。

解决方案 #

使用变量匹配任意参数名，同时去掉不必要的返回值类型约束，修改后的宏代码如下：

import scala.quoted.*

def rCode(formal: Expr[Int => Boolean])(using Quotes): Expr[Int => Boolean] =
  formal match {
    case '{ ($param: Int) => ${ body } } => 
      println(s"匹配成功，函数体：${body.show}")
    case f => 
      println(f.show)
  }
  formal

inline def r(inline formal: Int => Boolean): Int => Boolean = ${rCode('formal)}

这里的$param是一个变量，可以匹配任意名称的参数，同时去掉了: Boolean的约束，让模式更贴近编译器实际生成的AST结构。

结论 #

这是宏模式匹配的细节问题，并非Scala平台的遗留问题。核心是要注意宏模式匹配的严格性，避免写死参数名或冗余的类型标注，改用变量匹配来适配编译器生成的AST结构。

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