ANTLR解析Java文件

ANTLR解析Java文件

https://github.com/antlr/grammars-v4.git
grammars-v4里提供了Java和其他常用的antlr语法.

最近看的一个项目里实体类会使用 @Model(name=“xxx”) 标注实体类.用@ModelField(name=“xxx”)标注关联的实体.
这里使用antlr解析实体类之间的关联.

FileFilter filter = new FileFilter() {
    @Override
    public boolean accept(File file) {
        return file.getName().endsWith(".java");
    }
};

List<File> files = FileUtil.loopFiles(basePath, filter);

List<ObjCon> cons = Lists.newArrayList();
System.out.println("文件数: " + files.size());
int i = 0;
for (File file : files) {
    System.out.println("====== 第 " + i++ + "个文件");
    if (!file.getName().endsWith(".java")) {
        continue;
    }
    InputStream is = new FileInputStream(file);

    ANTLRInputStream input = new ANTLRInputStream(is);
    Java8Lexer lexer = new Java8Lexer(input);
    CommonTokenStream tokens = new CommonTokenStream(lexer);
    Java8Parser parser = new Java8Parser(tokens);
    // 开始分析语法
    ParseTree tree = parser.compilationUnit(); // parse

    // listener
    ObjCon curr = new ObjCon();
    cons.add(curr);
    MyJava8EntityUseListener testListener = new MyJava8EntityUseListener(parser, curr);

    // 新建一个 标准的 遍历器
    ParseTreeWalker walker = new ParseTreeWalker(); // create standard walker
    // 使用监听器 初始化 对 语法分析树的遍历
    walker.walk(testListener, tree);
}

listener类重写感兴趣的方法,这里使用
进入类声明方法,获取类上注解和类名.

@Override
public void enterNormalClassDeclaration(Java8Parser.NormalClassDeclarationContext ctx) {

    ctx.classModifier().forEach(modifier -> {
        if (modifier.annotation() != null
                && modifier.annotation().normalAnnotation() != null
                && modifier.annotation().normalAnnotation().typeName() != null
                && modifier.annotation().normalAnnotation().typeName().Identifier().getText() != null) {
            Java8Parser.NormalAnnotationContext normalAnnotation = modifier.annotation().normalAnnotation();

            // 注解名
            String annoName = normalAnnotation.typeName().Identifier().getText();
            if (annoName.equals("Model")) {
                // 注解属性 key-value
                normalAnnotation.elementValuePairList().elementValuePair().forEach(elem -> {
                    // 注解key标识符
                    String annoKey = elem.Identifier().getText();
                    if ("name".equals(annoKey)) {
                        // 注解value
                        objCon.setCnName(elem.elementValue().getText().replace("\"", ""));
                    }
                });
            }
        }
    });


    TerminalNode className = ctx.Identifier();
    System.out.println("类名: " + className);
    objCon.setName(className.getText());
    objCon.setEnCnName(objCon.getName() + objCon.getCnName());
}

进入field声明方法,将该类关联的带有@Relation注解的成员找出添加到list里.

/**
    * 进入field声明
    *
    * @param ctx
    */
@Override
public void enterFieldDeclaration(Java8Parser.FieldDeclarationContext ctx) {
    ctx.fieldModifier().forEach(modifier -> {
        if (modifier.annotation() != null
                && modifier.annotation().normalAnnotation() != null
                && modifier.annotation().normalAnnotation().typeName() != null
                && modifier.annotation().normalAnnotation().typeName().Identifier().getText() != null) {
            Java8Parser.NormalAnnotationContext anno = modifier.annotation().normalAnnotation();

            String txt = anno.typeName().Identifier().getText();
            if (txt.equals("ModelField")) {

            } else if (txt.equals("Relation")) {
                ObjMeta currObjMeta = new ObjMeta();
                globeCurrObjMeta = currObjMeta;
                objCon.getMetas().add(currObjMeta);

                anno.elementValuePairList().elementValuePair().forEach(elem -> {
                    elem.elementValue();
                    if (elem.Identifier().equals("modelClass")) {
                        //
                    }
                });
            }
        }
    });

    if (globeCurrObjMeta != null) {
        String classType = ctx.unannType().start.getText();
        globeCurrObjMeta.setClassName(classType);
        globeCurrObjMeta.setClassType(classType);
    }

    ctx.fieldModifier();
}

最后输出到obsidian里.