एक विधि प्रतीक और शरीर से एक विधि परिभाषा पेड़ बनाना

Question

क्या स्केल 2.10 में एक विधि परिभाषा पेड़ (यानी, DefDef) के बाईं ओर MethodSymbol को चालू करने का कोई सुविधाजनक तरीका है?

उदाहरण के लिए, मान लीजिए कि मैं एक मैक्रो बनाना चाहता हूं जो एक विशेषता का एक उदाहरण लेगा और कुछ डीबगिंग कार्यक्षमता के साथ उस विशेषता के सभी तरीकों को लपेट लेगा। निम्नलिखित मैं लिख सकते हैं:

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 

    //... 
} 

मैं एक नई अज्ञात वर्ग विशेषता को लागू करता है में इन तरीकों चिपके हुए और फिर उस वर्ग आप अगर आप एक पूरा काम कर उदाहरण here पा सकते हैं instantiating की बोरिंग व्यापार elided है ' रुचि रखते हैं

अब मैं यह लिख सकते हैं, उदाहरण के लिए:

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 

तो यह काम करता है, लेकिन केवल बहुत सरल में मामलों, यह नहीं होगा यदि विशेषता पैरामीटर सूची के साथ तरीकों, पहुँच संशोधक, एनोटेशन के साथ है, आदि

जो मैं वास्तव में चाहता हूं वह एक ऐसा कार्य है जो एक विधि प्रतीक और नए शरीर के लिए एक पेड़ लेगा और DefDef लौटाएगा। मैं हाथ से एक लिखना शुरू किया है, लेकिन यह इस तरह बारीकियों सामान का एक बहुत शामिल है:

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

कौन सा, कष्टप्रद वर्बोज़, और त्रुटि प्रवण है। क्या मुझे नए प्रतिबिंब एपीआई में ऐसा करने के लिए कुछ अच्छा तरीका याद आ रहा है?

Answer 1

मेरे सबसे अच्छे ज्ञान के लिए, प्रतीक से एक परिभाषित पेड़ पर जाने का कोई मानक तरीका नहीं है।

आपकी सर्वश्रेष्ठ शर्त शायद c.enclosingRun.units के माध्यम से फिर से शुरू हो जाएगी, unit.body पेड़ों में से प्रत्येक में आप जितनी बार जाते हैं। यदि आपको DefDef दिखाई देता है, जिसमें आपके प्रतीक के बराबर symbol है, तो आप अपने गंतव्य तक पहुंच गए हैं। upd। इसे पुन: उपयोग करने से पहले duplicate परिभाषित पेड़ को मत भूलना!

यह तकनीक दुनिया में सबसे सुविधाजनक चीज़ होने से बहुत दूर है, लेकिन इसे काम करना चाहिए।

Answer 2

आप निम्न का प्रयास कर सकते हैं। यह एकाधिक पैरामीटर और करीबी कार्यों और टाइप पैरामीटर के साथ काम करता है;)

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