From 085936e61cdd7cfe46ef69bcf9b109f5f2f4dd1c Mon Sep 17 00:00:00 2001
From: Shuduo Sang <sangshuduo@gmail.com>
Date: Fri, 17 Feb 2023 18:54:08 +0800
Subject: [PATCH] fix: taos shell prompt if not ws error (#20032)

---
 tools/shell/src/shellWebsocket.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/shell/src/shellWebsocket.c b/tools/shell/src/shellWebsocket.c
index b93cc02964..e848c07b66 100644
--- a/tools/shell/src/shellWebsocket.c
+++ b/tools/shell/src/shellWebsocket.c
@@ -218,8 +218,8 @@ void shellRunSingleCommandWebsocketImp(char *command) {
     res = ws_query_timeout(shell.ws_conn, command, shell.args.timeout);
     int code = ws_errno(res);
     if (code != 0 && !shell.stop_query) {
-      // websocket interface masked off first bit from standard error number.
-      if (TSDB_CODE_PAR_SYNTAX_ERROR == (code|0x80000000)) {
+      // if it's not a ws connection error
+      if (TSDB_CODE_WS_DSN_ERROR != (code&TSDB_CODE_WS_DSN_ERROR)) {
         et = taosGetTimestampUs();
         fprintf(stderr, "\nDB: error: %s (%.6fs)\n", ws_errstr(res), (et - st)/1E6);
         ws_free_result(res);