63 lines
1.2 KiB
Go
63 lines
1.2 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type WebsocketBag struct {
|
|
mu sync.Mutex
|
|
conns map[uuid.UUID]*context.CancelFunc
|
|
}
|
|
|
|
// Returns the websocket bag which contains all of the currently open websocket connections
|
|
// for the server instance.
|
|
func (s *Server) Websockets() *WebsocketBag {
|
|
s.wsBagLocker.Lock()
|
|
defer s.wsBagLocker.Unlock()
|
|
|
|
if s.wsBag == nil {
|
|
s.wsBag = &WebsocketBag{}
|
|
}
|
|
|
|
return s.wsBag
|
|
}
|
|
|
|
// Adds a new websocket connection to the stack.
|
|
func (w *WebsocketBag) Push(u uuid.UUID, cancel *context.CancelFunc) {
|
|
w.mu.Lock()
|
|
defer w.mu.Unlock()
|
|
|
|
if w.conns == nil {
|
|
w.conns = make(map[uuid.UUID]*context.CancelFunc)
|
|
}
|
|
|
|
w.conns[u] = cancel
|
|
}
|
|
|
|
// Removes a connection from the stack.
|
|
func (w *WebsocketBag) Remove(u uuid.UUID) {
|
|
w.mu.Lock()
|
|
delete(w.conns, u)
|
|
w.mu.Unlock()
|
|
}
|
|
|
|
// Cancels all of the stored cancel functions which has the effect of disconnecting
|
|
// every listening websocket for the server.
|
|
func (w *WebsocketBag) CancelAll() {
|
|
w.mu.Lock()
|
|
w.mu.Unlock()
|
|
|
|
if w.conns != nil {
|
|
for _, cancel := range w.conns {
|
|
c := *cancel
|
|
c()
|
|
}
|
|
}
|
|
|
|
// Reset the connections.
|
|
w.conns = make(map[uuid.UUID]*context.CancelFunc)
|
|
}
|