mirror of
https://github.com/livekit/livekit.git
synced 2026-04-01 00:05:40 +00:00
faa870de3d
* move callbacks out of messageRouter * OCD * more OCD * fix forwarder test * even more OCD * maximum OCD * package name collision, copy lock by value
47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
package telemetry
|
|
|
|
import (
|
|
"github.com/livekit/protocol/livekit"
|
|
"github.com/pion/interceptor"
|
|
"github.com/pion/rtcp"
|
|
)
|
|
|
|
func (t *telemetryService) NewStatsInterceptorFactory(participantID, identity string) *StatsInterceptorFactory {
|
|
return &StatsInterceptorFactory{
|
|
t: t,
|
|
participantID: participantID,
|
|
identity: identity,
|
|
}
|
|
}
|
|
|
|
type StatsInterceptorFactory struct {
|
|
t TelemetryService
|
|
participantID string
|
|
identity string
|
|
}
|
|
|
|
func (f *StatsInterceptorFactory) NewInterceptor(_ string) (interceptor.Interceptor, error) {
|
|
return &StatsInterceptor{
|
|
t: f.t,
|
|
participantID: f.participantID,
|
|
identity: f.identity,
|
|
}, nil
|
|
}
|
|
|
|
type StatsInterceptor struct {
|
|
interceptor.NoOp
|
|
|
|
t TelemetryService
|
|
participantID string
|
|
identity string
|
|
}
|
|
|
|
// BindRTCPWriter lets you modify any outgoing RTCP packets. It is called once per PeerConnection. The returned method
|
|
// will be called once per packet batch.
|
|
func (s *StatsInterceptor) BindRTCPWriter(writer interceptor.RTCPWriter) interceptor.RTCPWriter {
|
|
return interceptor.RTCPWriterFunc(func(pkts []rtcp.Packet, attributes interceptor.Attributes) (int, error) {
|
|
s.t.HandleRTCP(livekit.StreamType_UPSTREAM, s.participantID, pkts)
|
|
return writer.Write(pkts, attributes)
|
|
})
|
|
}
|