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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
#pragma once
#include "node.h"
#include <yql/essentials/core/sql_types/match_recognize.h>
#include <util/generic/ptr.h>
namespace NSQLTranslationV1 {
struct TNamedFunction {
TNodePtr callable; //Callable with some free args
TString name;
};
class TMatchRecognizeBuilder: public TSimpleRefCount<TMatchRecognizeBuilder> {
public:
TMatchRecognizeBuilder(
TPosition clausePos,
std::pair<TPosition, TVector<TNamedFunction>>&& partitioners,
std::pair<TPosition, TVector<TSortSpecificationPtr>>&& sortSpecs,
std::pair<TPosition, TVector<TNamedFunction>>&& measures,
std::pair<TPosition, NYql::NMatchRecognize::ERowsPerMatch>&& rowsPerMatch,
std::pair<TPosition, NYql::NMatchRecognize::TAfterMatchSkipTo>&& skipTo,
std::pair<TPosition, NYql::NMatchRecognize::TRowPattern>&& pattern,
std::pair<TPosition, TNodePtr>&& subset,
std::pair<TPosition, TVector<TNamedFunction>>&& definitions
)
: Pos(clausePos)
, Partitioners(std::move(partitioners))
, SortSpecs(std::move(sortSpecs))
, Measures(std::move(measures))
, RowsPerMatch(std::move(rowsPerMatch))
, SkipTo(std::move(skipTo))
, Pattern(std::move(pattern))
, Subset(std::move(subset))
, Definitions(definitions)
{}
TNodePtr Build(TContext& ctx, TString&& inputTable, ISource* source);
private:
TPosition Pos;
std::pair<TPosition, TVector<TNamedFunction>> Partitioners;
std::pair<TPosition, TVector<TSortSpecificationPtr>> SortSpecs;
std::pair<TPosition, TVector<TNamedFunction>> Measures;
std::pair<TPosition, NYql::NMatchRecognize::ERowsPerMatch> RowsPerMatch;
std::pair<TPosition, NYql::NMatchRecognize::TAfterMatchSkipTo> SkipTo;
std::pair<TPosition, NYql::NMatchRecognize::TRowPattern> Pattern;
std::pair<TPosition, TNodePtr> Subset;
std::pair<TPosition, TVector<TNamedFunction>> Definitions;
};
using TMatchRecognizeBuilderPtr=TIntrusivePtr<TMatchRecognizeBuilder> ;
class TMatchRecognizeVarAccessNode: public INode {
public:
TMatchRecognizeVarAccessNode(TPosition pos, const TString& var, const TString& column, bool theSameVar)
: INode(pos)
, Var(var)
, TheSameVar(theSameVar)
, Column(column)
{
}
TString GetVar() const {
return Var;
}
bool IsTheSameVar() const {
return TheSameVar;
}
TString GetColumn() const {
return Column;
}
bool DoInit(TContext& ctx, ISource* src) override;
TAstNode* Translate(TContext& ctx) const override {
return Node->Translate(ctx);
}
TPtr DoClone() const override {
YQL_ENSURE(!Node, "TMatchRecognizeVarAccessNode::Clone: Node must not be initialized");
auto copy = new TMatchRecognizeVarAccessNode(Pos, Var, Column, TheSameVar);
return copy;
}
protected:
void DoUpdateState() const override {
YQL_ENSURE(Node);
}
void DoVisitChildren(const TVisitFunc& func, TVisitNodeSet& visited) const final {
Y_DEBUG_ABORT_UNLESS(Node);
Node->VisitTree(func, visited);
}
private:
TNodePtr Node;
const TString Var;
const bool TheSameVar; //reference the same var as being defined by this expression;
const TString Column;
};
class TMatchRecognizeNavigate: public TAstListNode {
public:
TMatchRecognizeNavigate(TPosition pos, const TString& name, const TVector<TNodePtr>& args)
: TAstListNode(pos)
, Name(name)
, Args(args)
{
}
private:
TNodePtr DoClone() const override {
return new TMatchRecognizeNavigate(GetPos(), Name, CloneContainer(Args));
}
bool DoInit(TContext& ctx, ISource* src) override;
private:
const TString Name;
const TVector<TNodePtr> Args;
};
} // namespace NSQLTranslationV1
|
