Alessandro RosandGitHub
986e270862
Discarded frames, decode errors and lost packets were logged individually, then there was a mechanism that prevented more than 1 log entry per second from being printed, resulting in inaccurate reports. Now discarded frames, decode errors and lost packets are accurately counted, and their count is printed once every second.
97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
// Package test contains test utilities.
|
|
package test
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/bluenviron/mediamtx/internal/conf"
|
|
"github.com/bluenviron/mediamtx/internal/defs"
|
|
"github.com/bluenviron/mediamtx/internal/logger"
|
|
"github.com/bluenviron/mediamtx/internal/stream"
|
|
"github.com/bluenviron/mediamtx/internal/unit"
|
|
)
|
|
|
|
// SourceTester is a static source tester.
|
|
type SourceTester struct {
|
|
ctx context.Context
|
|
ctxCancel func()
|
|
stream *stream.Stream
|
|
reader stream.Reader
|
|
|
|
Unit chan unit.Unit
|
|
done chan struct{}
|
|
}
|
|
|
|
// NewSourceTester allocates a SourceTester.
|
|
func NewSourceTester(
|
|
createFunc func(defs.StaticSourceParent) defs.StaticSource,
|
|
resolvedSource string,
|
|
conf *conf.Path,
|
|
) *SourceTester {
|
|
ctx, ctxCancel := context.WithCancel(context.Background())
|
|
|
|
t := &SourceTester{
|
|
ctx: ctx,
|
|
ctxCancel: ctxCancel,
|
|
Unit: make(chan unit.Unit),
|
|
done: make(chan struct{}),
|
|
}
|
|
|
|
s := createFunc(t)
|
|
|
|
go func() {
|
|
s.Run(defs.StaticSourceRunParams{ //nolint:errcheck
|
|
Context: ctx,
|
|
ResolvedSource: resolvedSource,
|
|
Conf: conf,
|
|
})
|
|
close(t.done)
|
|
}()
|
|
|
|
return t
|
|
}
|
|
|
|
// Close closes the tester.
|
|
func (t *SourceTester) Close() {
|
|
t.ctxCancel()
|
|
t.stream.RemoveReader(t.reader)
|
|
<-t.done
|
|
}
|
|
|
|
// Log implements StaticSourceParent.
|
|
func (t *SourceTester) Log(_ logger.Level, _ string, _ ...interface{}) {
|
|
}
|
|
|
|
// SetReady implements StaticSourceParent.
|
|
func (t *SourceTester) SetReady(req defs.PathSourceStaticSetReadyReq) defs.PathSourceStaticSetReadyRes {
|
|
t.stream = &stream.Stream{
|
|
WriteQueueSize: 512,
|
|
UDPMaxPayloadSize: 1472,
|
|
Desc: req.Desc,
|
|
GenerateRTPPackets: req.GenerateRTPPackets,
|
|
Parent: t,
|
|
}
|
|
err := t.stream.Initialize()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
t.reader = NilLogger
|
|
|
|
t.stream.AddReader(t.reader, req.Desc.Medias[0], req.Desc.Medias[0].Formats[0], func(u unit.Unit) error {
|
|
t.Unit <- u
|
|
close(t.Unit)
|
|
return nil
|
|
})
|
|
|
|
t.stream.StartReader(t.reader)
|
|
|
|
return defs.PathSourceStaticSetReadyRes{
|
|
Stream: t.stream,
|
|
}
|
|
}
|
|
|
|
// SetNotReady implements StaticSourceParent.
|
|
func (t *SourceTester) SetNotReady(_ defs.PathSourceStaticSetNotReadyReq) {
|
|
}
|