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杀掉进程,效果如下: