refactor: wait for server before starting controllers

loft-sh · Jan 6, 2022 · 2d5b9f3 · 2d5b9f3
1 parent 00e7b1b
commit 2d5b9f3
Showing 1 changed file with 17 additions and 15 deletions.
diff --git a/cmd/vcluster/cmd/start.go b/cmd/vcluster/cmd/start.go
@@ -260,36 +260,38 @@ func ExecuteStart(options *context2.VirtualClusterOptions) error {
 		return errors.Wrap(err, "create controller context")
 	}
 
-	// start leader election for controllers
-	rawConfig, err := clientConfig.RawConfig()
+	// start the proxy
+	proxyServer, err := server.NewServer(ctx, options.RequestHeaderCaCert, options.ClientCaCert)
 	if err != nil {
 		return err
 	}
+
+	// start the proxy server in secure mode
 	go func() {
-		if ctx.Options.LeaderElect {
-			err = leaderelection.StartLeaderElection(ctx, scheme, func() error {
-				return startControllers(ctx, &rawConfig, serverVersion)
-			})
-		} else {
-			err = startControllers(ctx, &rawConfig, serverVersion)
-		}
+		err = proxyServer.ServeOnListenerTLS(options.BindAddress, options.Port, ctx.StopChan)
 		if err != nil {
-			klog.Fatalf("Error starting leader election: %v", err)
+			klog.Fatalf("Error serving: %v", err)
 		}
 	}()
 
-	// start the proxy
-	proxyServer, err := server.NewServer(ctx, options.RequestHeaderCaCert, options.ClientCaCert)
+	// start leader election for controllers
+	rawConfig, err := clientConfig.RawConfig()
 	if err != nil {
 		return err
 	}
 
-	// start the proxy server in secure mode
-	err = proxyServer.ServeOnListenerTLS(options.BindAddress, options.Port, ctx.StopChan)
+	if ctx.Options.LeaderElect {
+		err = leaderelection.StartLeaderElection(ctx, scheme, func() error {
+			return startControllers(ctx, &rawConfig, serverVersion)
+		})
+	} else {
+		err = startControllers(ctx, &rawConfig, serverVersion)
+	}
 	if err != nil {
-		return err
+		return errors.Wrap(err, "start controllers")
 	}
 
+	<-ctx.StopChan
 	return nil
 }