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
|
#include "object_processing.h"
#include <yql/essentials/core/sql_types/yql_callable_names.h>
namespace NSQLTranslationV1 {
using namespace NYql;
INode::TPtr TObjectProcessorImpl::BuildKeys() const {
auto keys = Y("Key");
keys = L(keys, Q(Y(Q("objectId"), Y("String", BuildQuotedAtom(Pos, ObjectId)))));
keys = L(keys, Q(Y(Q("typeId"), Y("String", BuildQuotedAtom(Pos, TypeId)))));
return keys;
}
TObjectProcessorImpl::TObjectProcessorImpl(TPosition pos, const TString& objectId, const TString& typeId, const TObjectOperatorContext& context)
: TBase(pos)
, TObjectOperatorContext(context)
, ObjectId(objectId)
, TypeId(typeId)
{
}
bool TObjectProcessorImpl::DoInit(TContext& ctx, ISource* src) {
Y_UNUSED(src);
Scoped->UseCluster(ServiceId, Cluster);
auto options = FillFeatures(BuildOptions());
auto keys = BuildKeys();
Add("block", Q(Y(
Y("let", "sink", Y("DataSink", BuildQuotedAtom(Pos, ServiceId), Scoped->WrapCluster(Cluster, ctx))),
Y("let", "world", Y(TString(WriteName), "world", "sink", keys, Y("Void"), Q(options))),
Y("return", ctx.PragmaAutoCommit ? Y(TString(CommitName), "world", "sink") : AstNode("world"))
)));
return TAstListNode::DoInit(ctx, src);
}
INode::TPtr TCreateObject::FillFeatures(INode::TPtr options) const {
if (!Features.empty()) {
auto features = Y();
for (auto&& i : Features) {
if (i.second.HasNode()) {
features->Add(Q(Y(BuildQuotedAtom(Pos, i.first), i.second.Build())));
} else {
features->Add(Q(Y(BuildQuotedAtom(Pos, i.first))));
}
}
options->Add(Q(Y(Q("features"), Q(features))));
}
if (!FeaturesToReset.empty()) {
auto reset = Y();
for (const auto& featureName : FeaturesToReset) {
reset->Add(BuildQuotedAtom(Pos, featureName));
}
options->Add(Q(Y(Q("resetFeatures"), Q(reset))));
}
return options;
}
TObjectOperatorContext::TObjectOperatorContext(TScopedStatePtr scoped)
: Scoped(scoped)
, ServiceId(Scoped->CurrService)
, Cluster(Scoped->CurrCluster)
{
}
}
|
