Go的优雅退出 - 技术栈

Go的优雅退出

奋斗的烧饵块2023-11-15 18:56

Go优雅退出/停机以前主要通过signal来实现，当然现在也是通过signal来实现，只是从go 1.16开始，新增了更加友好的API:

复制代码

func NotifyContext(parent context.Context, signals ...os.Signal) (ctx context.Context, stop context.CancelFunc)

该函数该如何使用？

Go 复制代码

package main

import (
	"context"
	"fmt"
	"os"
	"os/signal"
	"time"
)

func main() {
	// 在这里执行异步的其他代码
	go async()
	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)

	<-ctx.Done()  // 该语句done () 在接收到os.Interrupt时会停止阻塞强制return到下面
	fmt.Println("当前接收到了 os.Interrupt 指令，但是通过了Stop () 来 Reset")
	// 收尾中...
	stop() // 重置 os.Interrupt 信号，允许通过 Ctrl+C 强制退出
    // 执行收尾操作
	time.Sleep(3 * time.Second)
	fmt.Print("收尾成功，程序退出。")
}

func async() {
	for i := 0; i < 100; i++ {
		fmt.Println("我是协程")
		time.Sleep(1 * time.Second)
	}
}

如果使用stop，再次ctrl+c能直接强制退出，因为对应的signal通知已经被重置了：

如果不使用stop，则ctrl+c不能强制退出，假设收尾工作存在死锁等行为，则无法通过Ctrl+C杀掉进程，效果如下：