blob: 4eba367348100fafee980396067c63a7f65d4165 (
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
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
|
#include "yql_factory.h"
#include "yql_formatcode.h"
#include "yql_formattype.h"
#include "yql_formattypediff.h"
#include "yql_makecode.h"
#include "yql_maketype.h"
#include "yql_parsetypehandle.h"
#include "yql_reprcode.h"
#include "yql_serializetypehandle.h"
#include "yql_splittype.h"
#include "yql_typehandle.h"
#include "yql_typekind.h"
namespace NKikimr {
namespace NMiniKQL {
typedef IComputationNode* (*TYqlCallableComputationNodeBuilderFunc)(TCallable& callable, const TComputationNodeFactoryContext& ctx, ui32 exprCtxMutableIndex);
typedef THashMap<TString, TYqlCallableComputationNodeBuilderFunc> TYqlCallableComputationNodeBuilderFuncMap;
struct TYqlCallableComputationNodeBuilderFuncMapFiller {
TYqlCallableComputationNodeBuilderFuncMap Map;
TYqlCallableComputationNodeBuilderFuncMapFiller()
{
Map["FormatType"] = &WrapFormatType;
Map["FormatTypeDiff"] = &WrapFormatTypeDiff;
Map["ParseTypeHandle"] = &WrapParseTypeHandle;
Map["SerializeTypeHandle"] = &WrapSerializeTypeHandle;
Map["TypeHandle"] = &WrapTypeHandle;
Map["TypeKind"] = &WrapTypeKind;
Map["DataTypeComponents"] = &WrapSplitType<NYql::ETypeAnnotationKind::Data>;
Map["DataTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Data>;
Map["OptionalItemType"] = &WrapSplitType<NYql::ETypeAnnotationKind::Optional>;
Map["OptionalTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Optional>;
Map["ListItemType"] = &WrapSplitType<NYql::ETypeAnnotationKind::List>;
Map["ListTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::List>;
Map["StreamItemType"] = &WrapSplitType<NYql::ETypeAnnotationKind::Stream>;
Map["StreamTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Stream>;
Map["TupleTypeComponents"] = &WrapSplitType<NYql::ETypeAnnotationKind::Tuple>;
Map["TupleTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Tuple>;
Map["StructTypeComponents"] = &WrapSplitType<NYql::ETypeAnnotationKind::Struct>;
Map["StructTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Struct>;
Map["DictTypeComponents"] = &WrapSplitType<NYql::ETypeAnnotationKind::Dict>;
Map["DictTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Dict>;
Map["ResourceTypeTag"] = &WrapSplitType<NYql::ETypeAnnotationKind::Resource>;
Map["ResourceTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Resource>;
Map["TaggedTypeComponents"] = &WrapSplitType<NYql::ETypeAnnotationKind::Tagged>;
Map["TaggedTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Tagged>;
Map["VariantUnderlyingType"] = &WrapSplitType<NYql::ETypeAnnotationKind::Variant>;
Map["VariantTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Variant>;
Map["VoidTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Void>;
Map["NullTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Null>;
Map["EmptyListTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::EmptyList>;
Map["EmptyDictTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::EmptyDict>;
Map["CallableTypeComponents"] = &WrapSplitType<NYql::ETypeAnnotationKind::Callable>;
Map["CallableTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Callable>;
Map["PgTypeName"] = &WrapSplitType<NYql::ETypeAnnotationKind::Pg>;
Map["PgTypeHandle"] = &WrapMakeType<NYql::ETypeAnnotationKind::Pg>;
Map["FormatCode"] = &WrapFormatCode<false>;
Map["FormatCodeWithPositions"] = &WrapFormatCode<true>;
Map["SerializeCode"] = &WrapSerializeCode;
Map["WorldCode"] = &WrapMakeCode<NYql::TExprNode::World>;
Map["AtomCode"] = &WrapMakeCode<NYql::TExprNode::Atom>;
Map["ListCode"] = &WrapMakeCode<NYql::TExprNode::List>;
Map["FuncCode"] = &WrapMakeCode<NYql::TExprNode::Callable>;
Map["LambdaCode"] = &WrapMakeCode<NYql::TExprNode::Lambda>;
Map["ReprCode"] = &WrapReprCode;
}
};
TComputationNodeFactory GetYqlFactory(ui32 exprCtxMutableIndex) {
return [exprCtxMutableIndex](TCallable& callable, const TComputationNodeFactoryContext& ctx) -> IComputationNode* {
const auto& map = Singleton<TYqlCallableComputationNodeBuilderFuncMapFiller>()->Map;
auto it = map.find(callable.GetType()->GetName());
if (it == map.end())
return nullptr;
return it->second(callable, ctx, exprCtxMutableIndex);
};
}
TComputationNodeFactory GetYqlFactory() {
TComputationNodeFactory yqlFactory;
return [yqlFactory]
(TCallable& callable, const TComputationNodeFactoryContext& ctx) mutable -> IComputationNode* {
if (!yqlFactory) {
yqlFactory = GetYqlFactory(ctx.Mutables.CurValueIndex++);
}
return yqlFactory(callable, ctx);
};
}
}
}
|
