diff --git a/ws-server/main.go b/ws-server/main.go index b1e0d86..06f8be9 100644 --- a/ws-server/main.go +++ b/ws-server/main.go @@ -81,6 +81,7 @@ func (ws *WSConnection) Open() { log.Printf("Client connected") ws.ReadChan = make(chan string, 1024) ws.WriteChan = make(chan string, 1024) + ws.ErrorChan = make(chan error, 16) ws.conn.SetReadLimit(1024) ws.conn.SetReadDeadline(time.Now().Add(IDLE_TIMEOUT)) @@ -103,6 +104,7 @@ func (ws *WSConnection) Open() { case err := <-ws.ErrorChan: log.Printf("Error: %v", err) ws.conn.Close() + log.Printf("Client disconnected") connections[ws] = false return // case msg := <-wsBroadcast: