add lock

Signed-off-by: Ryan Leung <rleungx@gmail.com>
rleungx · Sep 26, 2023 · 0ea0068 · 0ea0068
1 parent a63ecc2
commit 0ea0068
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/schedule/schedulers/scheduler_controller.go b/pkg/schedule/schedulers/scheduler_controller.go
@@ -67,6 +67,8 @@ func NewController(ctx context.Context, cluster sche.SchedulerCluster, storage e
 
 // Wait waits on all schedulers to exit.
 func (c *Controller) Wait() {
+	c.Lock()
+	defer c.Unlock()
 	c.wg.Wait()
 }