Adding ipv6 to useragent (#1930)

Author: Paramadon

extension/agenthealth/handler/useragent/useragent.go

@@ -141,9 +141,15 @@ func (ua *userAgent) SetComponents(otelCfg *otelcol.Config, telegrafCfg *telegra
 		ua.inputs.Add(flagRunAsUser)
 	}
 
+	// Add ipv6 feature flag if dualstack endpoint is enabled
+	if os.Getenv(envconfig.AWS_USE_DUALSTACK_ENDPOINT) == "true" {
+		ua.feature.Add("ipv6")
+	}
+
 	ua.inputsStr.Store(componentsStr(typeInputs, ua.inputs))
 	ua.processorsStr.Store(componentsStr(typeProcessors, ua.processors))
 	ua.outputsStr.Store(componentsStr(typeOutputs, ua.outputs))
+	ua.featureStr.Store(componentsStr(typeFeature, ua.feature))
 	ua.notify()
 }
 

extension/agenthealth/handler/useragent/useragent_test.go

@@ -5,6 +5,7 @@ package useragent
 
 import (
 	"fmt"
+	"os"
 	"sync"
 	"testing"
 
@@ -284,3 +285,46 @@ func TestListen(t *testing.T) {
 	ua.SetContainerInsightsFlag()
 	wg.Wait()
 }
+
+func TestSetComponents_WithDualStackEndpoint(t *testing.T) {
+	// Save original env var
+	originalEnv := os.Getenv(envconfig.AWS_USE_DUALSTACK_ENDPOINT)
+	defer func() {
+		if originalEnv == "" {
+			os.Unsetenv(envconfig.AWS_USE_DUALSTACK_ENDPOINT)
+		} else {
+			os.Setenv(envconfig.AWS_USE_DUALSTACK_ENDPOINT, originalEnv)
+		}
+	}()
+
+	t.Run("ipv6_feature_added_when_dualstack_enabled", func(t *testing.T) {
+		os.Setenv(envconfig.AWS_USE_DUALSTACK_ENDPOINT, "true")
+		ua := newUserAgent()
+		ua.SetComponents(&otelcol.Config{}, &telegraf.Config{})
+
+		assert.Contains(t, ua.feature, "ipv6")
+		assert.Contains(t, ua.featureStr.Load(), "ipv6")
+		header := ua.Header(true)
+		assert.Contains(t, header, "feature:(ipv6)")
+	})
+
+	t.Run("ipv6_feature_not_added_when_dualstack_disabled", func(t *testing.T) {
+		os.Setenv(envconfig.AWS_USE_DUALSTACK_ENDPOINT, "false")
+		ua := newUserAgent()
+		ua.SetComponents(&otelcol.Config{}, &telegraf.Config{})
+
+		assert.NotContains(t, ua.feature, "ipv6")
+		header := ua.Header(true)
+		assert.NotContains(t, header, "ipv6")
+	})
+
+	t.Run("ipv6_feature_not_added_when_dualstack_not_set", func(t *testing.T) {
+		os.Unsetenv(envconfig.AWS_USE_DUALSTACK_ENDPOINT)
+		ua := newUserAgent()
+		ua.SetComponents(&otelcol.Config{}, &telegraf.Config{})
+
+		assert.NotContains(t, ua.feature, "ipv6")
+		header := ua.Header(true)
+		assert.NotContains(t, header, "ipv6")
+	})
+}