mirror of
https://github.com/bluenviron/mediamtx
synced 2024-12-20 05:40:22 +00:00
46 lines
733 B
Go
46 lines
733 B
Go
package path
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/aler9/gortsplib"
|
|
)
|
|
|
|
type reader interface {
|
|
OnFrame(int, gortsplib.StreamType, []byte)
|
|
}
|
|
|
|
type readersMap struct {
|
|
mutex sync.RWMutex
|
|
ma map[reader]struct{}
|
|
}
|
|
|
|
func newReadersMap() *readersMap {
|
|
return &readersMap{
|
|
ma: make(map[reader]struct{}),
|
|
}
|
|
}
|
|
|
|
func (m *readersMap) add(reader reader) {
|
|
m.mutex.Lock()
|
|
defer m.mutex.Unlock()
|
|
|
|
m.ma[reader] = struct{}{}
|
|
}
|
|
|
|
func (m *readersMap) remove(reader reader) {
|
|
m.mutex.Lock()
|
|
defer m.mutex.Unlock()
|
|
|
|
delete(m.ma, reader)
|
|
}
|
|
|
|
func (m *readersMap) forwardFrame(trackID int, streamType gortsplib.StreamType, buf []byte) {
|
|
m.mutex.RLock()
|
|
defer m.mutex.RUnlock()
|
|
|
|
for c := range m.ma {
|
|
c.OnFrame(trackID, streamType, buf)
|
|
}
|
|
}
|