mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
90 lines
1.8 KiB
Go
90 lines
1.8 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package eventsource
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
// Manager manages the eventsource Messengers
|
|
type Manager struct {
|
|
mutex sync.Mutex
|
|
|
|
messengers map[int64]*Messenger
|
|
connection chan struct{}
|
|
}
|
|
|
|
var manager *Manager
|
|
|
|
func init() {
|
|
manager = &Manager{
|
|
messengers: make(map[int64]*Messenger),
|
|
connection: make(chan struct{}, 1),
|
|
}
|
|
}
|
|
|
|
// GetManager returns a Manager and initializes one as singleton if there's none yet
|
|
func GetManager() *Manager {
|
|
return manager
|
|
}
|
|
|
|
// Register message channel
|
|
func (m *Manager) Register(uid int64) <-chan *Event {
|
|
m.mutex.Lock()
|
|
messenger, ok := m.messengers[uid]
|
|
if !ok {
|
|
messenger = NewMessenger(uid)
|
|
m.messengers[uid] = messenger
|
|
}
|
|
select {
|
|
case m.connection <- struct{}{}:
|
|
default:
|
|
}
|
|
m.mutex.Unlock()
|
|
return messenger.Register()
|
|
}
|
|
|
|
// Unregister message channel
|
|
func (m *Manager) Unregister(uid int64, channel <-chan *Event) {
|
|
m.mutex.Lock()
|
|
defer m.mutex.Unlock()
|
|
messenger, ok := m.messengers[uid]
|
|
if !ok {
|
|
return
|
|
}
|
|
if messenger.Unregister(channel) {
|
|
delete(m.messengers, uid)
|
|
}
|
|
}
|
|
|
|
// UnregisterAll message channels
|
|
func (m *Manager) UnregisterAll() {
|
|
m.mutex.Lock()
|
|
defer m.mutex.Unlock()
|
|
for _, messenger := range m.messengers {
|
|
messenger.UnregisterAll()
|
|
}
|
|
m.messengers = map[int64]*Messenger{}
|
|
}
|
|
|
|
// SendMessage sends a message to a particular user
|
|
func (m *Manager) SendMessage(uid int64, message *Event) {
|
|
m.mutex.Lock()
|
|
messenger, ok := m.messengers[uid]
|
|
m.mutex.Unlock()
|
|
if ok {
|
|
messenger.SendMessage(message)
|
|
}
|
|
}
|
|
|
|
// SendMessageBlocking sends a message to a particular user
|
|
func (m *Manager) SendMessageBlocking(uid int64, message *Event) {
|
|
m.mutex.Lock()
|
|
messenger, ok := m.messengers[uid]
|
|
m.mutex.Unlock()
|
|
if ok {
|
|
messenger.SendMessageBlocking(message)
|
|
}
|
|
}
|