Merge pull request #11 from kubescape/liveness

adding liveness probe

cmd/client/main.go

@@ -55,9 +55,9 @@ func main() {
 	// synchronizer server URL using service discovery
 	if services, err := config.LoadServiceURLs("/etc/config/services.json"); err != nil {
 		logger.L().Warning("failed discovering urls", helpers.Error(err))
-	} else {
-		logger.L().Debug("synchronizer server URL", helpers.String("synchronizer", services.GetSynchronizerUrl()))
-		cfg.InCluster.ServerUrl = services.GetSynchronizerUrl()
+	} else if serverUrl := services.GetSynchronizerUrl(); serverUrl != "" {
+		logger.L().Debug("synchronizer server URL", helpers.String("synchronizer", serverUrl))
+		cfg.InCluster.ServerUrl = serverUrl
 	}
 
 	cfg.InCluster.ValidateConfig()
@@ -89,6 +89,9 @@ func main() {
 		}),
 	}
 
+	// start liveness probe
+	utils.StartLivenessProbe()
+
 	for {
 		if err := start(ctx, cfg.InCluster, adapter, dialer); err != nil {
 			d := 5 * time.Second // TODO: use exponential backoff for retries

cmd/server/main.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/gobwas/ws"
 	pulsarconnector "github.com/kubescape/messaging/pulsar/connector"
+	"github.com/kubescape/synchronizer/utils"
 
 	"github.com/kubescape/go-logger"
 	"github.com/kubescape/go-logger/helpers"
@@ -63,6 +64,9 @@ func main() {
 		adapter = adapters.NewMockAdapter(false)
 	}
 
+	// start liveness probe
+	utils.StartLivenessProbe()
+
 	// websocket server
 	_ = http.ListenAndServe(":8080",
 		authentication.AuthenticationServerMiddleware(cfg.Backend.AuthenticationServer,

utils/utils.go

@@ -7,6 +7,7 @@ import (
 	"errors"
 	"flag"
 	"fmt"
+	"net/http"
 	"path/filepath"
 	"reflect"
 	"strconv"
@@ -186,3 +187,14 @@ func PulsarMessageIDtoString(msgID pulsar.MessageID) string {
 	msgIDstr := msgID.String() + ":" + batchStr
 	return msgIDstr
 }
+
+func StartLivenessProbe() {
+	http.HandleFunc("/healthz", func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+	})
+	go func() {
+		if err := http.ListenAndServe(":7888", nil); err != nil {
+			logger.L().Error("failed to start liveness probe", helpers.Error(err))
+		}
+	}()
+}