add cmd_code tag to metrics

MATRIXXSoftware · Dec 11, 2023 · 4e5b9c2 · 4e5b9c2
1 parent 85970fd
commit 4e5b9c2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/diameter.go b/diameter.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"net"
 	"os"
+	"strconv"
 	"time"
 
 	"github.com/fiorix/go-diameter/v4/diam"
@@ -182,9 +183,12 @@ func (c *DiameterClient) Send(msg *DiameterMessage) (*DiameterMessage, error) {
 	// Wait for Response
 	select {
 	case resp := <-c.hopIds[hopByHopID]:
+		tags := map[string]string{
+			"cmd_code": strconv.FormatUint(uint64(msg.diamMsg.Header.CommandCode), 10),
+		}
 		now := time.Now()
-		c.reportMetric(c.metrics.RequestDuration, time.Now(), metrics.D(now.Sub(sentAt)))
-		c.reportMetric(c.metrics.RequestCount, time.Now(), 1)
+		c.reportMetric(c.metrics.RequestDuration, time.Now(), metrics.D(now.Sub(sentAt)), tags)
+		c.reportMetric(c.metrics.RequestCount, time.Now(), 1, tags)
 
 		delete(c.hopIds, hopByHopID)
 

diff --git a/metrics.go b/metrics.go
@@ -21,7 +21,7 @@ func registerMetrics(vu modules.VU) DiameterMetrics {
 	return metrics
 }
 
-func (c *DiameterClient) reportMetric(metric *metrics.Metric, now time.Time, value float64) {
+func (c *DiameterClient) reportMetric(metric *metrics.Metric, now time.Time, value float64, tags map[string]string) {
 	state := c.vu.State()
 	ctx := c.vu.Context()
 	if state == nil || ctx == nil {
@@ -32,6 +32,7 @@ func (c *DiameterClient) reportMetric(metric *metrics.Metric, now time.Time, val
 		Time: now,
 		TimeSeries: metrics.TimeSeries{
 			Metric: metric,
+			Tags:   metrics.NewRegistry().RootTagSet().WithTagsFromMap(tags),
 		},
 		Value: value,
 	})