我有下一个代码,我想从值中提取默认参数。
//
def extractor[T] = macro extractorImpl[T]
def extractorImpl[T: c.WeakTypeTag](c: Context) = {
//first i got a type contructor
???
}
我尝试与attachments
but attachments.all
返回一个Set[Any]
与(例如)SymbolSourceAttachment(val name: String = "new name")
SymbolSourceAttachment
包含ValDef
但我不知道如何从中提取SymbolSourceAttachment
ValDef
.
顺便说一句,我应该得到一个Map[String, String]("name" -> "new name")
Example:
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")
既然你看到了SymbolSourceAttachment
,我假设您正在使用宏天堂(因为它是仅在天堂中使用的内部附件),所以我可以随意使用准引号:)
没有简单的方法可以获取 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)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)