Golang程序的优雅退出

我们知道，在使用Linux操作系统运行程序时，我们经常会使用 “Kill -9” 和 “Ctrl + C” 的形式退出程序。

所以，我们可以在程序中监听来自系统的退出信号，当主函数接收到退出信号时，再退出程序。像这样：

func main() {
	//接收退出信号的chanel
	sig := make(chan os.Signal)
	//指定哪些信号可以转发到chanel，如果没有列出，会将所有信号传递到chanel
	signal.Notify(sig, syscall.SIGINT, syscall.SIGKILL)

	fmt.Printf("接收到的信号: %v \n", <-sig)
	fmt.Println("主goroutine结束")
}

现在我们编译并运行程序，并通过按下 “Ctrl + C” 退出程序

 

这里监听到了 中断信号interrupt，我们成功接收到系统发送来的退出信号。但是现实中我们的程序中往往不只有一个主Goroutine，还有很多其他的子Goroutine运行着，那么我们就应该先通知子Goroutine结束任务，再结束整个程序的运行，像这样：

func main() {
	sig := make(chan os.Signal)
	signal.Notify(sig, syscall.SIGINT, syscall.SIGKILL)

	//通过context通知子goroutine结束
	ctx, cancel := context.WithCancel(context.Background())
	//使用wg确认所有子goroutine已经结束
	wg := sync.WaitGroup{}

	wg.Add(10)
	for i := 0; i < 10; i++ {
		go func(ctx context.Context, wg *sync.WaitGroup, i int) {
			defer wg.Done()

			//do something

			for {
				select {
				case <-ctx.Done():
					fmt.Printf("goroutine：%v 结束\n", i)
					return
				}
			}
		}(ctx, &wg, i)
	}

	fmt.Printf("接收到的信号: %v \n", <-sig)
	cancel()

	wg.Wait()
	fmt.Println("主goroutine结束")
}

现在编译运行，并通过按下 “Ctrl + C” 退出程序