Go优雅退出/停机以前主要通过signal来实现,当然现在也是通过signal来实现,只是从go 1.16开始,新增了更加友好的API:
func NotifyContext(parent context.Context, signals ...os.Signal) (ctx context.Context, stop context.CancelFunc)
该函数该如何使用?
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杀掉进程,效果如下: