如何从解析树到Java类文件

Question

我正在使用具有以下功能的命令行工具：

1. 使用扩展的Antlr4 Java9语法修改了.java文件。文件中的语法是Java，对包括目的的方法进行了一个修改，例如在此示例中：public void {Marketing} sendemail（）{} {}
2. 使用访客收集并删除所有目的。目的的收集和分析是程序的主要功能。
3. 编译并执行删除目的的Java文件。

我正在寻找实现步骤3的最简单，最有效的方法。建立完整的编译器的项目范围不超出我的项目范围，我希望在可能的情况下利用Java编译器并运行Javac。我已经考虑了以下方法，但是没有看似最佳的：

• 正如本文提出的那样：将AST汇回源代码。不过，在大型目录上可能会做很多工作。
• 使用ASM生成字节代码，尽管据我了解，我需要有效的Java源代码或类文件才能工作（ https://asm.ow2.io/asm4-guide.pdf ）。
• 构建一个Java编译器插件，以修改AST并在编译中的解析步骤中删除目的（ https://www.baeldung.com/java-build-compiler-plugin ）。我不确定在我可以修改AST之前的汇编是否会失败，因为语法无效。

任何输入都非常感谢。

Answer 1

您可以使用 TokenStreamRewriter 来获取没有目的节点的源代码（或完成许多其他重写任务）。下面是一个来自应用程序的示例，我有条件地将顶级 LIMIT 子句添加到 MySQL 查询：

/**
001     * Parses the query to see if there's already a top-level limit clause. If none was found, the query is
002     * rewritten to include a limit clause with the given values.
003     *
004     * @param query The query to check and modify.
005     * @param serverVersion The version of MySQL to use for checking.
006     * @param sqlMode The current SQL mode in the server.
007     * @param offset The limit offset to add.
008     * @param count The row count value to add.
009     *
010     * @returns The rewritten query if the original query is error free and contained no top-level LIMIT clause.
011     *          Otherwise the original query is returned.
012     */
013    public checkAndApplyLimits(query: string, serverVersion: number, sqlMode: string, offset: number,
014        count: number): [string, boolean] {
015
016        this.applyServerDetails(serverVersion, sqlMode);
017        const tree = this.startParsing(query, false, MySQLParseUnit.Generic);
018        if (!tree || this.errors.length > 0) {
019            return [query, false];
020        }
021
022        const rewriter = new TokenStreamRewriter(this.tokenStream);
023        const expressions = XPath.findAll(tree, "/query/simpleStatement//queryExpression", this.parser);
024        let changed = false;
025        if (expressions.size > 0) {
026            // There can only be one top-level query expression where we can add a LIMIT clause.
027            const candidate: ParseTree = expressions.values().next().value;
028
029            // Check if the candidate comes from a subquery.
030            let run: ParseTree | undefined = candidate;
031            let invalid = false;
032            while (run) {
033                if (run instanceof SubqueryContext) {
034                    invalid = true;
035                    break;
036                }
037
038                run = run.parent;
039            }
040
041            if (!invalid) {
042                // Top level query expression here. Check if there's already a LIMIT clause before adding one.
043                const context = candidate as QueryExpressionContext;
044                if (!context.limitClause() && context.stop) {
045                    // OK, ready to add an own limit clause.
046                    rewriter.insertAfter(context.stop, ` LIMIT ${offset}, ${count}`);
047                    changed = true;
048                }
049            }
040        }
051
052        return [rewriter.getText(), changed];
053    }

这段代码在做什么：

• 第 017 行：解析输入以获取解析树。如果你已经这样做了，当然可以传入解析树，而不用再次解析。
• 第 022 行使用您的令牌流准备一个新的 TokenStreamRewriter 实例。
• 第023行使用ANTLR4的XPATH功能来获取特定上下文类型的所有节点。您可以在这里一次性检索所有目的上下文。这也是你的第2）点的解决方案。
• 以下行仅检查是否必须添加新的 LIMIT 子句。对你来说没那么有趣。
• 第 046 行是操作令牌流的地方。在这种情况下，会添加一些内容，但您也可以 替换或删除节点。
• 第 052 行可能包含您最感兴趣的内容：它返回输入的原始文本，但应用了所有重写操作。

使用此代码，您可以创建一个临时 java 文件进行编译。它可用于同时执行列表中的两个操作（收集目的并删除它们）。