add health probes

Signed-off-by: Matthias Bertschy <[email protected]>

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"node-agent/pkg/healthmanager"
 	"strings"
 
 	"net/http"
@@ -240,6 +241,10 @@ func main() {
 	// Start the prometheusExporter
 	prometheusExporter.Start()
 
+	// Start the health manager
+	healthManager := healthmanager.NewHealthManager(mainHandler)
+	healthManager.Start(ctx)
+
 	// Start the container handler
 	err = mainHandler.Start(ctx)
 	if err != nil {

pkg/containerwatcher/container_watcher_interface.go

@@ -5,6 +5,7 @@ import (
 )
 
 type ContainerWatcher interface {
+	Ready() bool
 	Start(ctx context.Context) error
 	Stop()
 }

pkg/containerwatcher/container_watcher_mock.go

@@ -6,6 +6,10 @@ import (
 
 type ContainerWatcherMock struct{}
 
+func (c ContainerWatcherMock) Ready() bool {
+	return true
+}
+
 func (c ContainerWatcherMock) Start(_ context.Context) error {
 	return nil
 }

pkg/containerwatcher/v1/container_watcher.go

@@ -332,3 +332,7 @@ func (ch *IGContainerWatcher) Stop() {
 		ch.running = false
 	}
 }
+
+func (ch *IGContainerWatcher) Ready() bool {
+	return ch.running
+}

pkg/healthmanager/health_manager.go

@@ -0,0 +1,52 @@
+package healthmanager
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"node-agent/pkg/containerwatcher/v1"
+	"time"
+
+	"github.com/kubescape/go-logger"
+	"github.com/kubescape/go-logger/helpers"
+)
+
+type HealthManager struct {
+	containerWatcher *containerwatcher.IGContainerWatcher
+	port             int
+}
+
+func NewHealthManager(containerWatcher *containerwatcher.IGContainerWatcher) *HealthManager {
+	return &HealthManager{
+		containerWatcher: containerWatcher,
+		port:             8080,
+	}
+}
+
+func (h *HealthManager) Start(ctx context.Context) {
+	go func() {
+		http.HandleFunc("/livez", h.livenessProbe)
+		http.HandleFunc("/readyz", h.readinessProbe)
+		srv := &http.Server{
+			Addr:         fmt.Sprintf(":%d", h.port),
+			WriteTimeout: 15 * time.Second,
+			ReadTimeout:  15 * time.Second,
+		}
+		logger.L().Info("starting health manager", helpers.Int("port", h.port))
+		if err := srv.ListenAndServe(); err != nil {
+			logger.L().Ctx(ctx).Fatal("failed to start health manager", helpers.Error(err), helpers.Int("port", h.port))
+		}
+	}()
+}
+
+func (h *HealthManager) livenessProbe(w http.ResponseWriter, _ *http.Request) {
+	w.WriteHeader(http.StatusOK)
+}
+
+func (h *HealthManager) readinessProbe(w http.ResponseWriter, _ *http.Request) {
+	if h.containerWatcher.Ready() {
+		w.WriteHeader(http.StatusOK)
+	} else {
+		w.WriteHeader(http.StatusInternalServerError)
+	}
+}