feat(ws server): added in a websocket on port system to help with better logging

backend/cmd/services/websocket/client.go

@@ -0,0 +1,93 @@
+package socketio
+
+import (
+	"log"
+	"sync"
+
+	"github.com/google/uuid"
+	"github.com/gorilla/websocket"
+	logging "lst.net/utils/logger"
+)
+
+type Client struct {
+	ClientID  uuid.UUID
+	Conn      *websocket.Conn
+	APIKey    string
+	IPAddress string
+	UserAgent string
+	Send      chan []byte
+	Channels  map[string]bool // e.g., {"logs": true, "labels": true}
+}
+
+var clients = make(map[*Client]bool)
+
+var clientsLock sync.Mutex
+
+func init() {
+	var broadcast = make(chan string)
+	go func() {
+		for {
+			msg := <-broadcast
+
+			clientsLock.Lock()
+			for client := range clients {
+				if client.Channels["logs"] {
+					err := client.Conn.WriteMessage(websocket.TextMessage, []byte(msg))
+					if err != nil {
+						log.Println("Write error:", err)
+						client.Conn.Close()
+						//client.MarkDisconnected()
+						delete(clients, client)
+					}
+				}
+			}
+			clientsLock.Unlock()
+		}
+	}()
+}
+
+func StartBroadcasting(broadcaster chan logging.Message) {
+	go func() {
+		log.Println("StartBroadcasting goroutine started")
+		for {
+			msg := <-broadcaster
+			//log.Printf("Received msg on broadcaster: %+v\n", msg)
+			clientsLock.Lock()
+			for client := range clients {
+				if client.Channels[msg.Channel] {
+					log.Println("Sending message to client")
+					err := client.Conn.WriteJSON(msg)
+					if err != nil {
+						log.Println("Write error:", err)
+						client.Conn.Close()
+						client.MarkDisconnected()
+						delete(clients, client)
+					}
+				} else {
+					log.Println("Skipping client, channel mismatch")
+				}
+			}
+			clientsLock.Unlock()
+		}
+	}()
+}
+
+// func (c *Client) JoinChannel(name string) {
+// 	ch := GetOrCreateChannel(name)
+// 	c.Channels[name] = ch
+// 	ch.Register <- c
+// }
+
+// func (c *Client) LeaveChannel(name string) {
+// 	if ch, ok := c.Channels[name]; ok {
+// 		ch.Unregister <- c
+// 		delete(c.Channels, name)
+// 	}
+// }
+
+func (c *Client) Disconnect() {
+	// for _, ch := range c.Channels {
+	// 	ch.Unregister <- c
+	// }
+	close(c.Send)
+}