scala-macros - Scala 宏 : get param default value

Question

我有下一个代码，我想从 value 中提取默认参数。

//
def extractor[T] = macro extractorImpl[T]

def extractorImpl[T: c.WeakTypeTag](c: Context) = {
  //first i got a type contructor
  ???
}

我尝试使用 attachments 但 attachments.all 返回一个 Set[Any] 与(例如)SymbolSourceAttachment(val name : String = "新名字")

SymbolSourceAttachment 包含 ValDef 但我不知道如何从 SymbolSourceAttachment ValDef 中提取。

顺便说一句，我应该得到一个 Map[String, String]("name"-> "new name")

例子:

case class Person(name: String = "new name")

object Macro {
  def extractor[T] = macro extractorImpl[T]

  def extractorImpl[T: c.WeakTypeTag](c: Context) = {
    import c.universe._

    c.weakTypeOf[T].declarations.collect {
      case a: MethodSymbol if a.isConstructor =>
        a.paramss.collect {
          case b => b.collect {
            case c =>
              c.attachments.all {
                case d => println(showRaw(d)) // => SymbolSourceAttachment(val name: String = "new name")  
              }
          }
        }
      }
   } 
}

并且宏应该返回 Map("name"-> "new name")

Answer 1

由于您看到 SymbolSourceAttachment，我假设您使用的是宏天堂(因为它是仅在天堂中使用的内部附件)，所以我可以随意使用 quasiquotes :)

在 Scala 反射 API 中获取默认参数的值并不容易。您最好的方法是对为计算默认值而创建的方法的名称进行逆向工程，然后引用这些方法。

SymbolSourceAttachment 如果您的宏在编译案例类的同一编译运行中扩展，那么它会起作用，但它会在单独的编译下中断(附件不保存在类文件中)，并且它在原版 Scala 中不起作用(因为这个附件是天堂独有的)。

=== Macros.scala ===

import scala.reflect.macros.Context
import scala.language.experimental.macros

object Macros {
  def impl[T](c: Context)(T: c.WeakTypeTag[T]): c.Expr[Map[String, Any]] = {
    import c.universe._
    val classSym = T.tpe.typeSymbol
    val moduleSym = classSym.companionSymbol
    val apply = moduleSym.typeSignature.declaration(newTermName("apply")).asMethod
    // can handle only default parameters from the first parameter list
    // because subsequent parameter lists might depend on previous parameters
    val kvps = apply.paramss.head.map(_.asTerm).zipWithIndex.flatMap{ case (p, i) =>
      if (!p.isParamWithDefault) None
      else {
        val getterName = newTermName("apply$default$" + (i + 1))
        Some(q"${p.name.toString} -> $moduleSym.$getterName")
      }
    }
    c.Expr[Map[String, Any]](q"Map[String, Any](..$kvps)")
  }

  def extractor[T]: Map[String, Any] = macro impl[T]
}

=== Test.scala ===

case class C(x: Int = 2, y: String, z: Boolean = true)(t: String = "hello")

object Test extends App {
  println(Macros.extractor[C])
}

17:10 ~/Projects/Paradise2103/sandbox/src/main/scala (2.10.3)$ scalac Macros.scala && scalac Test.scala && scala Test
Map(x -> 2, z -> true)