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
|
#include "mkql_fromyson.h"
#include <yql/essentials/minikql/computation/mkql_computation_node_holders.h>
#include <yql/essentials/minikql/mkql_node_cast.h>
#include <yql/essentials/minikql/mkql_node_builder.h>
namespace NKikimr {
namespace NMiniKQL {
namespace {
template<bool IsOptional>
class TFromYsonSimpleTypeWrapper : public TMutableComputationNode<TFromYsonSimpleTypeWrapper<IsOptional>> {
typedef TMutableComputationNode<TFromYsonSimpleTypeWrapper<IsOptional>> TBaseComputation;
public:
TFromYsonSimpleTypeWrapper(TComputationMutables& mutables, IComputationNode* data, NUdf::TDataTypeId schemeType)
: TBaseComputation(mutables)
, Data(data)
, SchemeType(NUdf::GetDataSlot(schemeType))
{}
NUdf::TUnboxedValuePod DoCalculate(TComputationContext& ctx) const {
const auto& data = Data->GetValue(ctx);
if (IsOptional && !data) {
return NUdf::TUnboxedValuePod();
}
return SimpleValueFromYson(SchemeType, data.AsStringRef());
}
private:
void RegisterDependencies() const final {
this->DependsOn(Data);
}
IComputationNode* const Data;
const NUdf::EDataSlot SchemeType;
};
}
IComputationNode* WrapFromYsonSimpleType(TCallable& callable, const TComputationNodeFactoryContext& ctx) {
MKQL_ENSURE(callable.GetInputsCount() == 2, "Expected 2 args");
bool isOptional;
const auto dataType = UnpackOptionalData(callable.GetInput(0), isOptional);
const auto dataSchemeType = dataType->GetSchemeType();
MKQL_ENSURE(dataSchemeType == NUdf::TDataType<char*>::Id || dataSchemeType == NUdf::TDataType<NUdf::TYson>::Id,
"Expected String or Yson");
const auto schemeTypeData = AS_VALUE(TDataLiteral, callable.GetInput(1));
const auto schemeType = schemeTypeData->AsValue().Get<ui32>();
const auto data = LocateNode(ctx.NodeLocator, callable, 0);
if (isOptional) {
return new TFromYsonSimpleTypeWrapper<true>(ctx.Mutables, data, static_cast<NUdf::TDataTypeId>(schemeType));
} else {
return new TFromYsonSimpleTypeWrapper<false>(ctx.Mutables, data, static_cast<NUdf::TDataTypeId>(schemeType));
}
}
}
}
|
