WaitGroup提供3个方法实现优雅退出
1、Add() :每收到http/mq请求,会在计数器+1
2、Done():每执行完http/mq请求,会在计数器-1
3、Wait():计数器=0,即没有正在处理的 请求
import "sync"
import "os/signal"
import "fmt"
func main() {
var wg sync.WaitGroup
sig := make(chan os.Signal, 1)
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
httpHandle()
go func() {
sig := <-sig
wg.Wait()
done <- true
}()
<-done
fmt.println("quit success")
}
func httpHandle() {
wg.Add()
defer wg.Done()
xxx
}
signal.Notify的参数代表注册要接收的信号:没有信号参数表示接收所有的信号
syscall.SIGINT:接收ctrl+c
syscall.SIGTERM:程序退出
PS:若进程是通过supervisor信号停止,本质执行的是kill -9 xxx,若程序执行时间过长卡在wait仍会被强制退出