Merge pull request volcano-sh#2927 from rfyiamcool/fix/leaderelection…

…_signal_ctx

fix: set leaderelection's ctx that can sense signal in controller

cmd/controller-manager/app/server.go

@@ -39,6 +39,7 @@ import (
 	"volcano.sh/volcano/cmd/controller-manager/app/options"
 	"volcano.sh/volcano/pkg/controllers/framework"
 	"volcano.sh/volcano/pkg/kube"
+	"volcano.sh/volcano/pkg/signals"
 )
 
 const (
@@ -62,8 +63,10 @@ func Run(opt *options.ServerOption) error {
 
 	run := startControllers(config, opt)
 
+	ctx := signals.SetupSignalContext()
+
 	if !opt.EnableLeaderElection {
-		run(context.TODO())
+		run(ctx)
 		return fmt.Errorf("finished without leader elect")
 	}
 
@@ -97,7 +100,7 @@ func Run(opt *options.ServerOption) error {
 		return fmt.Errorf("couldn't create resource lock: %v", err)
 	}
 
-	leaderelection.RunOrDie(context.TODO(), leaderelection.LeaderElectionConfig{
+	leaderelection.RunOrDie(ctx, leaderelection.LeaderElectionConfig{
 		Lock:          rl,
 		LeaseDuration: leaseDuration,
 		RenewDeadline: renewDeadline,