Merge pull request #14 from InjectiveLabs/f/default-tags

feat: default tags

Author: hmoragrega

client.go

@@ -41,6 +41,7 @@ type StatterConfig struct {
 	EnvName              string        // dev/test/staging/prod
 	HostName             string        // hostname
 	Version              string        // version
+	DefaultTags          []interface{} // default tags for all metrics
 	StuckFunctionTimeout time.Duration // stuck time
 	MockingThreshold     time.Duration // mocking threshold
 	MockingEnabled       bool          // whether to enable mock statter, which only produce logs
@@ -52,6 +53,7 @@ type StatterConfig struct {
 }
 
 func (m *StatterConfig) BaseTags() []string {
+	defaultTags := Combine(m.DefaultTags...)
 	var baseTags []string
 
 	switch m.Agent {
@@ -63,6 +65,9 @@ func (m *StatterConfig) BaseTags() []string {
 		if len(config.HostName) > 0 {
 			baseTags = append(baseTags, "machine:"+config.HostName)
 		}
+		for k, v := range defaultTags {
+			baseTags = append(baseTags, k+":"+v)
+		}
 	// telegraf by default
 	default:
 		if len(config.EnvName) > 0 {
@@ -71,6 +76,9 @@ func (m *StatterConfig) BaseTags() []string {
 		if len(config.HostName) > 0 {
 			baseTags = append(baseTags, "machine", config.HostName)
 		}
+		for k, v := range defaultTags {
+			baseTags = append(baseTags, k, v)
+		}
 	}
 
 	return baseTags

client_test.go

@@ -0,0 +1,81 @@
+package metrics
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestBaseTags(t *testing.T) {
+	tests := []struct {
+		name     string
+		config   *StatterConfig
+		expected []string
+	}{
+		{
+			name: "Datadog agent with default tags",
+			config: &StatterConfig{
+				Agent:    DatadogAgent,
+				EnvName:  "prod",
+				HostName: "host1",
+				DefaultTags: []interface{}{
+					"key1", "value1",
+					"key2", "value2",
+				},
+			},
+			expected: []string{"env:prod", "machine:host1", "key1:value1", "key2:value2"},
+		},
+		{
+			name: "Telegraf agent with default tags",
+			config: &StatterConfig{
+				Agent:    TelegrafAgent,
+				EnvName:  "dev",
+				HostName: "host2",
+				DefaultTags: []interface{}{
+					"key1", "value1",
+					"key2", "value2",
+				},
+			},
+			expected: []string{"env", "dev", "machine", "host2", "key1", "value1", "key2", "value2"},
+		},
+		{
+			name: "Datadog agent with no default tags",
+			config: &StatterConfig{
+				Agent:       DatadogAgent,
+				EnvName:     "staging",
+				HostName:    "host3",
+				DefaultTags: nil,
+			},
+			expected: []string{"env:staging", "machine:host3"},
+		},
+		{
+			name: "Telegraf agent with no default tags",
+			config: &StatterConfig{
+				Agent:       TelegrafAgent,
+				EnvName:     "test",
+				HostName:    "host4",
+				DefaultTags: nil,
+			},
+			expected: []string{"env", "test", "machine", "host4"},
+		},
+		{
+			name: "Empty config",
+			config: &StatterConfig{
+				Agent:       DatadogAgent,
+				DefaultTags: nil,
+			},
+			expected: nil,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			t.Cleanup(func() {
+				config = nil
+			})
+			config = tt.config
+			result := tt.config.BaseTags()
+			assert.ElementsMatch(t, tt.expected, result)
+		})
+	}
+}