81 lines
2 KiB
Go
81 lines
2 KiB
Go
package ca
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"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
|
|
// the server and reload the configuration while running.
|
|
type StopReloader interface {
|
|
Stop() 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
|
|
// implementing the StopReloader interface, and when one of those signals is
|
|
// caught we'll run Stop (SIGINT, SIGTERM) or Reload (SIGHUP) on all servers.
|
|
func StopReloaderHandler(servers ...StopReloader) {
|
|
signals := make(chan os.Signal, 1)
|
|
signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP)
|
|
defer signal.Stop(signals)
|
|
|
|
for {
|
|
select {
|
|
case sig := <-signals:
|
|
switch sig {
|
|
case syscall.SIGHUP:
|
|
log.Println("reloading ...")
|
|
for _, server := range servers {
|
|
err := server.Reload()
|
|
if err != nil {
|
|
log.Printf("error reloading server: %+v", err)
|
|
}
|
|
}
|
|
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
|
|
}
|
|
}
|
|
}
|
|
}
|