fix: fix kill query crash issue

d5cfbebd · dapan1121 · 9a024590 · d5cfbebd · d5cfbebd
隐藏空白更改
内联并排

Showing with 6 addition and 2 deletion

source/client/src/clientMain.c source/client/src/clientMain.c +3 -1

tools/shell/src/shellEngine.c tools/shell/src/shellEngine.c +3 -1

未找到文件。
--- a/source/client/src/clientMain.c
+++ b/source/client/src/clientMain.c
@@ -222,7 +222,9 @@ void taos_kill_query(TAOS *taos) {

  int64_t  rid = *(int64_t *)taos;
  STscObj *pTscObj = acquireTscObj(rid);
-  stopAllRequests(pTscObj->pRequests);
+  if (pTscObj) {
+    stopAllRequests(pTscObj->pRequests);
+  }
  releaseTscObj(rid);
 }


--- a/tools/shell/src/shellEngine.c
+++ b/tools/shell/src/shellEngine.c
@@ -998,7 +998,9 @@ void *shellCancelHandler(void *arg) {
 		shell.stop_query = true;
 	} else {
 #endif
-		taos_kill_query(shell.conn);
+    if (shell.conn) {
+		  taos_kill_query(shell.conn);
+		}
 #ifdef WEBSOCKET
 	}
 #endif