Go语言程序终止时的清理策略：defer、信号处理与外部包装器

package main

import (
    "fmt"
    "os"
)

func main() {
    fmt.Println("程序开始运行...")

    // 示例1: 文件操作清理
    file, err := os.Create("example.txt")
    if err != nil {
        fmt.Println("创建文件失败:", err)
        return
    }
    // 使用 defer 确保文件在函数退出时关闭
    defer func() {
        fmt.Println("关闭文件: example.txt")
        file.Close()
    }()
    fmt.Fprintf(file, "Hello, Go defer!")

    // 示例2: 数据库连接清理
    // 假设这里有一个数据库连接对象 db
    // db := ConnectToDatabase() // 实际应用中会连接数据库
    // defer func() {
    //     fmt.Println("关闭数据库连接")
    //     // db.Close() // 调用实际的关闭方法
    // }()
    // fmt.Println("数据库操作进行中...")

    fmt.Println("程序主逻辑执行完毕。")
    // 当 main 函数退出时，defer 的函数会被执行
}

package main

import (
    "fmt"
    "os"
    "os/signal"
    "syscall"
    "time"
)

func main() {
    fmt.Println("程序启动，等待信号...")

    // 创建一个通道用于接收操作系统信号
    sigChan := make(chan os.Signal, 1)
    // 注册要捕获的信号：SIGINT (Ctrl+C), SIGTERM (kill 命令默认发送)
    signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)

    // 启动一个协程来处理信号
    go func() {
        sig := <-sigChan // 阻塞直到接收到信号
        fmt.Printf("\n接收到信号: %v，开始执行清理操作...\n", sig)

        // 在这里执行程序级的清理逻辑
        // 例如：关闭所有数据库连接、保存未完成的数据、刷新日志等
        time.Sleep(2 * time.Second) // 模拟清理耗时
        fmt.Println("清理操作完成，程序即将退出。")
        os.Exit(0) // 优雅退出
    }()

    // 程序主逻辑持续运行
    for i := 0; i < 5; i++ { // 简化循环次数以便演示
        fmt.Printf("程序运行中... %d\n", i)
        time.Sleep(1 * time.Second)
    }

    fmt.Println("主逻辑执行完毕，等待信号处理或自动退出。")
    // 如果主逻辑提前结束，但信号处理协程还在等待，程序会一直运行
    // 此时需要一种机制来协调，例如使用 context.WithCancel
    select {} // 阻塞主goroutine，直到信号处理协程调用 os.Exit(0)
}

#!/bin/bash

APP_LOG="/var/log/my_go_app.log"
CLEANUP_SCRIPT="/usr/local/bin/cleanup_resources.sh"

echo "Starting Go application..." | tee -a "$APP_LOG"
./my_go_app >> "$APP_LOG" 2>&1
APP_EXIT_CODE=$?

echo "Go application exited with code: $APP_EXIT_CODE" | tee -a "$APP_LOG"

echo "Executing cleanup script..." | tee -a "$APP_LOG"
# 传递Go应用的退出码给清理脚本
"$CLEANUP_SCRIPT" "$APP_EXIT_CODE" >> "$APP_LOG" 2>&1

echo "Cleanup finished." | tee -a "$APP_LOG"
exit "$APP_EXIT_CODE"

#!/bin/bash

# $1 是 Go 应用程序的退出码
APP_EXIT_CODE=$1

echo "Performing global cleanup based on exit code: $APP_EXIT_CODE"
# 例如：
# 1. 检查特定文件是否存在并删除
# 2. 清理临时目录
# 3. 发送告警通知
# 4. 关闭外部服务连接（如果它们是独立于Go应用生命周期的）
# 5. 确保某些外部资源（如云存储桶中的临时文件）被删除