本文讲解如何在 Go 命令行程序（如 HTTP 服务）接收到 Ctrl+C 或系统终止信号时，自动、安全地清理临时生成的 ZIP 文件，解决 defer 在异常终止下不执行的问题。
本文讲解如何在 go 命令行程序（如 http 服务）接收到 ctrl+c 或系统终止信号时，自动、安全地清理临时生成的 zip 文件，解决 `defer` 在异常终止下不执行的问题。
在 Go 中，defer 语句仅在函数正常返回时执行（包括 return、函数末尾自然结束），而通过 Ctrl+C 终止进程会向程序发送 SIGINT 信号，导致操作系统强制终止进程——此时 main 函数未完成、未返回，所有 defer 被跳过，文件无法清理。
要实现“程序退出前必执行清理”，必须主动监听操作系统信号，并在收到终止信号时同步执行清理逻辑。以下是推荐的健壮实现方式：
✅ 正确做法：使用 os/signal 监听终止信号
package main
import (
"fmt"
"log"
"net/http"
"os"
"os/signal"
"syscall"
)
var tempZipPath string // 全局变量保存待清理的 ZIP 路径
func cleanup() {
if tempZipPath != "" {
if err := os.Remove(tempZipPath); err != nil {
log.Printf("Warning: failed to remove %s: %v", tempZipPath, err)
} else {
log.Printf("Cleaned up temporary file: %s", tempZipPath)
}
}
}
func main() {
// 解析命令行参数（保持原有逻辑）
zipped := flag.Bool("z", false, "Zip folder and serve via HTTP (auto-delete on exit)")
save := flag.Bool("s", false, "Keep zipped file after server stops")
flag.Parse()
if len(flag.Args()) == 0 {
fmt.Println("Invalid usage. No file/folder specified. Use -h for help.")
return
}
if *zipped {
fmt.Println("Zipping...")
tempZipPath = ZipFile() // 假设此函数返回 ZIP 文件绝对路径
defer func() {
// 仍保留 defer 作为兜底：覆盖正常 return 场景（如启动失败）
if !*save && tempZipPath != "" {
os.Remove(tempZipPath)
}
}()
}
// ✅ 关键：启动信号监听 goroutine
sigs := make(chan os.Signal, 1)
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
go func() {
<h3>⚠️ 注意事项与最佳实践</h3>
-
不要阻塞
cleanup()：清理操作应轻量（如单个os.Remove）、无网络/IO 等待；若需异步或耗时操作，请设超时并记录警告。 -
defer仍是必要补充：它处理main自然返回（如参数错误提前退出）的场景，而信号处理覆盖强制终止，二者互补。 -
路径必须是绝对路径：
os.Remove对相对路径易出错，建议在ZipFile()中使用filepath.Abs()返回完整路径并赋值给tempZipPath。 -
避免竞态：
tempZipPath是全局变量，但仅由main单线程写入、信号 goroutine 读取，无需额外加锁。 -
极端情况免责：断电、
kill -9（SIGKILL不可捕获）、内核崩溃等仍会导致清理失败——生产环境应在程序启动时扫描并清理残留临时文件（例如检查*.zip.tmp模式）。
通过信号监听 + defer 双保险机制，你的 CLI 工具即可在保持简洁性的同时，实现可靠的资源清理，真正达成“关闭即清理”的用户体验。
