Go中的Ctrl+C：SIGINT拦截与"defer"清理机制解析

能否在Go中拦截Ctrl+C并像defer一样执行清理？深入解析

你是否曾经启动过一个Go服务器，按下Ctrl+C，然后思考：“如果在退出前能优雅地关闭数据库连接、保存缓存，或者简单地说声‘再见’，那该多好？”这个问题在Stack Overflow上非常热门（成千上万的浏览量！），因此我们决定详细解析它。今天，我们不仅会给出现成的解决方案，还会深入底层——看看Go如何处理信号，signal.Notify是什么，以及为什么defer在这里并不完全相关。

准备好了吗？让我们开始吧！

问题：Ctrl+C会立即终止程序

默认情况下，当你在终端中按下Ctrl+C时，操作系统会向进程发送SIGINT信号。如果Go程序不做任何处理，它会异常终止——所有goroutine都会被中断，defer函数不会执行。这就像拔掉电源插头：计算机关闭了，但数据可能会丢失。

一个经典例子：你正在编写一个Web服务器，打开了与PostgreSQL的连接，突然——Ctrl+C。如果不处理这个信号，连接会挂起，连接池也不会释放。经过几次这样的重启，数据库会说：“抱歉，我撑不住了。”

package main\n\nimport (\n    "fmt"\n    "os"\n    "os/signal"\n    "syscall"\n)\n\nfunc main() {\n    // ... 你的代码\n    fmt.Println("请按Ctrl+C以终止")\n    // 等待信号\n    sig := make(chan os.Signal, 1)\n    signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)\n    <-sig\n    fmt.Println("收到信号，正在退出...")\n}

这段代码已经能够捕获信号，但如果我们需要执行复杂的清理操作呢？这就是魔法开始的地方。

底层工作原理：signal.Notify与通道

signal.Notify函数会将你的通道注册为接收特定信号。当操作系统发送信号时，Go运行时会将其放入通道。如果通道是带缓冲的（我们设置了缓冲区大小为1），即使接收goroutine正忙，信号也不会丢失。

一个重要细节：signal.Notify会拦截信号，但不会阻止其传播。如果你不在通道中处理信号，程序仍然会在执行完defer（如果main函数中有defer）后终止。但如果你想执行清理操作，则需要等待它完成。

工作流程：

\n
• 操作系统 → 向进程发送SIGINT
\n
• Go运行时 → 将信号放入通过signal.Notify注册的通道
\n
• 你的goroutine → 从通道读取并启动清理操作
\n
• 清理完成后 → 调用os.Exit(0)或直接退出main函数
\n

但是，如何实现“defer风格”的清理呢？因为defer是在函数返回时执行的，而不是在收到信号时。答案是：使用defer + 阻塞main函数直到收到信号。

解决方案：Ctrl+C时的“defer风格”清理

思路很简单：我们编写在程序终止时应执行的清理函数，并将它们挂载到defer上。然后阻塞main函数直到收到信号。当信号到达时，main函数结束，所有defer都会执行。很优雅，对吧？

package main\n\nimport (\n    "fmt"\n    "os"\n    "os/signal"\n    "syscall"\n    "time"\n)\n\nfunc cleanup() {\n    fmt.Println("清理中：正在关闭连接...")\n    time.Sleep(2 * time.Second) // 模拟工作\n    fmt.Println("清理完成")\n}\n\nfunc main() {\n    // defer会在main函数任何方式退出时执行\n    defer cleanup()\n\n    // 信号通道\n    sig := make(chan os.Signal, 1)\n    signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)\n\n    fmt.Println("服务器已启动。请按Ctrl+C停止。")")\n    fmt.Println("正在执行清理...")")\n\n    //