-
Notifications
You must be signed in to change notification settings - Fork 0
/
signal.go
34 lines (27 loc) · 609 Bytes
/
signal.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
package easyshutdown
import (
"os"
"os/signal"
"syscall"
)
var shutdownSignals = []os.Signal{
os.Interrupt,
syscall.SIGTERM,
syscall.SIGINT,
}
var initOnce = make(chan struct{})
// signals subscribes to shutdownSignals and
// returns a closed stop channel when any of those signals are received.
func signals() <-chan struct{} {
close(initOnce)
stop := make(chan struct{})
signalChan := make(chan os.Signal, 2)
signal.Notify(signalChan, shutdownSignals...)
go func() {
<-signalChan
close(stop)
<-signalChan
os.Exit(1) // Two exit signals in a row? Exit directly.
}()
return stop
}