Reuse SetupSignalHandler from k8s.io/apiserver

Signed-off-by: Richard Wall <[email protected]>
kubernetes-sigs · Aug 30, 2019 · 0848b1d · 0848b1d
1 parent 22b0482
commit 0848b1d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/cmd/apiserver-boot/boot/util/util.go b/cmd/apiserver-boot/boot/util/util.go
@@ -28,9 +28,9 @@ import (
 	"strings"
 	"text/template"
 
-	"os/signal"
-
 	"github.com/markbates/inflect"
+
+	"sigs.k8s.io/apiserver-builder-alpha/cmd/apiserver-boot/boot/util/signal"
 )
 
 var Domain string
@@ -147,8 +147,7 @@ func CancelWhenSignaled(parent context.Context) context.Context {
 	ctx, cancel := context.WithCancel(parent)
 
 	go func() {
-		signalChannel := make(chan os.Signal)
-		signal.Notify(signalChannel, os.Interrupt, os.Kill)
+		signalChannel := signal.SetupSignalHandler()
 		<-signalChannel
 		cancel()
 	}()