diff options
Diffstat (limited to 'go/weed/signal_handling.go')
| -rw-r--r-- | go/weed/signal_handling.go | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/go/weed/signal_handling.go b/go/weed/signal_handling.go new file mode 100644 index 000000000..5d57c8fdb --- /dev/null +++ b/go/weed/signal_handling.go @@ -0,0 +1,27 @@ +// +build !plan9 + +package main + +import ( + "os" + "os/signal" + "syscall" +) + +func OnInterrupt(fn func()) { + // deal with control+c,etc + signalChan := make(chan os.Signal, 1) + signal.Notify(signalChan, + os.Interrupt, + os.Kill, + syscall.SIGHUP, + syscall.SIGINT, + syscall.SIGTERM, + syscall.SIGQUIT) + go func() { + for _ = range signalChan { + fn() + os.Exit(0) + } + }() +} |
