change handler func slightly

client.go

@@ -37,34 +37,36 @@ var upgrader = websocket.Upgrader{
 }
 
 // ServeWs handles WebSocket requests from clients.
-func ServeWs(hub *Hub, w http.ResponseWriter, r *http.Request) {
-	conn, err := upgrader.Upgrade(w, r, nil)
-	if err != nil {
-		log.Print("Failed to upgrade connection", "err", err)
-		return
-	}
-	conn.SetReadLimit(maxMessageSize)
-	if err := conn.SetReadDeadline(time.Now().Add(pongWait)); err != nil {
-		log.Print("Error setting read deadline", "err", err)
-	}
-	conn.SetPongHandler(func(string) error {
+func ServeWs(hub *Hub) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		conn, err := upgrader.Upgrade(w, r, nil)
+		if err != nil {
+			log.Print("Failed to upgrade connection", "err", err)
+			return
+		}
+		conn.SetReadLimit(maxMessageSize)
 		if err := conn.SetReadDeadline(time.Now().Add(pongWait)); err != nil {
 			log.Print("Error setting read deadline", "err", err)
 		}
-		return nil
-	})
-
-	client := &Client{
-		hub:      hub,
-		conn:     conn,
-		send:     make(chan Message, 256),
-		channels: sync.Map{},
-	}
+		conn.SetPongHandler(func(string) error {
+			if err := conn.SetReadDeadline(time.Now().Add(pongWait)); err != nil {
+				log.Print("Error setting read deadline", "err", err)
+			}
+			return nil
+		})
+
+		client := &Client{
+			hub:      hub,
+			conn:     conn,
+			send:     make(chan Message, 256),
+			channels: sync.Map{},
+		}
 
-	hub.clients.Store(client, true)
+		hub.clients.Store(client, true)
 
-	go client.writePump()
-	go client.readPump()
+		go client.writePump()
+		go client.readPump()
+	}
 }
 
 // readPump reads messages from the WebSocket connection.

cmd/main.go

@@ -17,9 +17,7 @@ func main() {
 
 	// Register routes
 	http.HandleFunc("/trigger", p.HandleTrigger(hub))
-	http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
-		p.ServeWs(hub, w, r)
-	})
+	http.HandleFunc("/ws", p.ServeWs(hub))
 	// Start the server
 	server := &http.Server{
 		Addr: "0.0.0.0:8945",

go.mod

@@ -1,5 +1,5 @@
 module github.com/biohackerellie/pushpop
 
-go 1.23.3
+go 1.23.4
 
 require github.com/gorilla/websocket v1.5.3