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

时间:2017-01-15 18:16:51

标签: scala scala-macros scala-meta

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]
  }
}

1 个答案:

答案 0 :(得分:1)

传递给宏注释的参数作为元树传递。

尽管可以通过Lit()提取器提取诸如Int / Double / String之类的文字。其他事情并非如此。

在meta

中解析时
  • @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))))

这意味着您没有对此事物的运行时访问权限。如果没有Semantic Api来解​​析符号等,你甚至无法正确地实例化对象。在scalameta 2和paradise 4中它可能可能,但它现在肯定是不可能的。您可以做的是使运行时模式匹配以检查值。

我在这里做了类似的事情(注意这是非常的WIP): 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传入的类型不正确,则运行时异常