WaitGroup优雅退出

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仍会被强制退出

来源：CSDN

作者：小青桔（1212）

链接：https://blog.csdn.net/ambzheng/article/details/104067469