From 6470cfc26e05b107ae914ef3d2736b4cae8d25a1 Mon Sep 17 00:00:00 2001
From: Haojun Liao <hjliao@taosdata.com>
Date: Fri, 6 Sep 2024 15:58:52 +0800
Subject: [PATCH] fix(query): fix memory leak.

---
 source/libs/executor/src/scanoperator.c | 6 +++++-
 source/libs/executor/src/sortoperator.c | 1 -
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/source/libs/executor/src/scanoperator.c b/source/libs/executor/src/scanoperator.c
index 7cc0aa0f16..42e7e4ac3b 100644
--- a/source/libs/executor/src/scanoperator.c
+++ b/source/libs/executor/src/scanoperator.c
@@ -5677,7 +5677,11 @@ int32_t startDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
   param->pOperator = pOperator;
 
   SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
-  QUERY_CHECK_NULL(ps, code, lino, _end, terrno);
+  if (ps == NULL) {
+    taosMemoryFree(param);
+    QUERY_CHECK_NULL(ps, code, lino, _end, terrno);
+  }
+
   ps->param = param;
   ps->onlyRef = false;
   code = tsortAddSource(pInfo->pSortHandle, ps);
diff --git a/source/libs/executor/src/sortoperator.c b/source/libs/executor/src/sortoperator.c
index 545e3febf1..9b659ac761 100644
--- a/source/libs/executor/src/sortoperator.c
+++ b/source/libs/executor/src/sortoperator.c
@@ -369,7 +369,6 @@ int32_t doOpenSortOperator(SOperatorInfo* pOperator) {
     return terrno;
   }
 
-  qInfo("---alloc:%p", pSource);
   pSource->param = pOperator->pDownstream[0];
   pSource->onlyRef = true;