行走ParseTree ANTLR4的正确方法

时间:2018-10-04 20:30:18

标签: c++ antlr4 parse-tree

我正在尝试使用带有C ++目标的ANTLR4实现TSql解析器。我抓取了语法文件here。该jar用于制作相应的源文件(由于冲突,将TSqlParser.cpp中的所有NULL更改为null)。我正在关注doc example(以及网络上的所有其他示例),该示例显示了如何实现一个正常运行的规则和侦听器。实现多个规则时,会发生此问题。

main.cpp:

#include <iostream>
#include <antlr4-runtime/antlr4-runtime.h>
#include "TSqlParser.h"
#include "TSqlLexer.h"

#include "listener.h"


int main(int argc, const char* argv[])
{
    std::ifstream stream;
    stream.open(argv[1]);

    antlr4::ANTLRInputStream input(stream);
    TSqlLexer lexer(&input);
    antlr4::CommonTokenStream tokens(&lexer);
    TSqlParser parser(&tokens);

    TreeShapeListener listener;

    //antlr4::tree::ParseTree *select_tree = parser.select_statement();
    antlr4::tree::ParseTree *update_tree = parser.update_statement();

    //antlr4::tree::ParseTreeWalker::DEFAULT.walk(&listener, select_tree);
    antlr4::tree::ParseTreeWalker::DEFAULT.walk(&listener, update_tree);

    return 0;
}

listener.h:

#include <antlr4-runtime/antlr4-runtime.h>
#include "TSqlParserBaseListener.h"

class TreeShapeListener : public TSqlParserBaseListener
{
public:

    void enterSelect_statement(TSqlParser::Select_statementContext*) override;
    void enterUpdate_statement(TSqlParser::Update_statementContext*) override;
};

listener.cpp:

#include "listener.h"

void TreeShapeListener::enterSelect_statement(TSqlParser::Select_statementContext *ctx)
{
    std::cout << "Hello Select\n";
}

void TreeShapeListener::enterUpdate_statement(TSqlParser::Update_statementContext *ctx)
{
    std::cout << "Hello Update\n";
}

这是一个小的测试sql文件:

SELECT TOP 1 NAME
FROM MYTABLE
WHERE SEQ = 6

UPDATE MYTABLE
SET NAME = 'Bob'
WHERE SEQ = 5

输出:

line 1:0 mismatched input 'SELECT' expecting {'UPDATE', 'WITH'}
Hello Update

如果我切换SQL语句,我只会得到Hello Update,所以我认为它只是吞噬了整个输入。一切似乎都需要我对所有内容进行硬编码才能得到我想要的。我的问题是,在解析文件时如何获得正确的侦听器来调用自己?我觉得我在这里缺少ANTLR非常重要的东西。在这里,我真的不需要完整的答案,我只需要有人指出正确的方向即可。它可能是一个API页面。自编写任何Java以来​​已经有一段时间了,但是它翻译的很好,我可以从那里弄清楚。

1 个答案:

答案 0 :(得分:1)

如果您通过以下方式构造树

antlr4::tree::ParseTree *update_tree = parser.update_statement();

您仅使用语法的一个子集,因为您的输入点是tsql_fileupdate_statement是SQL关键字UPDATE的规则,因此它不知道如何处理SELECT

相反,您应该通过以下方式构造树

antlr4::tree::ParseTree *tsql_file_tree = parser.tsql_file();