fixed bug for websocket close tip message

api/handler/markmap_handler.go

@@ -42,14 +42,8 @@ func (h *MarkMapHandler) Client(c *gin.Context) {
 
 	modelId := h.GetInt(c, "model_id", 0)
 	userId := h.GetInt(c, "user_id", 0)
-	logger.Info(modelId)
 
 	client := types.NewWsClient(ws)
-	if cli := h.clients.Get(userId); cli != nil {
-		cli.Close()
-	}
-
-	// 保存会话连接
 	h.clients.Put(userId, client)
 	go func() {
 		for {

api/handler/mj_handler.go

@@ -78,10 +78,6 @@ func (h *MidJourneyHandler) Client(c *gin.Context) {
 	}
 
 	client := types.NewWsClient(ws)
-	// close the existed connections
-	if cli := h.pool.Clients.Get(uint(userId)); cli != nil {
-		cli.Close()
-	}
 	h.pool.Clients.Put(uint(userId), client)
 	logger.Infof("New websocket connected, IP: %s", c.RemoteIP())
 }

api/handler/sd_handler.go

@@ -61,10 +61,6 @@ func (h *SdJobHandler) Client(c *gin.Context) {
 	}
 
 	client := types.NewWsClient(ws)
-	// close the existed connections
-	if cli := h.pool.Clients.Get(uint(userId)); cli != nil {
-		cli.Close()
-	}
 	h.pool.Clients.Put(uint(userId), client)
 	logger.Infof("New websocket connected, IP: %s", c.RemoteIP())
 }

web/src/views/ImageMj.vue

@@ -734,23 +734,7 @@ const connect = () => {
 
   _socket.addEventListener('close', () => {
     if (socket.value !== null) {
-      ElMessageBox.confirm(
+      connect()
-          '检测到您已经在其他客户端创建了新的连接，当前连接将被关闭！',
-          '提示',
-          {
-            dangerouslyUseHTMLString: true,
-            confirmButtonText: '重新连接',
-            cancelButtonText: '关闭',
-            type: 'warning',
-          }
-      ).then(() => {
-        connect()
-      }).catch(() => {
-        ElMessage({
-          type: 'info',
-          message: '连接已关闭',
-        })
-      })
     }
   });
 }

web/src/views/ImageSd.vue

@@ -577,23 +577,7 @@ const connect = () => {
 
   _socket.addEventListener('close', () => {
     if (socket.value !== null) {
-      ElMessageBox.confirm(
+      connect()
-          '检测到您已经在其他客户端创建了新的连接，当前连接将被关闭！',
-          '提示',
-          {
-            dangerouslyUseHTMLString: true,
-            confirmButtonText: '重新连接',
-            cancelButtonText: '关闭',
-            type: 'warning',
-          }
-      ).then(() => {
-        connect()
-      }).catch(() => {
-        ElMessage({
-          type: 'info',
-          message: '连接已关闭',
-        })
-      })
     }
   })
 }

web/src/views/MarkMap.vue

@@ -254,23 +254,7 @@ const connect = (userId) => {
   _socket.addEventListener('close', () => {
     loading.value = false
     if (socket.value !== null) {
-      ElMessageBox.confirm(
+      connect(userId)
-          '检测到您已经在其他客户端创建了新的连接，当前连接将被关闭！',
-          '提示',
-          {
-            dangerouslyUseHTMLString: true,
-            confirmButtonText: '重新连接',
-            cancelButtonText: '关闭',
-            type: 'warning',
-          }
-      ).then(() => {
-        connect(userId)
-      }).catch(() => {
-        ElMessage({
-          type: 'info',
-          message: '连接已关闭',
-        })
-      })
     }
   });
 }