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 webrtc metrtics.

Browse Source
This commit is contained in:
Miroslav Šedivý
2022-06-25 20:12:42 +02:00
parent 555fd803bc
commit 527b1f08f8
2 changed files with 142 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
internal/webrtc/manager.go
View File

@ -36,8 +36,9 @@ const rtcpPLIInterval = 3 * time.Second
func New(desktop types.DesktopManager, capture types.CaptureManager, config *config.WebRTC) *WebRTCManagerCtx {
return &WebRTCManagerCtx{
logger: log.With().Str("module", "webrtc").Logger(),
config: config,
logger: log.With().Str("module", "webrtc").Logger(),
config: config,
metrics: newMetrics(),
desktop: desktop,
capture: capture,
@ -47,9 +48,10 @@ func New(desktop types.DesktopManager, capture types.CaptureManager, config *con
}
type WebRTCManagerCtx struct {
logger zerolog.Logger
config *config.WebRTC
peerId int32
logger zerolog.Logger
config *config.WebRTC
metrics *metricsCtx
peerId int32
desktop types.DesktopManager
capture types.CaptureManager
@ -121,6 +123,7 @@ func (manager *WebRTCManagerCtx) ICEServers() []types.ICEServer {
func (manager *WebRTCManagerCtx) CreatePeer(session types.Session, videoID string) (*webrtc.SessionDescription, error) {
id := atomic.AddInt32(&manager.peerId, 1)
manager.metrics.NewConnection(session)
// add session id to logger context
logger := manager.logger.With().Str("session_id", session.ID()).Int32("peer_id", id).Logger()
@ -315,6 +318,8 @@ func (manager *WebRTCManagerCtx) CreatePeer(session types.Session, videoID strin
videoTrack.RemoveStream()
audioTrack.RemoveStream()
}
manager.metrics.SetState(session, state)
})
cursorImage := func(entry *cursor.ImageEntry) {
@ -392,5 +397,22 @@ func (manager *WebRTCManagerCtx) CreatePeer(session types.Session, videoID strin
})
})
go func() {
ticker := time.NewTicker(5 * time.Second)
defer ticker.Stop()
for range ticker.C {
if connection.ConnectionState() == webrtc.PeerConnectionStateClosed {
break
}
stats := connection.GetStats()
data, ok := stats["iceTransport"].(webrtc.TransportStats)
if ok {
manager.metrics.SetTransportStats(session, data)
}
}
}()
return offer, nil
}