61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
|
// Package messaging defines a structure to communication between inputs and outputs
|
||
|
|
package messaging
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Stream makes packages able to subscribe to an incoming stream
|
||
|
|
type Stream struct {
|
||
|
|
// Different qualities of this stream
|
||
|
|
qualities map[string]*Quality
|
||
|
|
|
||
|
|
// Mutex to lock outputs map
|
||
|
|
lockQualities sync.Mutex
|
||
|
|
|
||
|
|
// Count clients for statistics
|
||
|
|
nbClients int
|
||
|
|
}
|
||
|
|
|
||
|
|
func newStream() (s *Stream) {
|
||
|
|
s = &Stream{}
|
||
|
|
s.qualities = make(map[string]*Quality)
|
||
|
|
s.nbClients = 0
|
||
|
|
return s
|
||
|
|
}
|
||
|
|
|
||
|
|
// CreateQuality creates a new quality associated with this stream.
|
||
|
|
func (s *Stream) CreateQuality(name string) (quality *Quality) {
|
||
|
|
s.lockQualities.Lock()
|
||
|
|
quality = newQuality()
|
||
|
|
s.qualities[name] = quality
|
||
|
|
s.lockQualities.Unlock()
|
||
|
|
return quality
|
||
|
|
}
|
||
|
|
|
||
|
|
// DeleteQuality removes a quality.
|
||
|
|
func (s *Stream) DeleteQuality(name string) {
|
||
|
|
// Make sure we did not already close this output
|
||
|
|
s.lockQualities.Lock()
|
||
|
|
if _, ok := s.qualities[name]; ok {
|
||
|
|
s.qualities[name].Close()
|
||
|
|
delete(s.qualities, name)
|
||
|
|
}
|
||
|
|
s.lockQualities.Unlock()
|
||
|
|
}
|
||
|
|
|
||
|
|
// ClientCount returns the number of clients.
|
||
|
|
func (s *Stream) ClientCount() int {
|
||
|
|
return s.nbClients
|
||
|
|
}
|
||
|
|
|
||
|
|
// IncrementClientCount increments the number of clients.
|
||
|
|
func (s *Stream) IncrementClientCount() {
|
||
|
|
s.nbClients++
|
||
|
|
}
|
||
|
|
|
||
|
|
// DecrementClientCount decrements the number of clients.
|
||
|
|
func (s *Stream) DecrementClientCount() {
|
||
|
|
s.nbClients--
|
||
|
|
}
|