mirror of
https://github.com/go-gitea/gitea
synced 2024-08-17 12:50:52 +02:00
85 lines
1.8 KiB
Go
85 lines
1.8 KiB
Go
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
||
|
// Use of this source code is governed by a MIT-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
package eventsource
|
||
|
|
||
|
import (
|
||
|
"sync"
|
||
|
)
|
||
|
|
||
|
// Manager manages the eventsource Messengers
|
||
|
type Manager struct {
|
||
|
mutex sync.Mutex
|
||
|
|
||
|
messengers map[int64]*Messenger
|
||
|
}
|
||
|
|
||
|
var manager *Manager
|
||
|
|
||
|
func init() {
|
||
|
manager = &Manager{
|
||
|
messengers: make(map[int64]*Messenger),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// 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
|
||
|
}
|
||
|
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)
|
||
|
}
|
||
|
}
|