时髦 AST 转换 - 替换方法

Question

我正在尝试使用AST转换替换类的方法。

我首先检查该方法是否存在，然后将其删除（基于此）。

MethodNode methodNode = parentClass.getMethod(methodName, Parameter.EMPTY_ARRAY)
if (methodNode != null) {
     parentClass.getMethods().remove(methodNode)
} 

我看到集合上的大小发生了变化，但该方法仍然可以在类节点上访问。

删除该方法后，我想添加一个同名的新方法：

parentClass.addMethod(newMethodNode)

但是，这会导致以下错误：

 Repetitive method name/signature for method 'grails.plugin.Bar getBar()' in class 'grails.plugin.Foo'.

正确的方法是什么？

更新：由于我打算做的就是替换该方法的现有功能，因此我创建了一个新的块语句，并使用 methodNode.setCode（语句）在现有方法上设置了它。

Answer 1

听起来你想做的是用其他东西包装一个方法？下面是一个在try/finally中包装方法的例子。

import org.codehaus.groovy.ast.*
import org.codehaus.groovy.ast.expr.*
import org.codehaus.groovy.ast.stmt.*

methodNode.setCode(methodThatCreatesAst(methodNode.getCode()))

def methodThatCreatesAst(Statement statementToWrap) {
  BlockStatement methodBody = new BlockStatement()

  methodBody.addStatement(new TryCatchStatement(
    // try
    statementToWrap,
    // finally
    new ExpressionStatement(
      new MethodCallExpression(
        "myVar", "myMethod", new ArgumentListExpression()
      )
    )
  ))

  return methodBody
}

除此之外，我帮不上忙，因为我以前从未尝试过移除一个方法:)