如何在使用Antlr4解析PlSQL时提取有语法错误的行

Question

我正在使用这个Github repository中的PlSql语法文件。如果plsql文件有语法错误，我想在它的下面加下划线。我提供了以下代码片段来完成此操作：

public static class UnderlineListener extends BaseErrorListener {

    public void syntaxError(Recognizer<?, ?> recognizer,
                            Object offendingSymbol,
                            int line, int charPositionInLine,
                            String msg,
                            RecognitionException e)
    {
        System.err.println("line "+line+":"+charPositionInLine+" "+msg);
        underlineError(recognizer,(Token)offendingSymbol,
                       line, charPositionInLine);
    }

    protected void underlineError(Recognizer recognizer,
                                  Token offendingToken, int line,
                                  int charPositionInLine) {
        CommonTokenStream tokens =
            (CommonTokenStream)recognizer.getInputStream();
        String input = tokens.getTokenSource().getInputStream().toString();
        String[] lines = input.split("\n");
        String errorLine = lines[line - 1];
        System.err.println(errorLine);
        for (int i=0; i<charPositionInLine; i++) System.err.print(" ");
        int start = offendingToken.getStartIndex();
        int stop = offendingToken.getStopIndex();
        if ( start>=0 && stop>=0 ) {
        for (int i=start; i<=stop; i++) System.err.print("^");
        }
    System.err.println();
    }
}

虽然这在大多数情况下都很好用，但一些脚本语言，如PlSql，需要special handling for case-sensitivity。这意味着我必须使用CaseChangingCharStream，如下所示：

CharStream s = CharStreams.fromPath(Paths.get('test.sql'));
CaseChangingCharStream upper = new CaseChangingCharStream(s, true);
Lexer lexer = new SomeSQLLexer(upper);

现在，当我尝试使用String input = tokens.getTokenSource().getInputStream().toString();在UnderlineListener中获取输入文本时，无法获得test.sql的实际文本。这是因为getInputStream()返回的CaseChangingCharStream对象没有给出我的test.sql所需的实际文本。

在我的案例中，如何获得实际的文件文本？一种方法是将文件内容传递给UnderlineListener的构造函数，但我更喜欢使用上面的方法来获取实际的文件文本，因为它可以用于不使用CaseChangingCharStream的情况。

Answer 1

我已经找到了一种解决方法。CaseChangingCharStream.java的当前实现没有像getCharStream()这样的getter方法来访问final CharStream stream;属性。只需为它添加一个getter方法，就可以访问底层的CharStream对象，如下所示：

CaseChangingCharStream modifiedCharStream = (CaseChangingCharStream) tokens.getTokenSource().getInputStream();
String input = modifiedCharStream.getCharStream().toString();