mirror of
https://github.com/livekit/livekit.git
synced 2026-04-18 03:45:40 +00:00
bd42a39117
* include node id in prometheus metrics * static prom init and nodeID * update protocol dep
101 lines
2.8 KiB
Go
101 lines
2.8 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.CounterVec
|
|
promPacketBytes *prometheus.CounterVec
|
|
promNackTotal *prometheus.CounterVec
|
|
promPliTotal *prometheus.CounterVec
|
|
promFirTotal *prometheus.CounterVec
|
|
)
|
|
|
|
func initPacketStats(nodeID string) {
|
|
promPacketTotal = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "packet",
|
|
Name: "total",
|
|
ConstLabels: prometheus.Labels{"node_id": nodeID},
|
|
}, promPacketLabels)
|
|
promPacketBytes = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "packet",
|
|
Name: "bytes",
|
|
ConstLabels: prometheus.Labels{"node_id": nodeID},
|
|
}, promPacketLabels)
|
|
promNackTotal = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "nack",
|
|
Name: "total",
|
|
ConstLabels: prometheus.Labels{"node_id": nodeID},
|
|
}, promPacketLabels)
|
|
promPliTotal = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "pli",
|
|
Name: "total",
|
|
ConstLabels: prometheus.Labels{"node_id": nodeID},
|
|
}, promPacketLabels)
|
|
promFirTotal = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "fir",
|
|
Name: "total",
|
|
ConstLabels: prometheus.Labels{"node_id": nodeID},
|
|
}, promPacketLabels)
|
|
|
|
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 IncrementRTCP(direction Direction, nack, pli, fir int32) {
|
|
if nack > 0 {
|
|
promNackTotal.WithLabelValues(string(direction)).Add(float64(nack))
|
|
atomic.AddUint64(&atomicNackTotal, uint64(nack))
|
|
}
|
|
if pli > 0 {
|
|
promPliTotal.WithLabelValues(string(direction)).Add(float64(pli))
|
|
}
|
|
if fir > 0 {
|
|
promFirTotal.WithLabelValues(string(direction)).Add(float64(fir))
|
|
}
|
|
}
|