From 38470f378bb0cac735f3a47bea17647ba6a79e73 Mon Sep 17 00:00:00 2001
From: Paul Wells <paulwe@gmail.com>
Date: Sun, 26 May 2024 14:01:13 -0700
Subject: [PATCH] add message bytes metric (#2731)

---
 pkg/telemetry/prometheus/node.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/pkg/telemetry/prometheus/node.go b/pkg/telemetry/prometheus/node.go
index fe186e1dd..514220ca4 100644
--- a/pkg/telemetry/prometheus/node.go
+++ b/pkg/telemetry/prometheus/node.go
@@ -35,6 +35,7 @@ var (
 	initialized atomic.Bool
 
 	MessageCounter            *prometheus.CounterVec
+	MessageBytes              *prometheus.CounterVec
 	ServiceOperationCounter   *prometheus.CounterVec
 	TwirpRequestStatusCounter *prometheus.CounterVec
 
@@ -61,6 +62,16 @@ func Init(nodeID string, nodeType livekit.NodeType) error {
 		[]string{"type", "status"},
 	)
 
+	MessageBytes = prometheus.NewCounterVec(
+		prometheus.CounterOpts{
+			Namespace:   livekitNamespace,
+			Subsystem:   "node",
+			Name:        "message_bytes",
+			ConstLabels: prometheus.Labels{"node_id": nodeID, "node_type": nodeType.String()},
+		},
+		[]string{"type", "message_type"},
+	)
+
 	ServiceOperationCounter = prometheus.NewCounterVec(
 		prometheus.CounterOpts{
 			Namespace:   livekitNamespace,
@@ -103,6 +114,7 @@ func Init(nodeID string, nodeType livekit.NodeType) error {
 	)
 
 	prometheus.MustRegister(MessageCounter)
+	prometheus.MustRegister(MessageBytes)
 	prometheus.MustRegister(ServiceOperationCounter)
 	prometheus.MustRegister(TwirpRequestStatusCounter)
 	prometheus.MustRegister(promSysPacketGauge)