mirror of
https://github.com/hibiken/asynq.git
synced 2024-09-20 11:05:58 +08:00
31 lines
637 B
Go
31 lines
637 B
Go
|
// +build linux bsd darwin
|
||
|
|
||
|
package asynq
|
||
|
|
||
|
import (
|
||
|
"os"
|
||
|
"os/signal"
|
||
|
|
||
|
"golang.org/x/sys/unix"
|
||
|
|
||
|
"github.com/hibiken/asynq/internal/base"
|
||
|
)
|
||
|
|
||
|
// waitForSignals waits for signals and handle them.
|
||
|
// 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.
|
||
|
func (bg *Background) waitForSignals() {
|
||
|
sigs := make(chan os.Signal, 1)
|
||
|
signal.Notify(sigs, unix.SIGTERM, unix.SIGINT, unix.SIGTSTP)
|
||
|
for {
|
||
|
sig := <-sigs
|
||
|
if sig == unix.SIGTSTP {
|
||
|
bg.processor.stop()
|
||
|
bg.ps.SetStatus(base.StatusStopped)
|
||
|
continue
|
||
|
}
|
||
|
break
|
||
|
}
|
||
|
}
|