iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

Capture stream: Add- and RemoveListener.

Browse Source
This commit is contained in:
Miroslav Šedivý
2021-02-06 12:52:02 +01:00
parent 6756114e30
commit f05889bd56
3 changed files with 31 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
internal/capture/stream.go
View File

@ -3,8 +3,8 @@ package capture
import (
"fmt"
"sync"
"reflect"
"github.com/kataras/go-events"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
@ -20,7 +20,8 @@ type StreamManagerCtx struct {
pipelineStr string
pipeline *gst.Pipeline
sample chan types.Sample
emmiter events.EventEmmiter
listeners map[uintptr]func(sample types.Sample)
emitMu sync.Mutex
emitUpdate chan bool
emitStop chan bool
started bool
@ -31,7 +32,7 @@ func streamNew(codec codec.RTPCodec, pipelineStr string) *StreamManagerCtx {
logger: log.With().Str("module", "capture").Str("submodule", "stream").Logger(),
codec: codec,
pipelineStr: pipelineStr,
emmiter: events.New(),
listeners: map[uintptr]func(sample types.Sample){},
emitUpdate: make(chan bool),
emitStop: make(chan bool),
started: false,
@ -48,7 +49,11 @@ func streamNew(codec codec.RTPCodec, pipelineStr string) *StreamManagerCtx {
case <-manager.emitUpdate:
manager.logger.Debug().Msg("update emitting samples")
case sample := <-manager.sample:
manager.emmiter.Emit("sample", sample)
manager.emitMu.Lock()
for _, emit := range manager.listeners {
emit(sample)
}
manager.emitMu.Unlock()
}
}
}()
@ -67,10 +72,20 @@ func (manager *StreamManagerCtx) Codec() codec.RTPCodec {
return manager.codec
}
func (manager *StreamManagerCtx) OnSample(listener func(sample types.Sample)) {
manager.emmiter.On("sample", func(payload ...interface{}) {
listener(payload[0].(types.Sample))
})
func (manager *StreamManagerCtx) AddListener(listener func(sample types.Sample)) {
manager.emitMu.Lock()
defer manager.emitMu.Unlock()
ptr := reflect.ValueOf(listener).Pointer()
manager.listeners[ptr] = listener
}
func (manager *StreamManagerCtx) RemoveListener(listener func(sample types.Sample)) {
manager.emitMu.Lock()
defer manager.emitMu.Unlock()
ptr := reflect.ValueOf(listener).Pointer()
delete(manager.listeners, ptr)
}
func (manager *StreamManagerCtx) Start() error {