Start the grpc server after leader election

gitpod-io · Aug 18, 2023 · 5c7fa0b · 5c7fa0b
1 parent 7f968fa
commit 5c7fa0b
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/components/ws-manager-mk2/main.go b/components/ws-manager-mk2/main.go
@@ -170,13 +170,19 @@ func main() {
 		os.Exit(1)
 	}
 
-	wsmanService, err := setupGRPCService(cfg, mgr.GetClient(), maintenanceReconciler)
-	if err != nil {
-		setupLog.Error(err, "unable to start manager service")
-		os.Exit(1)
-	}
+	// Wait for leader election to start the GRPC server
+	go func() {
+		<-mgr.Elected()
+
+		wsmanService, err := setupGRPCService(cfg, mgr.GetClient(), maintenanceReconciler)
+		if err != nil {
+			setupLog.Error(err, "unable to start manager service")
+			os.Exit(1)
+		}
+
+		workspaceReconciler.OnReconcile = wsmanService.OnWorkspaceReconcile
+	}()
 
-	workspaceReconciler.OnReconcile = wsmanService.OnWorkspaceReconcile
 	if err = workspaceReconciler.SetupWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to setup workspace controller with manager", "controller", "Workspace")
 		os.Exit(1)