如何通过通道快速关闭Go协程？1000+长运行协程停止优化

你的方案是否正确？ #

你通过启动独立goroutine发送停止信号的方式是可行且正确的。

原代码的问题 #

原stopTest函数中，c <- true是无缓冲通道的阻塞发送操作：只有当对应的goroutine回到select语句接收信号时，发送操作才会完成。如果goroutine正卡在callClientTask()（最长30秒），manager的发送操作就会一直阻塞，导致1000个goroutine的停止信号要逐个等待，耗时极长。

你的改进思路的合理性 #

启动单独goroutine发送信号后，每个发送操作的阻塞都在独立的goroutine中，不会阻碍manager遍历所有通道。所有停止信号几乎同时发出，一旦某个goroutine完成当前callClientTask()迭代，回到select就会立刻接收信号并退出，确保所有goroutine都能在单次任务迭代（最多30秒）内停止。

不过你在发送后调用close(c)不是必须的——每个通道仅用于发送一次停止信号，关闭通道不会带来额外收益，但也不会引发问题。

更高效的优化方案 #

维护1000个独立的停止通道会浪费资源，Go提供了更简洁的广播式停止方案：

方案1：使用Context（推荐） #

Context是Go标准库中用于传递取消信号的惯用方式，支持一次性通知所有关联的goroutine：

import (
    "context"
    "log"
    "time"
)

func myThreadFunc(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            log.Debug("Stopping thread")
            return
        default:
            callClientTask()
        }
    }
}

func callClientTask() {
    // 模拟耗时外部API调用
    time.Sleep(5 * time.Second)
}

func manager() {
    ctx, cancel := context.WithCancel(context.Background())
    // 启动1000个goroutine
    for i := 0; i < 1000; i++ {
        go myThreadFunc(ctx)
    }

    // 等待超时后触发停止
    <-time.After(300 * time.Second)
    cancel() // 一次调用即可通知所有goroutine
}

调用cancel()后，所有监听ctx.Done()的goroutine都会立刻收到取消信号，无需维护大量通道，代码更简洁高效。

方案2：使用单个广播通道 #

通过关闭单个通道实现广播通知——通道关闭后，所有接收操作会立即返回：

import (
    "log"
    "time"
)

func myThreadFunc(stop <-chan struct{}) {
    for {
        select {
        case <-stop:
            log.Debug("Stopping thread")
            return
        default:
            callClientTask()
        }
    }
}

func callClientTask() {
    // 模拟耗时外部API调用
    time.Sleep(5 * time.Second)
}

func manager() {
    stopChan := make(chan struct{})
    // 启动1000个goroutine
    for i := 0; i < 1000; i++ {
        go myThreadFunc(stopChan)
    }

    // 等待超时后触发停止
    <-time.After(300 * time.Second)
    close(stopChan) // 关闭通道，所有goroutine同步收到信号
}

这种方式无需额外goroutine，仅通过关闭通道就能实现高效的广播停止，性能开销极低。

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