fix: conn connect order

taosdata · May 23, 2023 · bbb17e6 · bbb17e6
1 parent 56ee7eb
commit bbb17e6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/benchQuery.c b/src/benchQuery.c
@@ -492,6 +492,7 @@ static int multi_thread_specified_table_query(uint16_t iface, char* dbName) {
 #endif
            } else {
                 closeBenchConn(pThreadInfo->conn);
+                pThreadInfo->conn = NULL;
            }
 
            // need exit in loop

diff --git a/src/benchUtil.c b/src/benchUtil.c
@@ -310,12 +310,17 @@ SBenchConn* initBenchConn() {
 }
 
 void closeBenchConn(SBenchConn* conn) {
+    if(conn == NULL)
+       return ;
 #ifdef WEBSOCKET
     if (g_arguments->websocket) {
         ws_close(conn->taos_ws);
     } else {
 #endif
-        taos_close(conn->taos);
+        if(conn->taos) {
+            taos_close(conn->taos);
+            conn->taos = NULL;
+        }        
         if (conn->ctaos) {
             taos_close(conn->ctaos);
             conn->ctaos = NULL;