Fix default signal

xorkevin · Mar 14, 2023 · 693b6b1 · 693b6b1
1 parent ccfaeb8
commit 693b6b1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cmd.go b/cmd.go
@@ -139,9 +139,6 @@ func (c *Cmd) runInt(f func(ctx context.Context) error) {
 	if conf := c.opts.TermConfig; conf != nil {
 		signals = conf.TermSignals
 	}
-	if len(signals) == 0 {
-		signals = []os.Signal{os.Interrupt}
-	}
 	ksignal.Wait(ctx, signals...)
 
 	cancel()

diff --git a/util/ksignal/ksignal.go b/util/ksignal/ksignal.go
@@ -8,6 +8,9 @@ import (
 
 // Wait blocks until a signal is received
 func Wait(ctx context.Context, signals ...os.Signal) {
+	if len(signals) == 0 {
+		signals = []os.Signal{os.Interrupt}
+	}
 	notifyCtx, stop := signal.NotifyContext(ctx, signals...)
 	defer stop()
 	<-notifyCtx.Done()