luciano/htmgo
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

lock listeners

Browse source
This commit is contained in:
maddalax 2024-11-12 15:47:06 -06:00
parent c593c466a3
commit 79b15fe194
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
extensions/websocket/ws/manager.go
View file

@ -59,6 +59,7 @@ type SocketManager struct {
idToRoom *xsync.MapOf[string, string]
listeners []chan SocketEvent
listenersQueuedForDelete map[chan SocketEvent]bool
listenerLock sync.Mutex
goroutinesRunning atomic.Int32
opts *opts.ExtensionOpts
lock sync.Mutex
@ -179,16 +180,20 @@ func (manager *SocketManager) RunIntervalWithSocket(socketId string, interval ti
}
func (manager *SocketManager) Listen(listener chan SocketEvent) {
manager.listenerLock.Lock()
if manager.listeners == nil {
manager.listeners = make([]chan SocketEvent, 0)
}
if listener != nil {
manager.listeners = append(manager.listeners, listener)
}
manager.listenerLock.Unlock()
}
func (manager *SocketManager) RemoveListener(listener chan SocketEvent) {
manager.listenerLock.Lock()
manager.listenersQueuedForDelete[listener] = true
manager.listenerLock.Unlock()
}
func (manager *SocketManager) dispatch(event SocketEvent) {
@ -204,6 +209,7 @@ func (manager *SocketManager) dispatch(event SocketEvent) {
}
}()
manager.listenerLock.Lock()
if len(manager.listenersQueuedForDelete) > 0 {
newListener := make([]chan SocketEvent, 0)
for _, listener := range manager.listeners {
@ -224,6 +230,7 @@ func (manager *SocketManager) dispatch(event SocketEvent) {
}
wg.Wait()
manager.listenerLock.Unlock()
done <- struct{}{}
}