Add INT and TERM signal handler.

This commit is contained in:
Mariano Cano 2019-04-03 11:07:11 -07:00
parent 47eed2b914
commit 27b6ac0a58

View file

@ -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.