Question

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

//
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 = "new name")

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中获取默认参数值没有简单的方法。您最好的镜头是反向设计为计算默认值而创建的方法的名称，然后参考这些名称。

如果你的宏在编译case类的同一个编译运行中扩展，那么

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)

Scala Macro：获取param默认值

1 个答案: