mirror of
https://github.com/livekit/livekit.git
synced 2026-04-14 10:55:41 +00:00
c3a3fb569d
* add track publisher info in track subscribed event Signed-off-by: shishir gowda <shishir@livekit.io> * update protocol ver Signed-off-by: shishir gowda <shishir@livekit.io>
167 lines
5.9 KiB
Go
167 lines
5.9 KiB
Go
package telemetry
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/livekit/protocol/livekit"
|
|
"github.com/livekit/protocol/webhook"
|
|
)
|
|
|
|
const updateFrequency = time.Second * 10
|
|
|
|
//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 . TelemetryService
|
|
type TelemetryService interface {
|
|
// stats
|
|
TrackStats(streamType livekit.StreamType, participantID livekit.ParticipantID, trackID livekit.TrackID, stat *livekit.AnalyticsStat)
|
|
|
|
// events
|
|
RoomStarted(ctx context.Context, room *livekit.Room)
|
|
RoomEnded(ctx context.Context, room *livekit.Room)
|
|
ParticipantJoined(ctx context.Context, room *livekit.Room, participant *livekit.ParticipantInfo, clientInfo *livekit.ClientInfo, clientMeta *livekit.AnalyticsClientMeta)
|
|
ParticipantLeft(ctx context.Context, room *livekit.Room, participant *livekit.ParticipantInfo)
|
|
TrackPublished(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo)
|
|
TrackUnpublished(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo, ssrc uint32)
|
|
TrackSubscribed(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo, publisher *livekit.ParticipantInfo)
|
|
TrackUnsubscribed(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo)
|
|
TrackPublishedUpdate(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo)
|
|
TrackMaxSubscribedVideoQuality(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo, maxQuality livekit.VideoQuality)
|
|
RecordingStarted(ctx context.Context, ri *livekit.RecordingInfo)
|
|
RecordingEnded(ctx context.Context, ri *livekit.RecordingInfo)
|
|
ParticipantActive(ctx context.Context, participantID livekit.ParticipantID, clientMeta *livekit.AnalyticsClientMeta)
|
|
EgressStarted(ctx context.Context, info *livekit.EgressInfo)
|
|
EgressEnded(ctx context.Context, info *livekit.EgressInfo)
|
|
}
|
|
|
|
type doWorkFunc func()
|
|
|
|
type telemetryService struct {
|
|
internalService TelemetryServiceInternal
|
|
jobQueue chan doWorkFunc
|
|
}
|
|
|
|
const jobQueueBufferSize = 100
|
|
|
|
func NewTelemetryService(notifier webhook.Notifier, analytics AnalyticsService) TelemetryService {
|
|
t := &telemetryService{
|
|
internalService: NewTelemetryServiceInternal(notifier, analytics),
|
|
jobQueue: make(chan doWorkFunc, jobQueueBufferSize),
|
|
}
|
|
|
|
go t.run()
|
|
|
|
return t
|
|
}
|
|
|
|
func (t *telemetryService) run() {
|
|
|
|
ticker := time.NewTicker(updateFrequency)
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
t.internalService.SendAnalytics()
|
|
case job, ok := <-t.jobQueue:
|
|
if ok {
|
|
job()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) TrackStats(streamType livekit.StreamType, participantID livekit.ParticipantID, trackID livekit.TrackID, stats *livekit.AnalyticsStat) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.TrackStats(streamType, participantID, trackID, stats)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) RoomStarted(ctx context.Context, room *livekit.Room) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.RoomStarted(ctx, room)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) RoomEnded(ctx context.Context, room *livekit.Room) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.RoomEnded(ctx, room)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) ParticipantJoined(ctx context.Context, room *livekit.Room, participant *livekit.ParticipantInfo,
|
|
clientInfo *livekit.ClientInfo, clientMeta *livekit.AnalyticsClientMeta) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.ParticipantJoined(ctx, room, participant, clientInfo, clientMeta)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) ParticipantLeft(ctx context.Context, room *livekit.Room, participant *livekit.ParticipantInfo) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.ParticipantLeft(ctx, room, participant)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) TrackPublished(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.TrackPublished(ctx, participantID, track)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) TrackUnpublished(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo, ssrc uint32) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.TrackUnpublished(ctx, participantID, track, ssrc)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) TrackSubscribed(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo, publisher *livekit.ParticipantInfo) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.TrackSubscribed(ctx, participantID, track, publisher)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) TrackUnsubscribed(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.TrackUnsubscribed(ctx, participantID, track)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) RecordingStarted(ctx context.Context, ri *livekit.RecordingInfo) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.RecordingStarted(ctx, ri)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) RecordingEnded(ctx context.Context, ri *livekit.RecordingInfo) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.RecordingEnded(ctx, ri)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) TrackPublishedUpdate(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.TrackPublishedUpdate(ctx, participantID, track)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) ParticipantActive(ctx context.Context, participantID livekit.ParticipantID, clientMeta *livekit.AnalyticsClientMeta) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.ParticipantActive(ctx, participantID, clientMeta)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) TrackMaxSubscribedVideoQuality(ctx context.Context, participantID livekit.ParticipantID, track *livekit.TrackInfo, maxQuality livekit.VideoQuality) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.TrackMaxSubscribedVideoQuality(ctx, participantID, track, maxQuality)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) EgressStarted(ctx context.Context, info *livekit.EgressInfo) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.EgressStarted(ctx, info)
|
|
}
|
|
}
|
|
|
|
func (t *telemetryService) EgressEnded(ctx context.Context, info *livekit.EgressInfo) {
|
|
t.jobQueue <- func() {
|
|
t.internalService.EgressEnded(ctx, info)
|
|
}
|
|
}
|