1
- #include " mkql_block_tuple .h"
1
+ #include " mkql_block_container .h"
2
2
3
3
#include < ydb/library/yql/minikql/computation/mkql_block_impl.h>
4
4
@@ -15,9 +15,9 @@ namespace NMiniKQL {
15
15
16
16
namespace {
17
17
18
- class TBlockAsTupleExec {
18
+ class TBlockAsContainerExec {
19
19
public:
20
- TBlockAsTupleExec (const TVector<TType*>& argTypes, const std::shared_ptr<arrow::DataType>& returnArrowType)
20
+ TBlockAsContainerExec (const TVector<TType*>& argTypes, const std::shared_ptr<arrow::DataType>& returnArrowType)
21
21
: ArgTypes(argTypes)
22
22
, ReturnArrowType(returnArrowType)
23
23
{}
@@ -66,10 +66,10 @@ class TBlockAsTupleExec {
66
66
const std::shared_ptr<arrow::DataType> ReturnArrowType;
67
67
};
68
68
69
- std::shared_ptr<arrow::compute::ScalarKernel> MakeBlockAsTupleKernel (const TVector<TType*>& argTypes, TType* resultType) {
69
+ std::shared_ptr<arrow::compute::ScalarKernel> MakeBlockAsContainerKernel (const TVector<TType*>& argTypes, TType* resultType) {
70
70
std::shared_ptr<arrow::DataType> returnArrowType;
71
71
MKQL_ENSURE (ConvertArrowType (AS_TYPE (TBlockType, resultType)->GetItemType (), returnArrowType), " Unsupported arrow type" );
72
- auto exec = std::make_shared<TBlockAsTupleExec >(argTypes, returnArrowType);
72
+ auto exec = std::make_shared<TBlockAsContainerExec >(argTypes, returnArrowType);
73
73
auto kernel = std::make_shared<arrow::compute::ScalarKernel>(ConvertToInputTypes (argTypes), ConvertToOutputType (resultType),
74
74
[exec](arrow::compute::KernelContext* ctx, const arrow::compute::ExecBatch& batch, arrow::Datum* res) {
75
75
return exec->Exec (ctx, batch, res);
@@ -81,17 +81,17 @@ std::shared_ptr<arrow::compute::ScalarKernel> MakeBlockAsTupleKernel(const TVect
81
81
82
82
} // namespace
83
83
84
- IComputationNode* WrapBlockAsTuple (TCallable& callable, const TComputationNodeFactoryContext& ctx) {
84
+ IComputationNode* WrapBlockAsContainer (TCallable& callable, const TComputationNodeFactoryContext& ctx) {
85
85
TComputationNodePtrVector argsNodes;
86
86
TVector<TType*> argsTypes;
87
87
for (ui32 i = 0 ; i < callable.GetInputsCount (); ++i) {
88
88
argsNodes.push_back (LocateNode (ctx.NodeLocator , callable, i));
89
89
argsTypes.push_back (callable.GetInput (i).GetStaticType ());
90
90
}
91
91
92
- auto kernel = MakeBlockAsTupleKernel (argsTypes, callable.GetType ()->GetReturnType ());
92
+ auto kernel = MakeBlockAsContainerKernel (argsTypes, callable.GetType ()->GetReturnType ());
93
93
return new TBlockFuncNode (ctx.Mutables , callable.GetType ()->GetName (), std::move (argsNodes), argsTypes, *kernel, kernel);
94
94
}
95
95
96
- }
97
- }
96
+ } // namespace NMiniKQL
97
+ } // namespace NKikimr
0 commit comments