36 lines
535 B
Go
36 lines
535 B
Go
|
// Copyright 2022 Listware
|
||
|
|
||
|
package agent
|
||
|
|
||
|
import (
|
||
|
"os"
|
||
|
"os/signal"
|
||
|
"syscall"
|
||
|
)
|
||
|
|
||
|
func (a *Agent) osSignalCtrl() {
|
||
|
sigChan := make(chan os.Signal, 1)
|
||
|
signal.Notify(sigChan,
|
||
|
syscall.SIGINT,
|
||
|
syscall.SIGTERM,
|
||
|
syscall.SIGQUIT,
|
||
|
syscall.SIGHUP,
|
||
|
syscall.SIGUSR1,
|
||
|
syscall.SIGUSR2,
|
||
|
)
|
||
|
go func() {
|
||
|
for {
|
||
|
select {
|
||
|
case sig := <-sigChan:
|
||
|
switch sig {
|
||
|
case syscall.SIGTERM, syscall.SIGQUIT, syscall.SIGINT:
|
||
|
log.Infof("Get Stop signal")
|
||
|
a.cancel()
|
||
|
}
|
||
|
case <-a.ctx.Done():
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
}()
|
||
|
}
|