Merge 9439737 into 41ad1d4

server/server.go

@@ -119,6 +119,7 @@ func WsHandler(w http.ResponseWriter, r *http.Request, config *config.Config) {
 		messageType, p, err := conn.ReadMessage()
 		if err != nil {
 			log.Printf("Error reading message: %v", err)
+			removeFromClients(conn) // Remove the faulty connection
 			break
 		}
 
@@ -128,6 +129,19 @@ func WsHandler(w http.ResponseWriter, r *http.Request, config *config.Config) {
 	}
 }
 
+func removeFromClients(conn *websocket.Conn) {
+	lock.Lock()
+	defer lock.Unlock()
+	for i, client := range clients {
+		if client.Conn == conn {
+			// Remove the client safely without memory leak
+			clients[i] = clients[len(clients)-1]
+			clients = clients[:len(clients)-1]
+			break
+		}
+	}
+}
+
 // 处理收到的信息
 func processWSMessage(msg []byte) {
 	var genericMap map[string]interface{}