ClickHouse/utils/syntax-analyzer/main.cpp
fastio fc9c440d3e
Fix the unexpected behaviour of show tables when antlr parser enabled (#18431)
* Fix the unexpected behaviour of show tables when antlr parser enabled

* fix compile error
2021-01-04 23:42:57 +03:00

64 lines
1.9 KiB
C++

#include <IO/WriteBufferFromOStream.h>
#include <Parsers/IAST.h>
#include <Parsers/New/AST/Query.h>
#include <Parsers/New/parseQuery.h>
#include <Parsers/parseQuery.h>
#include <Parsers/ParserQuery.h>
#include <support/Any.h>
#include <iostream>
#include <iterator>
#include <string>
using namespace DB;
int main(int argc, const char **)
{
if (argc > 1)
{
std::cerr << "No arguments needed. Reads query from input until EOF" << std::endl;
return 1;
}
std::istreambuf_iterator<char> begin(std::cin), end;
std::string query(begin, end);
{
std::vector<String> queries;
splitMultipartQuery(query, queries, 10000000, 10000);
for (const auto & q : queries)
{
std::cout << std::endl << "Query:" << std::endl;
std::cout << q << std::endl;
ParserQuery parser(q.data() + q.size());
ASTPtr orig_ast = parseQuery(parser, q, 10000000, 10000);
std::cout << std::endl << "New AST:" << std::endl;
auto new_ast = parseQuery(q, "");
new_ast->dump();
auto old_ast = new_ast->convertToOld();
if (orig_ast)
{
std::cout << std::endl << "Original AST:" << std::endl;
WriteBufferFromOStream buf(std::cout, 1);
orig_ast->dumpTree(buf);
std::cout << std::endl << "Original query:" << std::endl;
orig_ast->format({buf, false});
std::cout << std::endl;
}
if (old_ast)
{
std::cout << std::endl << "Converted AST:" << std::endl;
WriteBufferFromOStream buf(std::cout, 1);
old_ast->dumpTree(buf);
std::cout << std::endl << "Converted query:" << std::endl;
old_ast->format({buf, false});
std::cout << std::endl;
}
}
}
}