httpserver.go

@@ -16,7 +16,7 @@ import (
 )
 
 type WsConnections struct {
-	connections map[*WsConnection]struct{}
+	connections map[*websocket.Conn]struct{}
 	mutex       sync.Mutex
 }
 
@@ -24,24 +24,19 @@ func (wc *WsConnections) Add(c *websocket.Conn) {
 	wc.mutex.Lock()
 	defer wc.mutex.Unlock()
 
-	wc.connections[NewWsConnection(c)] = struct{}{}
+	wc.connections[c] = struct{}{}
 }
 
 func (wc *WsConnections) Delete(c *websocket.Conn) {
 	wc.mutex.Lock()
 	defer wc.mutex.Unlock()
 
-	for k := range wc.connections {
+	delete(wc.connections, c)
-		if k.w == c {
-			delete(wc.connections, k)
-			break
-		}
-	}
 }
 
 func (wc *WsConnections) Send(message interface{}) {
 	for conn := range wc.connections {
-		go func(conn *WsConnection) { _ = conn.Send(message) }(conn)
+		go func(conn *websocket.Conn) { _ = conn.WriteJSON(message) }(conn)
 	}
 }
 
@@ -54,7 +49,7 @@ var upgrader = websocket.Upgrader{
 }
 
 var wsConnections = &WsConnections{
-	connections: make(map[*WsConnection]struct{})}
+	connections: make(map[*websocket.Conn]struct{})}
 
 func httpServer(listenAddress string) {
 	if listenAddress == "none" {

make.bat

@@ -1 +1 @@
-go build -ldflags "-s -w" -buildmode=pie -trimpath
+go build -ldflags "-s -w -H windowsgui" -buildmode=pie -trimpath

wconn.go

@@ -1,23 +0,0 @@
-package main
-
-import (
-	"sync"
-
-	"github.com/gorilla/websocket"
-)
-
-type WsConnection struct {
-	w  *websocket.Conn
-	mu sync.Mutex
-}
-
-func NewWsConnection(w *websocket.Conn) *WsConnection {
-	return &WsConnection{w: w}
-}
-
-func (w *WsConnection) Send(message any) error {
-	w.mu.Lock()
-	defer w.mu.Unlock()
-
-	return w.w.WriteJSON(message)
-}