Add health check extension for liveness & readiness probes

Author: HussainRaza28

translator/translate/otel/extension/healthcheck.go

@@ -0,0 +1,42 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: MIT
+
+package extension
+
+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_test.go

@@ -0,0 +1,29 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: MIT
+
+package extension
+
+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

@@ -22,6 +22,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"
 	"github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/extension/entitystore"
 	"github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/extension/server"
 	pipelinetranslator "github.com/aws/amazon-cloudwatch-agent/translator/translate/otel/pipeline"
@@ -94,6 +95,8 @@ func Translate(jsonConfig interface{}, os string) (*otelcol.Config, error) {
 		pipelines.Translators.Extensions.Set(server.NewTranslator())
 	}
 
+	pipelines.Translators.Extensions.Set(extension.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)