94 lines
1.9 KiB
Go
94 lines
1.9 KiB
Go
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)
|
|
}
|