Add health check extension for liveness & readiness probes

translator/tocwconfig/tocwconfig_test.go

@@ -934,6 +934,37 @@ func verifyToYamlTranslation(t *testing.T, input interface{}, expectedYamlFilePa
 		yamlStr := toyamlconfig.ToYamlConfig(yamlConfig)
 		require.NoError(t, yaml.Unmarshal([]byte(yamlStr), &actual))
 
+		// Remove health_check/health_check extension from both expected and actual for comparison
+		// This allows tests to pass when the health check extension is dynamically added
+		removeHealthCheckExtension := func(data interface{}) interface{} {
+			if dataMap, ok := data.(map[string]interface{}); ok {
+				if extensions, exists := dataMap["extensions"]; exists {
+					if extMap, ok := extensions.(map[string]interface{}); ok {
+						delete(extMap, "health_check/health_check")
+					}
+				}
+				if service, exists := dataMap["service"]; exists {
+					if serviceMap, ok := service.(map[string]interface{}); ok {
+						if serviceExtensions, exists := serviceMap["extensions"]; exists {
+							if extSlice, ok := serviceExtensions.([]interface{}); ok {
+								var filteredExtensions []interface{}
+								for _, ext := range extSlice {
+									if extStr, ok := ext.(string); ok && extStr != "health_check/health_check" {
+										filteredExtensions = append(filteredExtensions, ext)
+									}
+								}
+								serviceMap["extensions"] = filteredExtensions
+							}
+						}
+					}
+				}
+			}
+			return data
+		}
+
+		expected = removeHealthCheckExtension(expected)
+		actual = removeHealthCheckExtension(actual)
+
 		//assert.NoError(t, os.WriteFile(expectedYamlFilePath, []byte(yamlStr), 0644)) // useful for regenerating YAML
 		opt := cmpopts.SortSlices(func(x, y interface{}) bool {
 			return pretty.Sprint(x) < pretty.Sprint(y)

translator/translate/otel/extension/healthcheck/translator.go

@@ -0,0 +1,42 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: MIT
+
+package healthcheck
+
+import (
+	"sync"
+
+	"go.opentelemetry.io/collector/component"
+	"go.opentelemetry.io/collector/confmap"
+
+	"github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/common"
+)
+
+type healthCheckTranslator struct {
+	name string
+	mux  sync.RWMutex
+}
+
+var _ common.Translator[component.Config, component.ID] = (*healthCheckTranslator)(nil)
+
+func NewHealthCheckTranslator() common.Translator[component.Config, component.ID] {
+	return &healthCheckTranslator{name: "health_check"}
+}
+
+func (t *healthCheckTranslator) ID() component.ID {
+	t.mux.RLock()
+	defer t.mux.RUnlock()
+	return component.NewIDWithName(component.MustNewType("health_check"), t.name)
+}
+
+func (t *healthCheckTranslator) Translate(_ *confmap.Conf) (component.Config, error) {
+	cfg := &struct {
+		Endpoint string `mapstructure:"endpoint"`
+		Path     string `mapstructure:"path"`
+	}{
+		Endpoint: "0.0.0.0:13133",
+		Path:     "/",
+	}
+
+	return cfg, nil
+}

translator/translate/otel/extension/healthcheck/translator_test.go

@@ -0,0 +1,29 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: MIT
+
+package healthcheck
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"go.opentelemetry.io/collector/confmap"
+)
+
+func TestHealthCheckTranslator(t *testing.T) {
+	translator := NewHealthCheckTranslator()
+	assert.Equal(t, "health_check", translator.ID().Type().String())
+
+	conf := confmap.New()
+	cfg, err := translator.Translate(conf)
+	assert.NoError(t, err)
+
+	// Assert the config has the expected fields
+	healthCheckCfg, ok := cfg.(*struct {
+		Endpoint string `mapstructure:"endpoint"`
+		Path     string `mapstructure:"path"`
+	})
+	assert.True(t, ok)
+	assert.Equal(t, "0.0.0.0:13133", healthCheckCfg.Endpoint)
+	assert.Equal(t, "/", healthCheckCfg.Path)
+}

translator/translate/otel/translate_otel.go

@@ -23,6 +23,7 @@ import (
 	"github.com/aws/amazon-cloudwatch-agent/translator/context"
 	"github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/common"
 	"github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/extension/entitystore"
+	healthcheckextension "github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/extension/healthcheck"
 	"github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/extension/server"
 	pipelinetranslator "github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/pipeline"
 	"github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/pipeline/applicationsignals"
@@ -94,6 +95,8 @@ func Translate(jsonConfig interface{}, os string) (*otelcol.Config, error) {
 		pipelines.Translators.Extensions.Set(server.NewTranslator())
 	}
 
+	pipelines.Translators.Extensions.Set(healthcheckextension.NewHealthCheckTranslator())
+
 	cfg := &otelcol.Config{
 		Receivers:  map[component.ID]component.Config{},
 		Exporters:  map[component.ID]component.Config{},

translator/translate/otel/translate_otel_test.go

@@ -20,6 +20,31 @@ import (
 	"github.com/aws/amazon-cloudwatch-agent/translator/util/eksdetector"
 )
 
+func TestHealthCheckExtension(t *testing.T) {
+	agent.Global_Config.Region = "us-east-1"
+	input := map[string]interface{}{
+		"metrics": map[string]interface{}{
+			"metrics_collected": map[string]interface{}{
+				"cpu": map[string]interface{}{},
+			},
+		},
+	}
+
+	cfg, err := Translate(input, "linux")
+	require.NoError(t, err)
+	require.NotNil(t, cfg)
+
+	// Verify that the health check extension is registered
+	extensionFound := false
+	for _, ext := range cfg.Service.Extensions {
+		if ext.Type().String() == "health_check" {
+			extensionFound = true
+			break
+		}
+	}
+	assert.True(t, extensionFound, "Health check extension should be registered")
+}
+
 func TestTranslator(t *testing.T) {
 	agent.Global_Config.Region = "us-east-1"
 	testutil.SetPrometheusRemoteWriteTestingEnv(t)