72 lines
1.3 KiB
Go
72 lines
1.3 KiB
Go
package channels
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
|
|
"github.com/cskr/pubsub"
|
|
)
|
|
|
|
type ChannelManager struct {
|
|
channels map[string]*Channel
|
|
channelsLock sync.RWMutex
|
|
pubsub *pubsub.PubSub
|
|
}
|
|
|
|
func NewChannelManager() *ChannelManager {
|
|
mgr := &ChannelManager{
|
|
pubsub: pubsub.New(1),
|
|
channels: map[string]*Channel{},
|
|
}
|
|
return mgr
|
|
}
|
|
|
|
func (manager *ChannelManager) Channel(uuid string) *Channel {
|
|
manager.channelsLock.RLock()
|
|
defer manager.channelsLock.RUnlock()
|
|
|
|
channel, ok := manager.channels[uuid]
|
|
if !ok {
|
|
return nil
|
|
}
|
|
|
|
return channel
|
|
}
|
|
|
|
func (manager *ChannelManager) Close(uuid string) (err error) {
|
|
manager.channelsLock.Lock()
|
|
defer manager.channelsLock.Unlock()
|
|
|
|
_, ok := manager.channels[uuid]
|
|
if !ok {
|
|
err = errors.New("channel uuid is not known")
|
|
return
|
|
}
|
|
|
|
manager.pubsub.Pub(manager.channels[uuid], "close")
|
|
delete(manager.channels, uuid)
|
|
|
|
return
|
|
}
|
|
|
|
func (manager *ChannelManager) Open(uuid string) (channel *Channel, err error) {
|
|
manager.channelsLock.Lock()
|
|
defer manager.channelsLock.Unlock()
|
|
|
|
if _, ok := manager.channels[uuid]; ok {
|
|
err = errors.New("channel uuid is already in use")
|
|
return
|
|
}
|
|
|
|
channel = &Channel{Id: uuid}
|
|
manager.channels[uuid] = channel
|
|
|
|
manager.pubsub.Pub(channel, "open")
|
|
|
|
return
|
|
}
|
|
|
|
func (manager *ChannelManager) Shutdown() {
|
|
manager.pubsub.Shutdown()
|
|
}
|