6

Author: aledbf

components/ws-manager-mk2/main.go

@@ -10,6 +10,7 @@ import (
 	"flag"
 	"fmt"
 	"net"
+	"net/http"
 	"os"
 
 	// Import all Kubernetes client auth plugins (e.g. Azure, GCP, OIDC, etc.)
@@ -135,6 +136,8 @@ func main() {
 		os.Exit(1)
 	}
 
+	var grpcServerStarted bool
+
 	go func() {
 		<-mgr.Elected()
 
@@ -165,6 +168,8 @@ func main() {
 			os.Exit(1)
 		}
 
+		grpcServerStarted = true
+
 		workspaceReconciler.OnReconcile = wsmanService.OnWorkspaceReconcile
 
 		if err = workspaceReconciler.SetupWithManager(mgr); err != nil {
@@ -199,6 +204,17 @@ func main() {
 		os.Exit(1)
 	}
 
+	if err := mgr.AddHealthzCheck("grpc-server", func(req *http.Request) error {
+		if grpcServerStarted {
+			return nil
+		}
+
+		return fmt.Errorf("ws-manager GRPC server not started")
+	}); err != nil {
+		setupLog.Error(err, "unable to set up ready check")
+		os.Exit(1)
+	}
+
 	setupLog.Info("starting manager")
 	if err := mgr.Start(ctrl.SetupSignalHandler()); err != nil {
 		setupLog.Error(err, "problem running manager")