使用 Go 语言优雅地处理程序退出时的清理工作

package main

import (
    "log"
    "os"
    "os/signal"
    "time"
)

func main() {
    sigchan := make(chan os.Signal, 1) // 创建一个 buffered channel，避免信号丢失
    signal.Notify(sigchan, os.Interrupt)

    go func() {
        sig := <-sigchan
        log.Println("Received signal:", sig)

        // 在这里执行退出前的清理操作
        log.Println("Performing cleanup tasks...")
        time.Sleep(2 * time.Second) // 模拟清理操作耗时

        log.Println("Cleanup finished. Exiting...")
        os.Exit(0) // 显式退出程序
    }()

    // 主程序逻辑，例如启动 HTTP 服务器
    log.Println("Starting the main program...")
    for i := 0; i < 5; i++ {
        log.Println("Doing some work...", i)
        time.Sleep(1 * time.Second)
    }

    log.Println("Main program finished.")
}

package main

import (
    "context"
    "fmt"
    "log"
    "net/http"
    "os"
    "os/signal"
    "sync"
    "time"
)

func main() {
    // 创建一个 HTTP 服务器
    server := &http.Server{
        Addr:    ":8080",
        Handler: http.HandlerFunc(handler),
    }

    // 创建一个 context，用于控制服务器的生命周期
    ctx, cancel := context.WithCancel(context.Background())

    // 创建一个 WaitGroup，用于等待所有 goroutine 完成
    var wg sync.WaitGroup

    // 启动一个 goroutine 来监听信号
    wg.Add(1)
    go func() {
        defer wg.Done()

        sigchan := make(chan os.Signal, 1)
        signal.Notify(sigchan, os.Interrupt)

        sig := <-sigchan
        log.Println("Received signal:", sig)

        // 关闭服务器
        log.Println("Shutting down the server...")
        shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 5*time.Second)
        defer shutdownCancel()
        if err := server.Shutdown(shutdownCtx); err != nil {
            log.Fatalf("Server shutdown failed: %v", err)
        }
        log.Println("Server shutdown gracefully.")

        // 取消 context，通知其他 goroutine 停止工作
        cancel()
    }()

    // 启动 HTTP 服务器
    wg.Add(1)
    go func() {
        defer wg.Done()
        log.Println("Starting the server...")
        if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
            log.Fatalf("Server listen failed: %v", err)
        }
        log.Println("Server stopped.")
    }()

    // 等待所有 goroutine 完成
    <-ctx.Done() // 等待 context 被取消
    log.Println("Waiting for all goroutines to finish...")
    wg.Wait()
    log.Println("All goroutines finished. Exiting...")
}

func handler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintln(w, "Hello, World!")
}