diff --git a/internal/websocket/client_commands.go b/internal/websocket/client_commands.go index d46f451..c53f45b 100644 --- a/internal/websocket/client_commands.go +++ b/internal/websocket/client_commands.go @@ -18,7 +18,7 @@ func handleClientCommands(ws *websocket.Conn) { _, message, err := ws.ReadMessage() if err != nil { - log.Printf("Client disconnected: %v", err) + log.Printf("Client disconnected from %s (%v)", ws.NetConn().RemoteAddr(), err) delete(Clients, ws) break } diff --git a/internal/websocket/handle_connections.go b/internal/websocket/handle_connections.go index be49918..c80605a 100644 --- a/internal/websocket/handle_connections.go +++ b/internal/websocket/handle_connections.go @@ -27,6 +27,8 @@ func HandleConnections(w http.ResponseWriter, r *http.Request) { } defer ws.Close() + log.Printf("Client connected from %v", ws.NetConn().RemoteAddr()) + // Register the new client mu.Lock() Clients[ws] = &models.Client{