traefik/cmd/context.go

23 lines
425 B
Go
Raw Normal View History

2018-03-14 12:14:03 +00:00
package cmd
import (
"context"
"os"
"os/signal"
"syscall"
)
2018-08-06 18:00:03 +00:00
// ContextWithSignal creates a context canceled when SIGINT or SIGTERM are notified
2018-03-14 12:14:03 +00:00
func ContextWithSignal(ctx context.Context) context.Context {
newCtx, cancel := context.WithCancel(ctx)
signals := make(chan os.Signal)
signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
go func() {
select {
case <-signals:
cancel()
}
}()
return newCtx
}