JavaParser：解析和生成Java代码

Question

我已经阅读了JavaParser手册，并开始构建自己的示例。我想要实现的是阅读Java代码并在其上插入新的代码行。具体来说，我想在每个if和while语句之前初始化一个计数器，并在语句体内部对计数器进行递增。我这样做的目的是为一组指定的运行运行运行新代码，并观察每个分支执行了多少次。我使用JavaParser来解析和添加代码，因为我想自动生成和运行所有东西。

例如，我们有以下简单的代码：

if (x>4) { 
   x=4; 
} 
else { 
   x=4; 
} 
while (i<x) {
   i++;
} 

在解析之后，我想有如下内容：

int countA=0; 
int countB=0;            
if (x>4) { 
   x=4; 
   countA++;
} 
else { 
   x=4; 
   countB++;
} 
int countC=0;
while (i<x) {
   i++;
   countC++;
} 

到目前为止，我的代码是：

public static void main (String[] args) throws Exception {          
              CompilationUnit cu = StaticJavaParser.parse("class X{void x(){" +
              "    if (x>4) {  " +
              "      x=4;  " +
              "    }  " +
              "    else {  " +
              "        x=4;  " +
              "    }  " +
              "    while (i<x) {  " +
              "        i++;  " +
              "    }  " +
              "    }}");
                              
              cu.findAll(Statement.class).forEach(statement -> {             
                  if (statement.isIfStmt()) {
                      //System.out.println(statement.getChildNodes().get(0));
                     // System.out.println(statement.getParentNodeForChildren().toString());
                     // statement.getChildNodes().add(statement.getChildNodes().get(0));
                
                     // System.out.println(statement.toString());
                  }
                  if (statement.isWhileStmt()) {
                     // System.out.println();
                  }
              });
              System.out.println(cu);
            }
}

我尝试过一些东西（推荐的外线），但没有成功。我的主要问题是，我找不到在给定的语句的childNodes末尾注入任何代码行的方法，这应该是计数器的增量，或者在parentnodes末尾注入任何代码行，这将是初始化。有什么方法可以实现我所描述的吗？




Answer 1

代码中的问题如下：

if (statement.isIfStmt()) {

该条件清楚地表明您正在处理if语句，因此实际上无法添加指令。您需要做的是获取n1元素（其中n是if语句的索引），然后使用：

cu.findAll(Statement.class).get(2).asBlockStmt().addStatement("countA++;")

有几种方法可以定义要添加的表达式，为了清晰起见，我在这里使用了一个普通字符串，但请检查工作示例中的实际语法。

暂时你可以使用同样的技术，但它会更复杂。为简单起见，将使用以下方法在方法开头添加计数器声明：

((BlockStmt)cu.findAll(Statement.class).get(6).getParentNode().get()).addStatement(0, new StringLiteralExpr("int b = 1;"));

剥除

• 所有语句列表中的第六个语句是while循环
• 此语句的父节点是该方法的主块，是要添加指令的位置
• 需要将此对象类强制转换为BlockStmt才能使用addStatement方法
• StringLiteralExpr是Expression的一个子类，在这里用于从字符串创建实例。您还可以探索许多其他不同的实现

我复制了你的代码，制作了一个工作的JUnit 5，这就是我得到的结果

公共类ParserTest{

String counterName = "count";
Integer counterIndex = 1;

@Test
public void main() {

    CompilationUnit cu = StaticJavaParser.parse("class X{void x(){" + "    if (x>4) {  " + "      x=4;  "
            + "    }  " + "    else {  " + "        x=4;  " + "    }  " + "    while (i<x) {  " + "        i++;  "
            + "    }  " + "    }}");

    cu.findAll(Statement.class).forEach(statement -> {
        if (statement.isIfStmt()) {

            // Add counter declaration in main block for the if
            String counterNameAndIndexIf = counterName + counterIndex++;
            // Create expression using StaticJavaParser
            Expression ifCounterExpression = StaticJavaParser
                    .parseVariableDeclarationExpr(" int " + counterNameAndIndexIf + " = 0");
            ((BlockStmt) statement.getParentNode().get()).addStatement(0, ifCounterExpression);

            // Add counter declaration in main block for the else
            String counterNameAndIndexElse = counterName + counterIndex++;
            // Create expression using StaticJavaParser
            Expression elseCounterExpression = StaticJavaParser
                    .parseVariableDeclarationExpr("int " + counterNameAndIndexElse + " = 0");
            ((BlockStmt) statement.getParentNode().get()).addStatement(0, elseCounterExpression);

            // Add if increment
            Expression ifIncrementExpression = StaticJavaParser.parseExpression(counterNameAndIndexIf + "++");
            ((BlockStmt) statement.getChildNodes().get(1)).addStatement(ifIncrementExpression);

            // Add else increment
            Expression elseIncrementExpression = StaticJavaParser.parseExpression(counterNameAndIndexElse + "++");
            ((BlockStmt) statement.getChildNodes().get(2)).addStatement(elseIncrementExpression);
        }
        if (statement.isWhileStmt()) {
            String counterNameAndIndexWhile = counterName + counterIndex++;
            Expression whileCounterExpression = StaticJavaParser
                    .parseVariableDeclarationExpr(" int " + counterNameAndIndexWhile + " = 0");
            ((BlockStmt) statement.getParentNode().get()).addStatement(0, whileCounterExpression);

            // Add while increment
            Expression whileIncrementExpression = StaticJavaParser.parseExpression(counterNameAndIndexWhile + "++");
            ((BlockStmt) statement.getChildNodes().get(1)).addStatement(whileIncrementExpression);

        }
    });
    System.out.println(cu);
}

而结果就是

class X {

void x() {
    int count3 = 0;
    int count2 = 0;
    int count1 = 0;
    if (x > 4) {
        x = 4;
        count1++;
    } else {
        x = 4;
        count2++;
    }
    while (i < x) {
        i++;
        count3++;
    }
}

希望这有帮助。如果您需要关于如何构建这些对象的说明，请告诉我。当然，有更聪明的方法和代码可以被重构，但我想让它尽可能清晰地为您描述。

干杯

编辑：为了支持语句嵌套和声明方法主块中的所有计数器变量，可以使用以下方法

statement.findRootNode().getChildNodes().get(0).getChildNodes().get(1).getChildNodes().get(2)

这种方法将从根开始自上而下，始终指向主块。使用它添加计数器变量。