在Groovy中疲惫地走过AST树

Question

这与我的问题有关，即拦截给定类中对字段的所有访问，而不仅仅是以与Groovy“属性”样式访问一致的方式进行的访问。您可以在这里查看：在groovy中拦截LOCAL属性访问。

我发现的一种肯定会解决我的问题的方法是在编译时使用AST重写任何具有属性访问的非属性访问。例如，如果一个类如下所示：

class Foo {
  def x = 1
  def getter() {
    x
  }
  def getProperty(String name) {
    this."$name" ++
  }
}

foo = new Foo()
assert foo.getter() == 1
assert foo.x == 2

这些断言语句将会工作，因为getter方法直接访问x，而foo.x通过getProperty("x ")在返回前递增x。

经过一些反复试验，我可以使用AST转换来更改代码的行为，这样“getter”方法中的表达式“x”实际上是作为属性而不是本地字段访问的。到目前为止一切都很好！

现在，我该如何访问给定类中本地字段的所有访问权限？我一直在梳理互联网，寻找某种AST树行者助手，但没有找到一个。我真的需要为此处的所有 38 种表达式类型实现表达式 walker http://groovy.codehaus.org/api/org/codehaus/groovy/ast/expr/package-summary.html 此处的所有 18 种语句类型 http://groovy.codehaus.org/api/org/codehaus/groovy/ast/stmt/package-summary.html？这似乎是某人已经写过的东西（因为它首先是构建AST树不可或缺的一部分），但我似乎找不到它。

Answer 1

格伦

您正在寻找某种类型的访问者。Groovy定义了一些（记录不充分的）访问者，您可以使用这些访问者。对于您的问题，我没有确切的答案，但我可以为您提供一些方向。

下面的代码片段显示了如何遍历一个类的AST并打印所有方法名:

class TypeSystemUsageVisitor extends ClassCodeVisitorSupport {

@Override
public void visitExpression(MethodNode node) {
    super.visitMethod(node)
    println node.name
}

@Override
protected SourceUnit getSourceUnit() {
    // I don't know ho I should implement this, but it makes no difference
    return null;
} 
}

这就是我如何使用上面定义的访问者

def visitor = new TypeSystemUsageVisitor()
def sourceFile = new File("path/to/Class.groovy")
def ast = new AstBuilder().buildFromString(CompilePhase.CONVERSION, false, sourceFile.text).find { it.class == ClassNode.class }
ast.visitContents(visitor)

参观者会替你横穿这棵树。它们有访问*方法，您可以覆盖这些方法并对其执行任何操作。我认为适合您的问题的访问者是CodeVisitorSupport，它有一个visitVariableExpression方法。

我建议您阅读groovyConsole附带的AST浏览器代码，以获得更多关于如何使用Groovy AST访问者的示例。另外，看看CodeVisitorSupport的api文档。