Есть ли способ получить все операторы в методе без ручного обхода AST? - PullRequest
Новая
       66

Есть ли способ получить все операторы в методе без ручного обхода AST?

0 голосов
/ 18 октября 2019

просто интересно, есть ли способ получить все операторы в методе без ручного обхода AST метода.

До сих пор я пытался получить все операторы путем ручного обхода через метод. Дерево, которое, как правило, является плохой идеей и сложно сделать. Уровни AST Если кто-то может помочь, это будет высоко ценится.

1 Ответ

1 голос
/ 21 октября 2019

Вам нужно использовать другого посетителя, если вы хотите просмотреть все утверждения всех уровней глубины AST.

На основе вашего примера: 

/**
 * Visitor which counts expression statements
 */
GenericVisitorAdapter<ExpressionStmt, Integer> expressionsCountVisitor = new GenericVisitorAdapter<ExpressionStmt, Integer>() {

        /**
         * This is based on your example
         */
        @Override
        public ExpressionStmt visit(ExpressionStmt expressionStmt, Integer count) {
            if(expressionStmt.toString().contains("=")) {
                count++;
            }
            return super.visit(expressionStmt, count);
        }

        /**
         * But better to check for AssignExpr instead of ".contains("=")"
         */
        @Override
        public ExpressionStmt visit(AssignExpr assignExpr, Integer count) {
            // counts each "=" assign expression
            count++;
            return super.visit(assignExpr, count);
        }

    };

// And then use this visitor for each MethodDecalration

public void visit(MethodDeclaration methodDeclaration, Object arg) {
    Integer total = 0;
    // traverses down the hierarchy of this method declaration
    methodDeclaration.accept(expressionsCountVisitor, total);
    super.visit(methodDeclaration, arg);
    System.out.println(total);
}
...