将类型参数传递给scala元宏/注释


package scalaworld.macros
import scala.meta._
class Argument(arg: Int) extends scala.annotation.StaticAnnotation {
inline def apply(defn: Any): Any = meta {
println(this.structure)
val arg = this match {
// The argument needs to be a literal like `1` or a string like `"foobar"`.
// You can't pass in a variable name.
case q"new $_(${Lit(arg: Int)})"                      => arg
// Example if you have more than one argument.
case q"new $_(${Lit(arg: Int)}, ${Lit(foo: String)})" => arg
case _                                                => ??? // default     value
}
println(s"Arg is $arg")
defn.asInstanceOf[Stat]
}
}

我想修改上面的宏并添加类型参数[A]。我尝试了以下操作,但它没有编译

package scalaworld.macros
import scala.meta._
class Argument2[A](arg: A) extends scala.annotation.StaticAnnotation {
inline def apply(defn: Any): Any = meta {
println(this.structure)
val arg = this match {
case q"new $_(${Lit(arg: A)})"                      => arg
case q"new $_(${Lit(arg: A)}, ${Lit(foo: String)})" => arg
case _                                              => ???
}
println(s"Arg is $arg")
defn.asInstanceOf[Stat]
}
}

传递到宏注释中的参数作为子树传递。

尽管可以通过Lit()提取器提取Int/DDouble/String等文本。其他事情并非如此。

在元中解析时

  • @someMacro(1)变为@someMacro(Lit(1))
  • @someMacro("Foo")变为@someMacro(Lit("Foo"))

其他一切都会像正常情况下一样通过

  • @someMacro(foo)变为@someMacro(Term.Name("foo"))
  • @someMacro(Option(2))变为@someMacro(Term.Apply(Term.Name("Option"), Seq(Lit(2))))

这意味着您没有运行时访问权限。如果没有语义Api来解析符号等,你甚至无法正确实例化对象。它在scalameta2和天堂4中可能是可能的,但现在绝对不可能。您可以做的是进行运行时模式匹配来检查值。

我在这里做了一些类似的事情(注意这是非常在制品):https://github.com/DavidDudson/Elysium/blob/master/gen/src/main/scala/nz/daved/elysium/gen/MacroAnnotation.scala

具体参见https://github.com/DavidDudson/Elysium/blob/master/gen/src/main/scala/nz/daved/elysium/gen/MacroAnnotation.scala#L149

注意:这意味着在运行时(在该示例中恰好是编译时),如果传入的arg类型不正确,则运行时异常

最新更新