diff options
Diffstat (limited to 'cmd/rr/http/workers.go')
| -rw-r--r-- | cmd/rr/http/workers.go | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/cmd/rr/http/workers.go b/cmd/rr/http/workers.go index 394874fa..d04e37fb 100644 --- a/cmd/rr/http/workers.go +++ b/cmd/rr/http/workers.go @@ -36,6 +36,8 @@ import ( "os" "strconv" "time" + "os/signal" + "syscall" ) var ( @@ -59,6 +61,9 @@ func init() { ) rr.CLI.AddCommand(workersCommand) + + signal.Notify(stopSignal, syscall.SIGTERM) + signal.Notify(stopSignal, syscall.SIGINT) } func workersHandler(cmd *cobra.Command, args []string) (err error) { @@ -87,6 +92,8 @@ func workersHandler(cmd *cobra.Command, args []string) (err error) { tm.Clear() for { select { + case <-stopSignal: + return nil case <-time.NewTicker(time.Millisecond * 500).C: tm.MoveCursor(1, 1) showWorkers(client) @@ -94,7 +101,6 @@ func workersHandler(cmd *cobra.Command, args []string) (err error) { } } - <-stopSignal return nil } |