Получение методов, вызываемых внутри основного метода с использованием JavaParser - PullRequest
Новая
       25

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

0 голосов
/ 25 июня 2018

Я использую библиотеку JavaParser для анализа кода Java и доступа к токенам кода Java.

Ниже мой код 

import java.util.Vector;
import com.github.javaparser.JavaParser;
import com.github.javaparser.ast.CompilationUnit;
import com.github.javaparser.ast.body.MethodDeclaration;
import com.github.javaparser.ast.expr.VariableDeclarationExpr;
import com.github.javaparser.ast.stmt.BlockStmt;
import com.github.javaparser.ast.type.ClassOrInterfaceType;
import com.github.javaparser.ast.visitor.VoidVisitorAdapter;
import java.io.FileInputStream;


public class MethodParser {
    public static void main(String[] args) throws Exception {
        // creates an input stream for the file to be parsed

        FileInputStream in = new FileInputStream("F:\\Projects\\Parse.java");

        CompilationUnit cu;
        try {
            // parse the file
            cu = JavaParser.parse(in);
        } finally {
            in.close();
        }
        cu.accept(new MethodVisitor(), null);
    }
    private static class MethodVisitor extends VoidVisitorAdapter<Void> {
        @Override

 public void visit(MethodDeclaration n, Void arg) {
            /* here you can access the attributes of the method.
             this method will be called for all methods in this
             CompilationUnit, including inner class methods */

            String x =String.valueOf(n.getBody() );

            n.ifAssertStmt(n);

            System.out.println(x);
            super.visit(n, arg);
        }
        public void visitVariables(MethodDeclaration n, Void arg) {


            String x =String.valueOf(n.getBody());
            System.out.println(x);
            super.visit(n, arg);
        }

    }

}

ниже приведен код java-файла, прочитанного JavaParser (Parse.java) 

public class Parse{

public void printFirstName(){
System.out.println("My First name is John");
}

public void printLastName(){
System.out.println("My Last name is John");
}


public static void main(String[] args){

Parse A = new Parse();
A.printFirstName();

}
}

Выход out выводит содержимое всех тел методов в классе parse.java. Но я хочу получить методы, вызываемые в методе main (например, printFirstName в этом сценарии) Как я могу это сделать?

1 Ответ

0 голосов
/ 27 июня 2018

Я бы использовал посетителя и: 1) установите флаг в visitMethodDeclaration, проверяя, является ли метод, который вы посещаете, основным (имя, статический флаг, тип параметров) 2) в методе visitMethodCallExpression я буду печатать вызовы, только если был установлен флаг в точке 1

...