58 lines
879 B
Go
58 lines
879 B
Go
package server
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"g.fc.run/ai/byre/internal/api"
|
|
)
|
|
|
|
type hub struct {
|
|
mu sync.Mutex
|
|
subs map[chan api.Event]func(api.Event) bool
|
|
}
|
|
|
|
func newHub() *hub {
|
|
return &hub{subs: map[chan api.Event]func(api.Event) bool{}}
|
|
}
|
|
|
|
func (h *hub) subscribe(wanted func(api.Event) bool) chan api.Event {
|
|
h.mu.Lock()
|
|
defer h.mu.Unlock()
|
|
ch := make(chan api.Event, 256)
|
|
h.subs[ch] = wanted
|
|
return ch
|
|
}
|
|
|
|
func (h *hub) unsubscribe(ch chan api.Event) {
|
|
h.mu.Lock()
|
|
defer h.mu.Unlock()
|
|
_, ok := h.subs[ch]
|
|
if !ok {
|
|
return
|
|
}
|
|
delete(h.subs, ch)
|
|
close(ch)
|
|
}
|
|
|
|
func (h *hub) publish(ev api.Event) {
|
|
h.mu.Lock()
|
|
defer h.mu.Unlock()
|
|
for ch, wanted := range h.subs {
|
|
if !wanted(ev) {
|
|
continue
|
|
}
|
|
select {
|
|
case ch <- ev:
|
|
default:
|
|
delete(h.subs, ch)
|
|
close(ch)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (h *hub) count() int {
|
|
h.mu.Lock()
|
|
defer h.mu.Unlock()
|
|
return len(h.subs)
|
|
}
|