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
|
#include "mkql_match_recognize_list.h"
namespace NKikimr::NMiniKQL::NMatchRecognize {
namespace {
class TIterator : public TTemporaryComputationValue<TIterator> {
public:
TIterator(TMemoryUsageInfo* memUsage, const TSparseList& parent)
: TTemporaryComputationValue<TIterator>(memUsage)
, Parent(parent)
, Current(Parent.Begin())
{}
private:
bool Skip() final {
return ++Current != Parent.End();
}
bool Next(NUdf::TUnboxedValue& value) final {
if (!Skip()) {
return false;
}
value = Current->second.Value;
return true;
}
bool NextPair(NUdf::TUnboxedValue& key, NUdf::TUnboxedValue& payload) final {
if (!Next(payload)) {
return false;
}
key = NUdf::TUnboxedValuePod(Current->first);
return true;
}
const TSparseList& Parent;
TSparseList::iterator Current;
};
class TKeysIterator : public TTemporaryComputationValue<TKeysIterator> {
public:
TKeysIterator(TMemoryUsageInfo* memUsage, const TSparseList& parent)
: TTemporaryComputationValue<TKeysIterator>(memUsage)
, Parent(parent)
, Current(Parent.Begin())
{}
private:
bool Skip() final {
return ++Current != Parent.End();
}
bool Next(NUdf::TUnboxedValue& key) final {
if (!Skip()) {
return false;
}
key = NUdf::TUnboxedValuePod(Current->first);
return true;
}
const TSparseList& Parent;
TSparseList::iterator Current;
};
} // anonymous namespace
TListValue::TListValue(TMemoryUsageInfo* memUsage, const TSparseList& list)
: TComputationValue<TListValue>(memUsage)
, List(list)
{}
bool TListValue::HasFastListLength() const {
return true;
}
ui64 TListValue::GetListLength() const {
return GetDictLength();
}
ui64 TListValue::GetEstimatedListLength() const {
return GetListLength();
}
NUdf::TUnboxedValue TListValue::GetListIterator() const {
return GetPayloadsIterator();
}
bool TListValue::HasListItems() const {
return HasDictItems();
}
NUdf::IBoxedValuePtr TListValue::ToIndexDictImpl(const NUdf::IValueBuilder& builder) const {
Y_UNUSED(builder);
return const_cast<TListValue*>(this);
}
ui64 TListValue::GetDictLength() const {
return List.Size();
}
NUdf::TUnboxedValue TListValue::GetDictIterator() const {
return NUdf::TUnboxedValuePod(new TIterator(GetMemInfo(), List));
}
NUdf::TUnboxedValue TListValue::GetKeysIterator() const {
return NUdf::TUnboxedValuePod(new TKeysIterator(GetMemInfo(), List));
}
NUdf::TUnboxedValue TListValue::GetPayloadsIterator() const {
return NUdf::TUnboxedValuePod(new TIterator(GetMemInfo(), List));
}
bool TListValue::Contains(const NUdf::TUnboxedValuePod& key) const {
return List.Contains(key.Get<ui64>());
}
NUdf::TUnboxedValue TListValue::Lookup(const NUdf::TUnboxedValuePod& key) const {
return List.Get(key.Get<ui64>());
}
bool TListValue::HasDictItems() const {
return !List.Empty();
}
} // namespace NKikimr::NMiniKQL::NMatchRecognize
|
