From 2a248056466945bbd227f48dc71c076adcdcfbb3 Mon Sep 17 00:00:00 2001
From: 54liuyao <54liuyao@163.com>
Date: Fri, 1 Nov 2024 11:34:18 +0800
Subject: [PATCH] add block data capacity

---
 source/libs/executor/src/groupoperator.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/source/libs/executor/src/groupoperator.c b/source/libs/executor/src/groupoperator.c
index 9cf2a3ea17..0f85c3346d 100644
--- a/source/libs/executor/src/groupoperator.c
+++ b/source/libs/executor/src/groupoperator.c
@@ -1263,7 +1263,10 @@ static SSDataBlock* buildStreamPartitionResult(SOperatorInfo* pOperator) {
   QUERY_CHECK_CONDITION((hasRemainPartion(pInfo)), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
   SPartitionDataInfo* pParInfo = (SPartitionDataInfo*)pInfo->parIte;
   blockDataCleanup(pDest);
-  int32_t      rows = taosArrayGetSize(pParInfo->rowIds);
+  int32_t rows = taosArrayGetSize(pParInfo->rowIds);
+  code = blockDataEnsureCapacity(pDest, rows);
+  QUERY_CHECK_CODE(code, lino, _end);
+
   SSDataBlock* pSrc = pInfo->pInputDataBlock;
   for (int32_t i = 0; i < rows; i++) {
     int32_t rowIndex = *(int32_t*)taosArrayGet(pParInfo->rowIds, i);