50 lines
890 B
Go
50 lines
890 B
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"sync"
|
|
|
|
"github.com/coder/websocket"
|
|
"github.com/coder/websocket/wsjson"
|
|
"github.com/zuadi/tecamino-dbm.git/models"
|
|
)
|
|
|
|
var (
|
|
clients = make(map[*Client]bool)
|
|
clientsMu sync.Mutex
|
|
)
|
|
|
|
type Client struct {
|
|
conn *websocket.Conn
|
|
ctx context.Context
|
|
}
|
|
|
|
func registerClient(c *Client) {
|
|
clientsMu.Lock()
|
|
defer clientsMu.Unlock()
|
|
clients[c] = true
|
|
log.Printf("Client connected (%d total)", len(clients))
|
|
}
|
|
|
|
func unregisterClient(c *Client) {
|
|
clientsMu.Lock()
|
|
defer clientsMu.Unlock()
|
|
delete(clients, c)
|
|
log.Printf("Client disconnected (%d total)", len(clients))
|
|
}
|
|
|
|
func broadcast(data models.JsonData) {
|
|
clientsMu.Lock()
|
|
defer clientsMu.Unlock()
|
|
|
|
for c := range clients {
|
|
go func(client *Client) {
|
|
err := wsjson.Write(client.ctx, client.conn, data)
|
|
if err != nil {
|
|
log.Printf("Broadcast error: %v", err)
|
|
}
|
|
}(c)
|
|
}
|
|
}
|