feat(conntrack-metrics): add conntrack_total_connections metric and f…

…ix comments
microsoft · Jan 21, 2025 · 3fa4154 · 3fa4154
1 parent 191464d
commit 3fa4154
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 13 deletions.
diff --git a/pkg/metrics/metrics.go b/pkg/metrics/metrics.go
@@ -182,6 +182,12 @@ func InitializeMetrics() {
 		ConntrackBytesReplyDescription,
 	)
 
+	ConntrackTotalConnections = exporter.CreatePrometheusGaugeVecForMetric(
+		exporter.DefaultRegistry,
+		utils.ConntrackTotalConnectionsName,
+		ConntrackTotalConnectionsDescription,
+	)
+
 	isInitialized = true
 	metricsLogger.Info("Metrics initialized")
 }

diff --git a/pkg/metrics/types.go b/pkg/metrics/types.go
@@ -45,10 +45,11 @@ const (
 	lostEventsCounterDescription                     = "Number of events lost in control plane"
 
 	// Conntrack metrics
-	ConntrackPacketForwardDescription = "Number of forward packets"
-	ConntrackPacketReplyDescription   = "Number of reply packets"
-	ConntrackBytesForwardDescription  = "Number of forward bytes"
-	ConntrackBytesReplyDescription    = "Number of reply bytes"
+	ConntrackPacketForwardDescription    = "Number of forward packets"
+	ConntrackPacketReplyDescription      = "Number of reply packets"
+	ConntrackBytesForwardDescription     = "Number of forward bytes"
+	ConntrackBytesReplyDescription       = "Number of reply bytes"
+	ConntrackTotalConnectionsDescription = "Total number of connections"
 )
 
 // Metric Counters
@@ -97,10 +98,11 @@ var (
 	InfinibandStatusParamsGauge GaugeVec
 
 	// Conntrack
-	ConntrackPacketsForward GaugeVec
-	ConntrackPacketsReply   GaugeVec
-	ConntrackBytesForward   GaugeVec
-	ConntrackBytesReply     GaugeVec
+	ConntrackPacketsForward   GaugeVec
+	ConntrackPacketsReply     GaugeVec
+	ConntrackBytesForward     GaugeVec
+	ConntrackBytesReply       GaugeVec
+	ConntrackTotalConnections GaugeVec
 )
 
 func ToPrometheusType(metric interface{}) prometheus.Collector {

diff --git a/pkg/plugin/conntrack/conntrack_linux.go b/pkg/plugin/conntrack/conntrack_linux.go
@@ -24,7 +24,7 @@ import (
 	"go.uber.org/zap"
 )
 
-var conntrackMetricsEnabled = false // global variable to enable conntrack metrics
+var conntrackMetricsEnabled = false // conntrack metrics global variable
 
 //go:generate go run github.com/cilium/ebpf/cmd/bpf2go@master -cflags "-g -O2 -Wall -D__TARGET_ARCH_${GOARCH} -Wall" -target ${GOARCH} -type ct_v4_key conntrack ./_cprog/conntrack.c -- -I../lib/_${GOARCH} -I../lib/common/libbpf/_src -I../lib/common/libbpf/_include/linux -I../lib/common/libbpf/_include/uapi/linux -I../lib/common/libbpf/_include/asm
 
@@ -91,7 +91,7 @@ func GenerateDynamic(ctx context.Context, dynamicHeaderPath string, conntrackMet
 	if err != nil {
 		return errors.Wrap(err, "failed to write conntrack dynamic header")
 	}
-	// set a global variable to enable conntrack metrics
+	// set a global variable
 	if conntrackMetrics == 1 {
 		conntrackMetricsEnabled = true
 	}
@@ -126,8 +126,8 @@ func (ct *Conntrack) Run(ctx context.Context) error {
 			var keysToDelete []conntrackCtV4Key
 
 			// metrics counters
-			var packetsCountForward, packetsCountReply uint32
-			var bytesCountForward, bytesCountReply uint64
+			var packetsCountForward, packetsCountReply, totConnections uint32 = 0, 0, 0
+			var bytesCountForward, bytesCountReply uint64 = 0, 0
 
 			iter := ct.ctMap.Iterate()
 			for iter.Next(&key, &value) {
@@ -148,8 +148,8 @@ func (ct *Conntrack) Run(ctx context.Context) error {
 				// Add conntrack metrics.
 				if conntrackMetricsEnabled {
 					// Basic metrics, node-level
-					// for each ct_entry increment counters
 					ctMeta := value.ConntrackMetadata
+					totConnections++
 					packetsCountForward += ctMeta.PacketsForwardCount
 					packetsCountReply += ctMeta.PacketsReplyCount
 					bytesCountForward += ctMeta.BytesForwardCount
@@ -181,6 +181,7 @@ func (ct *Conntrack) Run(ctx context.Context) error {
 				metrics.ConntrackBytesForward.WithLabelValues().Set(float64(bytesCountForward))
 				metrics.ConntrackPacketsReply.WithLabelValues().Set(float64(packetsCountReply))
 				metrics.ConntrackBytesReply.WithLabelValues().Set(float64(bytesCountReply))
+				metrics.ConntrackTotalConnections.WithLabelValues().Set(float64(totConnections))
 			}
 
 			// Delete the conntrack entries

diff --git a/pkg/utils/metric_names.go b/pkg/utils/metric_names.go
@@ -40,6 +40,7 @@ const (
 	ConntrackPacketsReplyGaugeName   = "conntrack_packets_reply"
 	ConntrackBytesForwardGaugeName   = "conntrack_bytes_forward"
 	ConntrackBytesReplyGaugeName     = "conntrack_bytes_reply"
+	ConntrackTotalConnectionsName    = "conntrack_total_connections"
 )
 
 // IsAdvancedMetric is a helper function to determine if a name is an advanced metric