Bir yöntem sembolünden ve bir gövdeden bir yöntem tanımlama ağacı oluşturma

Question

MethodSymbol'u Scala 2.10'da bir yöntem tanımlama ağacının (ör. DefDef) sol tarafına döndürmenin kullanışlı bir yolu var mı? Örneğin, bir özelliğin bir örneğini alacak ve bu tüm trait yöntemlerini bazı hata ayıklama işlevleri ile dolduracak bir makro oluşturmak istediğimi varsayalım. Yazabilirim aşağıdaki: Ben özelliği uygulayan yeni bir anonim sınıfta bu yöntemleri yapışmasını ve daha sonra bu sınıf-sen eğer tam bir çalışma örneği here bulabilirsiniz nesnelleştirilme sıkıcı iş elided ettik

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

object WrapperExample { 
    def wrap[A](a: A): A = macro wrap_impl[A] 

    def wrap_impl[A: c.WeakTypeTag](c: Context)(a: c.Expr[A]) = { 
    import c.universe._ 

    val wrapped = weakTypeOf[A] 
    val f = Select(reify(Predef).tree, "println") 

    val methods = wrapped.declarations.collect { 
     case m: MethodSymbol if !m.isConstructor => DefDef(
     Modifiers(Flag.OVERRIDE), 
     m.name, 
     Nil, Nil, 
     TypeTree(), 
     Block(
      Apply(f, c.literal("Calling: " + m.name.decoded).tree :: Nil), 
      Select(a.tree, m.name) 
     ) 
    ) 
    }.toList 

    //... 
} 

' ilgileniyorum.

Şimdi örneğin, bu yazabilirsiniz:

scala> trait X { def foo = 1; def bar = 'a } 
defined trait X 

scala> val x = new X {} 
x: X = $anon$1@15dd533 

scala> val w: X = WrapperExample.wrap[X](x) 
w: X = $1$$1@27c3a4a3 

scala> w.foo 
Calling: foo 
res0: Int = 1 

scala> w.bar 
Calling: bar 
res1: Symbol = 'a 

Yani çalışıyor, ama sadece çok basit durumlarda-o alışkanlık özelliği erişim değiştiricileri, ek açıklamalarla parametre listeleri ile yöntemlerini, varsa,

Gerçekten istediğim bir yöntem sembolü ve yeni gövde için bir ağaç alacak ve bir DefDef döndürecektir. Ben elle tek yazmaya başladım, ama böyle keman bir sürü içerir: can sıkıcı ayrıntılı ve hataya açık olan

List(if (method.isImplicit) Some(Flag.IMPLICIT) else None, ...) 

. Yeni Yansıma API'sinde bunu yapmanın daha güzel bir yolunu mu duyuyorum?

Answer 1

Bildiğim kadarıyla, bir sembolden tanımlayıcı bir ağaca gitmenin standart bir yolu yoktur.

En iyi bahsiniz, muhtemelen, c.enclosingRun.units aracılığıyla yinelemeniz ve unit.body ağaçlarının her birine yeniden girmenizdir. Sembolünüze eşit bir symbol olan bir DefDef görüyorsanız, hedefinize ulaştınız demektir. Post. Yeniden tanımlamadan önce tanımlayan ağaç duplicate unutma!

Bu teknik, dünyadaki en uygun şey olmaktan çok uzaktır, ancak çalışmalıdır.

Answer 2

Aşağıdakileri deneyebilirsiniz. Bu, birden çok parametre ve kıvrımlı işlevler ve tür parametreleriyle çalışır;)

val methods = wrapped.declarations.collect { 
    case m: MethodSymbol if !m.isConstructor => DefDef(
    Modifiers(Flag.OVERRIDE), 
    m.name, 
    m.typeParams.map(TypeDef(_)), 
    m.paramss.map(_.map(ValDef(_))), 
    TypeTree(m.returnType), 
    Block(
     Apply(f, c.literal("Calling: " + m.name.decoded).tree :: Nil), 
     m.paramss.foldLeft(Select(a.tree.duplicate, m.name): Tree)((prev, params) => 
     Apply(prev, params.map(p => Ident(p.name))) 
    ) 
    ) 
) 
}.toList