better handling of os signals
This commit is contained in:
27
go/weed/signal_handling.go
Normal file
27
go/weed/signal_handling.go
Normal file
@@ -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)
|
||||
}
|
||||
}()
|
||||
}
|
||||
Reference in New Issue
Block a user