From bb0be3c0d4a621da5cbcea18f92c30f70c8e5eaf Mon Sep 17 00:00:00 2001 From: buddha23 <8994556+buddha23@user.noreply.gitee.com> Date: Fri, 11 Nov 2022 01:19:30 +0000 Subject: [PATCH] update omnioperator/omniop-spark-extension-ock/cpp/src/shuffle/ock_merge_reader.cpp. Signed-off-by: buddha23 <8994556+buddha23@user.noreply.gitee.com> --- .../cpp/src/shuffle/ock_merge_reader.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/omnioperator/omniop-spark-extension-ock/cpp/src/shuffle/ock_merge_reader.cpp b/omnioperator/omniop-spark-extension-ock/cpp/src/shuffle/ock_merge_reader.cpp index fe21d3347..d8b53dd71 100644 --- a/omnioperator/omniop-spark-extension-ock/cpp/src/shuffle/ock_merge_reader.cpp +++ b/omnioperator/omniop-spark-extension-ock/cpp/src/shuffle/ock_merge_reader.cpp @@ -15,12 +15,15 @@ using namespace ock::dopspark; bool OckMergeReader::Initialize(const int32_t *typeIds, uint32_t colNum) { mColNum = colNum; - mVectorBatch = new (std::nothrow) VBDataDesc(colNum); + mVectorBatch = std::make_shares(); if (UNLIKELY(mVectorBatch == nullptr)) { LOG_ERROR("Failed to new instance for vector batch description"); return false; } + if (UNLIKELY(!mVectorBatch->Initialize(colNum))) { + LOG_ERROR() + } mColTypeIds.reserve(colNum); for (uint32_t index = 0; index < colNum; ++index) { mColTypeIds.emplace_back(typeIds[index]); -- Gitee