From fa3f970748bf3794781477cbb4ff12ea021320b7 Mon Sep 17 00:00:00 2001
From: dapan1121 <89396746@qq.com>
Date: Mon, 10 May 2021 09:22:10 +0800
Subject: [PATCH] fix mem leak

---
 src/client/src/tscPrepare.c     | 7 +++++--
 tests/script/api/batchprepare.c | 2 ++
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/client/src/tscPrepare.c b/src/client/src/tscPrepare.c
index bc333625ef..c8a3b963f0 100644
--- a/src/client/src/tscPrepare.c
+++ b/src/client/src/tscPrepare.c
@@ -1384,8 +1384,11 @@ int taos_stmt_close(TAOS_STMT* stmt) {
     free(normal->parts);
     free(normal->sql);
   } else {
-    taosHashCleanup(pStmt->mtb.pTableHash);
-    taosHashCleanup(pStmt->mtb.pTableBlockHashList);
+    if (pStmt->multiTbInsert) {
+      taosHashCleanup(pStmt->mtb.pTableHash);
+      pStmt->mtb.pTableBlockHashList = tscDestroyBlockHashTable(pStmt->mtb.pTableBlockHashList, true);
+      taosHashCleanup(pStmt->pSql->cmd.pTableBlockHashList);
+    }
   }
 
   taos_free_result(pStmt->pSql);
diff --git a/tests/script/api/batchprepare.c b/tests/script/api/batchprepare.c
index 5303b53329..f3ea95abfa 100644
--- a/tests/script/api/batchprepare.c
+++ b/tests/script/api/batchprepare.c
@@ -2050,6 +2050,8 @@ int sql_perf_s1(TAOS     *taos) {
     free(sql[i]);
   }
 
+  free(sql);
+
   return 0;
 }