mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
6067367acd
* Handle bitrate change by finding the stream with closest bitrate as peer * Convert video id into bitrate when creating peer or changing bitrate * Try to fix prometheus panic * Revert metrics label name change * minor fixes. * bitrate selector. * skip if moving to the same stream. * no closure for getting target bitrate. * fix: high res switch to lo video, stream bitrate out of range * revert dev config change. * white space. Co-authored-by: Aleksandar Sukovic <aleksandar.sukovic@gmail.com>
106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package capture
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"math"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"github.com/demodesk/neko/pkg/types"
|
|
"github.com/demodesk/neko/pkg/types/codec"
|
|
)
|
|
|
|
type BucketsManagerCtx struct {
|
|
logger zerolog.Logger
|
|
codec codec.RTPCodec
|
|
streams map[string]*StreamSinkManagerCtx
|
|
streamIDs []string
|
|
}
|
|
|
|
func bucketsNew(codec codec.RTPCodec, streams map[string]*StreamSinkManagerCtx, streamIDs []string) *BucketsManagerCtx {
|
|
logger := log.With().
|
|
Str("module", "capture").
|
|
Str("submodule", "buckets").
|
|
Logger()
|
|
|
|
return &BucketsManagerCtx{
|
|
logger: logger,
|
|
codec: codec,
|
|
streams: streams,
|
|
streamIDs: streamIDs,
|
|
}
|
|
}
|
|
|
|
func (m *BucketsManagerCtx) shutdown() {
|
|
m.logger.Info().Msgf("shutdown")
|
|
}
|
|
|
|
func (m *BucketsManagerCtx) destroyAll() {
|
|
for _, stream := range m.streams {
|
|
if stream.Started() {
|
|
stream.destroyPipeline()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (m *BucketsManagerCtx) recreateAll() error {
|
|
for _, stream := range m.streams {
|
|
if stream.Started() {
|
|
err := stream.createPipeline()
|
|
if err != nil && !errors.Is(err, types.ErrCapturePipelineAlreadyExists) {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *BucketsManagerCtx) IDs() []string {
|
|
return m.streamIDs
|
|
}
|
|
|
|
func (m *BucketsManagerCtx) Codec() codec.RTPCodec {
|
|
return m.codec
|
|
}
|
|
|
|
func (m *BucketsManagerCtx) SetReceiver(receiver types.Receiver) error {
|
|
receiver.OnBitrateChange(func(bitrate int) error {
|
|
stream, ok := m.findNearestStream(bitrate)
|
|
if !ok {
|
|
return fmt.Errorf("no stream found for bitrate %d", bitrate)
|
|
}
|
|
|
|
return receiver.SetStream(stream)
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *BucketsManagerCtx) findNearestStream(bitrate int) (ss *StreamSinkManagerCtx, ok bool) {
|
|
minDiff := math.MaxInt
|
|
for _, s := range m.streams {
|
|
streamBitrate, err := s.Bitrate()
|
|
if err != nil {
|
|
m.logger.Error().Err(err).Msgf("failed to get bitrate for stream %s", s.ID())
|
|
continue
|
|
}
|
|
|
|
diffAbs := int(math.Abs(float64(bitrate - streamBitrate)))
|
|
|
|
if diffAbs < minDiff {
|
|
minDiff, ss = diffAbs, s
|
|
}
|
|
}
|
|
ok = ss != nil
|
|
return
|
|
}
|
|
|
|
func (m *BucketsManagerCtx) RemoveReceiver(receiver types.Receiver) error {
|
|
receiver.OnBitrateChange(nil)
|
|
receiver.RemoveStream()
|
|
return nil
|
|
}
|