forked from TrueCloudLab/certificates
Add INT and TERM signal handler.
This commit is contained in:
parent
47eed2b914
commit
27b6ac0a58
1 changed files with 32 additions and 0 deletions
32
ca/signal.go
32
ca/signal.go
|
@ -7,6 +7,12 @@ import (
|
||||||
"syscall"
|
"syscall"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Stopper is the interface that external commands can implement to stop the
|
||||||
|
// server.
|
||||||
|
type Stopper interface {
|
||||||
|
Stop() error
|
||||||
|
}
|
||||||
|
|
||||||
// StopReloader is the interface that external commands can implement to stop
|
// StopReloader is the interface that external commands can implement to stop
|
||||||
// the server and reload the configuration while running.
|
// the server and reload the configuration while running.
|
||||||
type StopReloader interface {
|
type StopReloader interface {
|
||||||
|
@ -14,6 +20,32 @@ type StopReloader interface {
|
||||||
Reload() error
|
Reload() error
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// StopHandler watches SIGINT, SIGTERM on a list of servers implementing the
|
||||||
|
// Stopper interface, and when one of those signals is caught we'll run Stop
|
||||||
|
// (SIGINT, SIGTERM) on all servers.
|
||||||
|
func StopHandler(servers ...Stopper) {
|
||||||
|
signals := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
|
||||||
|
defer signal.Stop(signals)
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case sig := <-signals:
|
||||||
|
switch sig {
|
||||||
|
case syscall.SIGINT, syscall.SIGTERM:
|
||||||
|
log.Println("shutting down ...")
|
||||||
|
for _, server := range servers {
|
||||||
|
err := server.Stop()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("error stopping server: %s", err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// StopReloaderHandler watches SIGINT, SIGTERM and SIGHUP on a list of servers
|
// StopReloaderHandler watches SIGINT, SIGTERM and SIGHUP on a list of servers
|
||||||
// implementing the StopReloader interface, and when one of those signals is
|
// implementing the StopReloader interface, and when one of those signals is
|
||||||
// caught we'll run Stop (SIGINT, SIGTERM) or Reload (SIGHUP) on all servers.
|
// caught we'll run Stop (SIGINT, SIGTERM) or Reload (SIGHUP) on all servers.
|
||||||
|
|
Loading…
Reference in a new issue