如何在主进程中断时让Goroutine优雅退出？附代码优化疑问

1. done通道是否有必要？ #

有必要。原代码里main监听ctx.Done()只是收到终止信号，不代表work协程已经完成所有清理操作并退出。done通道的作用就是让main等待work协程真正执行完收尾逻辑（比如示例里的fmt.Println("work finished")）后再退出，避免main提前结束导致协程未完成的工作被强行中断。

如果去掉done通道，main在<-ctx.Done()之后会直接执行收尾并退出，此时work协程可能还在处理退出前的步骤，最终这些步骤可能无法完整执行。

2. 仅通过context能否实现Goroutine的优雅终止？ #

不能。context的核心作用是传递终止信号，通知协程需要停止工作，但它无法保证协程完成所有退出前的操作。要实现真正的"优雅终止"，必须搭配同步机制（比如done通道、sync.WaitGroup）让主协程等待所有工作协程完成清理后再退出。

比如只靠context的话，main触发cancel()后直接退出，工作协程收到ctx.Done()信号后的收尾代码可能还没执行完，程序就已经结束了，达不到优雅终止的效果。

3. 多个Worker Goroutine的处理方式 #

当存在多个工作协程时，用sync.WaitGroup会比多个done通道更简洁，示例代码如下：

package main

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

func work(ctx context.Context, wg *sync.WaitGroup, workerID int) {
    defer wg.Done()
    fmt.Printf("worker %d starting its loop\n", workerID)
    out:
    for {
        select {
        case <-ctx.Done():
            fmt.Printf("ctx.Done() inside worker %d\n", workerID)
            break out
        default:
            time.Sleep(time.Second)
            fmt.Printf("worker %d heartbeat\n", workerID)
        }
    }
    fmt.Printf("worker %d finished\n", workerID)
}

func main() {
    interrupt := make(chan os.Signal, 1)
    signal.Notify(interrupt, syscall.SIGINT, syscall.SIGTERM)
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel()

    var wg sync.WaitGroup
    workerCount := 3
    wg.Add(workerCount)
    for i := 0; i < workerCount; i++ {
        go work(ctx, &wg, i)
    }

    // 等待中断信号
    <-interrupt
    cancel()
    fmt.Println("cancel triggered, waiting for workers to finish")
    // 等待所有worker完成
    wg.Wait()
    fmt.Println("Main finished")
}

这段代码里，sync.WaitGroup负责追踪所有工作协程的状态：启动前调用wg.Add()标记需要等待的协程数，每个协程退出前调用wg.Done()标记自身完成，main通过wg.Wait()等待所有协程完成收尾工作后再退出，实现多协程的优雅终止。

内容的提问来源于stack exchange，提问作者Greendrake