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

问题描述

我正在使用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 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);

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

我该如何获取实际的文件文本?一种方法是将文件内容传递给UnderlineListener的构造函数,但是我宁愿坚持使用上述获取实际文件文本的方法,因为它可以用于CaseChangingCharStream不是使用。

解决方法

我找到了一种解决方法。 CaseChangingCharStream.java的当前实现没有像getCharStream()这样的getter方法来访问final CharStream stream;属性。只需为其添加一个getter方法,便可以如下访问基础CharStream对象:

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