1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
#include "postgres_parser.hpp"
#include "pg_functions.hpp"
#include "parser/parser.hpp"
#include "parser/scansup.hpp"
#include "common/keywords.hpp"
namespace duckdb {
PostgresParser::PostgresParser() : success(false), parse_tree(nullptr), error_message(""), error_location(0) {}
void PostgresParser::Parse(const std::string &query) {
duckdb_libpgquery::pg_parser_init();
duckdb_libpgquery::parse_result res;
pg_parser_parse(query.c_str(), &res);
success = res.success;
if (success) {
parse_tree = res.parse_tree;
} else {
error_message = std::string(res.error_message);
error_location = res.error_location;
}
}
vector<duckdb_libpgquery::PGSimplifiedToken> PostgresParser::Tokenize(const std::string &query) {
duckdb_libpgquery::pg_parser_init();
auto tokens = duckdb_libpgquery::tokenize(query.c_str());
duckdb_libpgquery::pg_parser_cleanup();
return std::move(tokens);
}
PostgresParser::~PostgresParser() {
duckdb_libpgquery::pg_parser_cleanup();
}
duckdb_libpgquery::PGKeywordCategory PostgresParser::IsKeyword(const std::string &text) {
return duckdb_libpgquery::is_keyword(text.c_str());
}
vector<duckdb_libpgquery::PGKeyword> PostgresParser::KeywordList() {
// FIXME: because of this, we might need to change the libpg_query library to use duckdb::vector
vector<duckdb_libpgquery::PGKeyword> tmp(duckdb_libpgquery::keyword_list());
return tmp;
}
void PostgresParser::SetPreserveIdentifierCase(bool preserve) {
duckdb_libpgquery::set_preserve_identifier_case(preserve);
}
}
|
