From 6b38e5b8fb083b8e784ae9cee778e00e40e732cb Mon Sep 17 00:00:00 2001
From: Alex Duan <417921451@qq.com>
Date: Mon, 8 Jan 2024 10:44:19 +0800
Subject: [PATCH] feat: websocket support bi with appending &conn_mode=1 string
 to dsn

---
 tools/shell/src/shellEngine.c    |  4 +---
 tools/shell/src/shellWebsocket.c | 11 ++++++++++-
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/tools/shell/src/shellEngine.c b/tools/shell/src/shellEngine.c
index f6f834f3d7..23424cea98 100644
--- a/tools/shell/src/shellEngine.c
+++ b/tools/shell/src/shellEngine.c
@@ -1318,9 +1318,7 @@ int32_t shellExecute() {
  if(shell.args.is_bi_mode) { 
   // need set bi mode
   printf("Set BI mode is true.\n");
-#ifdef WEBSOCKET
-    //ws_taos_set_conn_mode(shell.ws_conn, TAOS_CONN_MODE_BI, 1);
-#else
+#ifndef WEBSOCKET
     taos_set_conn_mode(shell.conn, TAOS_CONN_MODE_BI, 1);
 #endif
  }
diff --git a/tools/shell/src/shellWebsocket.c b/tools/shell/src/shellWebsocket.c
index 8c3a46aa05..2d3aa27a3b 100644
--- a/tools/shell/src/shellWebsocket.c
+++ b/tools/shell/src/shellWebsocket.c
@@ -30,7 +30,16 @@ int shell_conn_ws_server(bool first) {
   fprintf(stdout, "trying to connect %s****** ", cuttedDsn);
   fflush(stdout);
   for (int i = 0; i < shell.args.timeout; i++) {
-    shell.ws_conn = ws_connect_with_dsn(shell.args.dsn);
+    if(shell.args.is_bi_mode) {
+      size_t len = strlen(shell.args.dsn);
+      char * dsn = taosMemoryMalloc(len + 32);
+      sprintf(dsn, "%s&conn_mode=1", shell.args.dsn);
+      shell.ws_conn = ws_connect_with_dsn(dsn);
+      taosMemoryFree(dsn);
+    } else {
+      shell.ws_conn = ws_connect_with_dsn(shell.args.dsn);
+    }
+
     if (NULL == shell.ws_conn) {
       int errNo = ws_errno(NULL);
       if (0xE001 == errNo) {