mirror of
https://github.com/livekit/livekit.git
synced 2026-05-12 14:35:11 +00:00
David Colburn
bf46e998b2
* more buffer stats for analytics * update names * fix jitter and lost rate * don't return on participantLeft if they never published
90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package prometheus
|
|
|
|
import (
|
|
"sync/atomic"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
type Direction string
|
|
|
|
const (
|
|
Incoming Direction = "incoming"
|
|
Outgoing Direction = "outgoing"
|
|
)
|
|
|
|
var (
|
|
atomicBytesIn uint64
|
|
atomicBytesOut uint64
|
|
atomicPacketsIn uint64
|
|
atomicPacketsOut uint64
|
|
atomicNackTotal uint64
|
|
|
|
promPacketLabels = []string{"direction"}
|
|
|
|
promPacketTotal = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "packet",
|
|
Name: "total",
|
|
}, promPacketLabels)
|
|
promPacketBytes = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "packet",
|
|
Name: "bytes",
|
|
}, promPacketLabels)
|
|
promNackTotal = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "nack",
|
|
Name: "total",
|
|
}, promPacketLabels)
|
|
promPliTotal = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "pli",
|
|
Name: "total",
|
|
}, promPacketLabels)
|
|
promFirTotal = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "fir",
|
|
Name: "total",
|
|
}, promPacketLabels)
|
|
)
|
|
|
|
func initPacketStats() {
|
|
prometheus.MustRegister(promPacketTotal)
|
|
prometheus.MustRegister(promPacketBytes)
|
|
prometheus.MustRegister(promNackTotal)
|
|
prometheus.MustRegister(promPliTotal)
|
|
prometheus.MustRegister(promFirTotal)
|
|
}
|
|
|
|
func IncrementPackets(direction Direction, count uint64) {
|
|
promPacketTotal.WithLabelValues(string(direction)).Add(float64(count))
|
|
if direction == Incoming {
|
|
atomic.AddUint64(&atomicPacketsIn, count)
|
|
} else {
|
|
atomic.AddUint64(&atomicPacketsOut, count)
|
|
}
|
|
}
|
|
|
|
func IncrementBytes(direction Direction, count uint64) {
|
|
promPacketBytes.WithLabelValues(string(direction)).Add(float64(count))
|
|
if direction == Incoming {
|
|
atomic.AddUint64(&atomicBytesIn, count)
|
|
} else {
|
|
atomic.AddUint64(&atomicBytesOut, count)
|
|
}
|
|
}
|
|
|
|
func IncrementNack(direction Direction) {
|
|
promNackTotal.WithLabelValues(string(direction)).Add(1)
|
|
atomic.AddUint64(&atomicNackTotal, 1)
|
|
}
|
|
|
|
func IncrementPLI(direction Direction) {
|
|
promPliTotal.WithLabelValues(string(direction)).Add(1)
|
|
}
|
|
|
|
func IncrementFIR(direction Direction) {
|
|
promFirTotal.WithLabelValues(string(direction)).Add(1)
|
|
}
|