2020-04-06 03:14:42 +08:00
|
|
|
// +build linux bsd darwin
|
|
|
|
|
|
|
|
package asynq
|
|
|
|
|
|
|
|
import (
|
|
|
|
"os"
|
|
|
|
"os/signal"
|
|
|
|
|
|
|
|
"golang.org/x/sys/unix"
|
|
|
|
)
|
|
|
|
|
2020-04-06 05:56:06 +08:00
|
|
|
// waitForSignals waits for signals and handles them.
|
2020-04-06 03:14:42 +08:00
|
|
|
// It handles SIGTERM, SIGINT, and SIGTSTP.
|
|
|
|
// SIGTERM and SIGINT will signal the process to exit.
|
|
|
|
// SIGTSTP will signal the process to stop processing new tasks.
|
2020-04-12 23:16:42 +08:00
|
|
|
func (srv *Server) waitForSignals() {
|
|
|
|
srv.logger.Info("Send signal TSTP to stop processing new tasks")
|
|
|
|
srv.logger.Info("Send signal TERM or INT to terminate the process")
|
2020-04-10 23:47:43 +08:00
|
|
|
|
2020-04-06 03:14:42 +08:00
|
|
|
sigs := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(sigs, unix.SIGTERM, unix.SIGINT, unix.SIGTSTP)
|
|
|
|
for {
|
|
|
|
sig := <-sigs
|
|
|
|
if sig == unix.SIGTSTP {
|
2020-04-13 02:00:45 +08:00
|
|
|
srv.Quiet()
|
2020-04-06 03:14:42 +08:00
|
|
|
continue
|
|
|
|
}
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|