2024-03-14 23:26:33 +08:00
|
|
|
//go:build linux || dragonfly || freebsd || netbsd || openbsd || darwin
|
2020-04-06 03:14:42 +08:00
|
|
|
|
|
|
|
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 {
|
2021-03-23 21:20:54 +08:00
|
|
|
srv.Stop()
|
2020-04-06 03:14:42 +08:00
|
|
|
continue
|
2024-12-09 15:08:36 +08:00
|
|
|
} else {
|
|
|
|
srv.Stop()
|
|
|
|
break
|
2020-04-06 03:14:42 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-09-27 08:33:29 +08:00
|
|
|
|
|
|
|
func (s *Scheduler) waitForSignals() {
|
|
|
|
s.logger.Info("Send signal TERM or INT to stop the scheduler")
|
|
|
|
sigs := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(sigs, unix.SIGTERM, unix.SIGINT)
|
|
|
|
<-sigs
|
|
|
|
}
|