mirror of
https://github.com/livekit/livekit.git
synced 2026-04-01 04:25:39 +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
94 lines
2.9 KiB
Go
94 lines
2.9 KiB
Go
package prometheus
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/livekit/protocol/livekit"
|
|
"github.com/livekit/protocol/utils"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
const livekitNamespace string = "livekit"
|
|
|
|
var (
|
|
MessageCounter *prometheus.CounterVec
|
|
ServiceOperationCounter *prometheus.CounterVec
|
|
)
|
|
|
|
func init() {
|
|
nodeID, _ := utils.LocalNodeID()
|
|
MessageCounter = prometheus.NewCounterVec(
|
|
prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "node",
|
|
Name: "messages",
|
|
ConstLabels: prometheus.Labels{"node_id": nodeID},
|
|
},
|
|
[]string{"type", "status"},
|
|
)
|
|
|
|
ServiceOperationCounter = prometheus.NewCounterVec(
|
|
prometheus.CounterOpts{
|
|
Namespace: livekitNamespace,
|
|
Subsystem: "node",
|
|
Name: "service_operation",
|
|
ConstLabels: prometheus.Labels{"node_id": nodeID},
|
|
},
|
|
[]string{"type", "status", "error_type"},
|
|
)
|
|
|
|
prometheus.MustRegister(MessageCounter)
|
|
prometheus.MustRegister(ServiceOperationCounter)
|
|
|
|
initPacketStats(nodeID)
|
|
initRoomStats(nodeID)
|
|
}
|
|
|
|
func UpdateCurrentNodeStats(nodeStats *livekit.NodeStats) error {
|
|
updatedAtPrevious := nodeStats.UpdatedAt
|
|
nodeStats.UpdatedAt = time.Now().Unix()
|
|
secondsSinceLastUpdate := nodeStats.UpdatedAt - updatedAtPrevious
|
|
|
|
err := updateCurrentNodeSystemStats(nodeStats)
|
|
updateCurrentNodeRoomStats(nodeStats)
|
|
updateCurrentNodePacketStats(nodeStats, secondsSinceLastUpdate)
|
|
|
|
return err
|
|
}
|
|
|
|
func updateCurrentNodeRoomStats(nodeStats *livekit.NodeStats) {
|
|
nodeStats.NumClients = atomic.LoadInt32(&atomicParticipantTotal)
|
|
nodeStats.NumRooms = atomic.LoadInt32(&atomicRoomTotal)
|
|
nodeStats.NumTracksIn = atomic.LoadInt32(&atomicTrackPublishedTotal)
|
|
nodeStats.NumTracksOut = atomic.LoadInt32(&atomicTrackSubscribedTotal)
|
|
}
|
|
|
|
func updateCurrentNodePacketStats(nodeStats *livekit.NodeStats, secondsSinceLastUpdate int64) {
|
|
if secondsSinceLastUpdate == 0 {
|
|
return
|
|
}
|
|
|
|
bytesInPrevious := nodeStats.BytesIn
|
|
bytesOutPrevious := nodeStats.BytesOut
|
|
packetsInPrevious := nodeStats.PacketsIn
|
|
packetsOutPrevious := nodeStats.PacketsOut
|
|
nackTotalPrevious := nodeStats.NackTotal
|
|
|
|
nodeStats.BytesIn = atomic.LoadUint64(&atomicBytesIn)
|
|
nodeStats.BytesOut = atomic.LoadUint64(&atomicBytesOut)
|
|
nodeStats.PacketsIn = atomic.LoadUint64(&atomicPacketsIn)
|
|
nodeStats.PacketsOut = atomic.LoadUint64(&atomicPacketsOut)
|
|
nodeStats.NackTotal = atomic.LoadUint64(&atomicNackTotal)
|
|
|
|
nodeStats.BytesInPerSec = perSec(bytesInPrevious, nodeStats.BytesIn, secondsSinceLastUpdate)
|
|
nodeStats.BytesOutPerSec = perSec(bytesOutPrevious, nodeStats.BytesOut, secondsSinceLastUpdate)
|
|
nodeStats.PacketsInPerSec = perSec(packetsInPrevious, nodeStats.PacketsIn, secondsSinceLastUpdate)
|
|
nodeStats.PacketsOutPerSec = perSec(packetsOutPrevious, nodeStats.PacketsOut, secondsSinceLastUpdate)
|
|
nodeStats.NackPerSec = perSec(nackTotalPrevious, nodeStats.NackTotal, secondsSinceLastUpdate)
|
|
}
|
|
|
|
func perSec(prev, curr uint64, secs int64) float32 {
|
|
return float32(curr-prev) / float32(secs)
|
|
}
|