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

add waitgroups.

Browse Source
This commit is contained in:
Miroslav Šedivý
2021-09-02 00:00:56 +02:00
parent 7e2f173460
commit f18ae2b972
3 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/capture/screencast.go
View File

@ -16,6 +16,7 @@ import (
type ScreencastManagerCtx struct {
logger zerolog.Logger
mu sync.Mutex
wg sync.WaitGroup
pipelineStr string
pipeline *gst.Pipeline
enabled bool
@ -39,8 +40,11 @@ func screencastNew(enabled bool, pipelineStr string) *ScreencastManagerCtx {
emitUpdate: make(chan bool),
}
manager.wg.Add(1)
go func() {
manager.logger.Debug().Msg("started emitting samples")
defer manager.wg.Done()
ticker := time.NewTicker(screencastTimeout)
defer ticker.Stop()
@ -66,10 +70,12 @@ func screencastNew(enabled bool, pipelineStr string) *ScreencastManagerCtx {
}
func (manager *ScreencastManagerCtx) shutdown() {
manager.logger.Info().Msgf("shutting down")
manager.logger.Info().Msgf("shutdown")
manager.destroyPipeline()
manager.emitStop <- true
manager.wg.Wait()
}
func (manager *ScreencastManagerCtx) Enabled() bool {

8
internal/capture/stream.go
View File

@ -15,6 +15,7 @@ import (
type StreamManagerCtx struct {
logger zerolog.Logger
mu sync.Mutex
wg sync.WaitGroup
codec codec.RTPCodec
pipelineStr func() string
pipeline *gst.Pipeline
@ -42,8 +43,11 @@ func streamNew(codec codec.RTPCodec, pipelineStr func() string, video_id string)
started: false,
}
manager.wg.Add(1)
go func() {
manager.logger.Debug().Msg("started emitting samples")
defer manager.wg.Done()
for {
select {
@ -66,7 +70,7 @@ func streamNew(codec codec.RTPCodec, pipelineStr func() string, video_id string)
}
func (manager *StreamManagerCtx) shutdown() {
manager.logger.Info().Msgf("shutting down")
manager.logger.Info().Msgf("shutdown")
manager.emitMu.Lock()
for key := range manager.listeners {
@ -75,7 +79,9 @@ func (manager *StreamManagerCtx) shutdown() {
manager.emitMu.Unlock()
manager.destroyPipeline()
manager.emitStop <- true
manager.wg.Wait()
}
func (manager *StreamManagerCtx) Codec() codec.RTPCodec {