golang 优雅退出任务方式模板

package main

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

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel()

    wg := &sync.WaitGroup{}
    wg.Add(1)
    go worker(ctx)

    signalChan := make(chan os.Signal, 1)
    signal.Notify(signalChan, syscall.SIGINT, syscall.SIGTERM)
    <-signalChan

    fmt.Println("Signal received. Shutting down...")

    cancel()
    wg.Wait()

    fmt.Println("Graceful shutdown complete.")
}

func worker(ctx context.Context) {
    defer fmt.Println("Worker shutting down...")

    for {
        select {
        case <-ctx.Done():
            return
        default:
            fmt.Println("Working...")
            time.Sleep(1 * time.Second)
        }
    }
}