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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
#pragma once
#include "sql_translation.h"
#include <yql/essentials/parser/proto_ast/gen/v1_proto_split/SQLv1Parser.pb.main.h>
#include <util/string/split.h>
namespace NSQLTranslationV1 {
using namespace NSQLv1Generated;
class TSqlQuery: public TSqlTranslation {
public:
TSqlQuery(TContext& ctx, NSQLTranslation::ESqlMode mode, bool topLevel)
: TSqlTranslation(ctx, mode)
, TopLevel(topLevel)
{
}
TNodePtr Build(const TSQLv1ParserAST& ast);
TNodePtr Build(const std::vector<::NSQLv1Generated::TRule_sql_stmt_core>& ast);
bool Statement(TVector<TNodePtr>& blocks, const TRule_sql_stmt_core& core);
private:
bool DeclareStatement(const TRule_declare_stmt& stmt);
bool ExportStatement(const TRule_export_stmt& stmt);
bool AlterTableAction(const TRule_alter_table_action& node, TAlterTableParameters& params);
bool AlterExternalTableAction(const TRule_alter_external_table_action& node, TAlterTableParameters& params);
bool AlterTableAddColumn(const TRule_alter_table_add_column& node, TAlterTableParameters& params);
bool AlterTableDropColumn(const TRule_alter_table_drop_column& node, TAlterTableParameters& params);
bool AlterTableAlterColumn(const TRule_alter_table_alter_column& node, TAlterTableParameters& params);
bool AlterTableAddFamily(const TRule_family_entry& node, TAlterTableParameters& params);
bool AlterTableAlterFamily(const TRule_alter_table_alter_column_family& node, TAlterTableParameters& params);
bool AlterTableSetTableSetting(const TRule_alter_table_set_table_setting_uncompat& node, TTableSettings& tableSettings, ETableType tableType);
bool AlterTableSetTableSetting(const TRule_alter_table_set_table_setting_compat& node, TTableSettings& tableSettings, ETableType tableType);
bool AlterTableResetTableSetting(const TRule_alter_table_reset_table_setting& node, TTableSettings& tableSettings, ETableType tableType);
bool AlterTableAddIndex(const TRule_alter_table_add_index& node, TAlterTableParameters& params);
void AlterTableDropIndex(const TRule_alter_table_drop_index& node, TAlterTableParameters& params);
void AlterTableRenameTo(const TRule_alter_table_rename_to& node, TAlterTableParameters& params);
bool AlterTableAddChangefeed(const TRule_alter_table_add_changefeed& node, TAlterTableParameters& params);
bool AlterTableAlterChangefeed(const TRule_alter_table_alter_changefeed& node, TAlterTableParameters& params);
void AlterTableDropChangefeed(const TRule_alter_table_drop_changefeed& node, TAlterTableParameters& params);
void AlterTableRenameIndexTo(const TRule_alter_table_rename_index_to& node, TAlterTableParameters& params);
bool AlterTableAlterIndex(const TRule_alter_table_alter_index& node, TAlterTableParameters& params);
TNodePtr PragmaStatement(const TRule_pragma_stmt& stmt, bool& success);
void AddStatementToBlocks(TVector<TNodePtr>& blocks, TNodePtr node);
bool ParseTableStoreFeatures(std::map<TString, TDeferredAtom> & result, const TRule_alter_table_store_action & actions);
bool AlterTableAlterColumnDropNotNull(const TRule_alter_table_alter_column_drop_not_null& node, TAlterTableParameters& params);
TNodePtr Build(const TRule_delete_stmt& stmt);
TNodePtr Build(const TRule_update_stmt& stmt);
TSourcePtr Build(const TRule_set_clause_choice& stmt);
bool FillSetClause(const TRule_set_clause& node, TVector<TString>& targetList, TVector<TNodePtr>& values);
TSourcePtr Build(const TRule_set_clause_list& stmt);
TSourcePtr Build(const TRule_multiple_column_assignment& stmt);
template<class TNode>
void ParseStatementName(const TNode& node, TString& internalStatementName, TString& humanStatementName) {
internalStatementName.clear();
humanStatementName.clear();
const auto& descr = AltDescription(node);
TVector<TString> parts;
if (!Ctx.Settings.Antlr4Parser) {
const auto pos = descr.find(": ");
Y_DEBUG_ABORT_UNLESS(pos != TString::npos);
Split(TString(descr.begin() + pos + 2, descr.end()), "_", parts);
} else {
Split(descr, "_", parts);
}
Y_DEBUG_ABORT_UNLESS(parts.size() > 1);
parts.pop_back();
for (auto& part: parts) {
part.to_upper(0, 1);
internalStatementName += part;
if (!humanStatementName.empty()) {
humanStatementName += ' ';
}
humanStatementName += to_upper(part);
}
}
const bool TopLevel;
};
} // namespace NSQLTranslationV1
|
