summaryrefslogtreecommitdiffstats
path: root/yql/essentials/sql/v1/select_yql_window.cpp
blob: 126fd267e741573b469de17b925a69ac054b6501 (plain) (blame)
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
#include "select_yql_window.h"

#include "node.h"

namespace NSQLTranslationV1 {

IYqlWindowLikeNode::IYqlWindowLikeNode(TPosition position)
    : INode(std::move(position))
{
}

bool IYqlWindowLikeNode::SetYqlSelectWindowName(TContext& ctx, TString windowName) {
    Y_UNUSED(ctx);
    WindowName_ = std::move(windowName);
    return true;
}

TString IYqlWindowLikeNode::GetWindowName() const {
    return WindowName_;
}

class TYqlWindow final: public IYqlWindowLikeNode, public TYqlWindowArgs {
public:
    TYqlWindow(TPosition position, TYqlWindowArgs&& args)
        : IYqlWindowLikeNode(std::move(position))
        , TYqlWindowArgs(std::move(args))
    {
    }

    bool DoInit(TContext& ctx, ISource* src) override {
        if (!NSQLTranslationV1::Init(ctx, src, Args)) {
            return false;
        }

        Apply_ = Y("YqlWin", Q(Name), Q(GetWindowName()), Q(Y()), Y("Void"));
        for (const auto& arg : Args) {
            Apply_ = L(std::move(Apply_), arg);
        }

        return true;
    }

    TAstNode* Translate(TContext& ctx) const override {
        return Apply_->Translate(ctx);
    }

    TNodePtr DoClone() const override {
        return new TYqlWindow(*this);
    }

private:
    TNodePtr Apply_;
};

TNodePtr BuildYqlWindow(TPosition position, TYqlWindowArgs&& args) {
    return new TYqlWindow(std::move(position), std::move(args));
}

} // namespace NSQLTranslationV1