From be38b3d602d2a8582a113e0ac59c668f2c554dfa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20Majdand=C5=BEi=C4=87?=
 <david.majdandzic@hotmail.com>
Date: Thu, 20 Jun 2024 17:51:48 +0200
Subject: [PATCH] Fix error handling

---
 ws-server/main.go | 2 ++
 1 file changed, 2 insertions(+)

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: