Go的优雅退出

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