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

发布了171 篇原创文章 · 获赞 72 · 访问量 8851

猜你喜欢

转载自blog.csdn.net/ambzheng/article/details/104067469