Go的优雅退出

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

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

 该函数该如何使用？

package mainimport ("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杀掉进程，效果如下：