Golang中Context取消后如何停止向Channel发送数据

你当前代码存在4个核心缺陷，导致无法实现预期效果：

• select语句的分支选择是随机的：当ctx.Done()返回的关闭channel和待发送的业务channel同时处于就绪状态时，Go不会优先选择取消分支，可能出现cancel调用后，仍有goroutine命中发送分支写入数据
• channel关闭时机错误：主协程收到第一个值就立刻关闭channel，但此时其他goroutine可能还没响应取消信号，一旦尝试向已关闭的channel写入数据会直接触发panic
• 多余的goroutine嵌套+死循环：每个worker函数内部又启动了一个匿名goroutine跑死循环，增加了取消信号传递的延迟，也提高了逻辑出错的概率
• 无缓冲channel的设计问题：第一个写入channel的goroutine必须等主协程接收才能继续执行cancel，这期间其他goroutine也可能同时尝试写入，出现竞争

package main

import (
	"context"
	"fmt"
	"log"
	"sync"
	"time"
)

func main() {
	ctx, cancel := context.WithCancel(context.Background())
	// 缓冲大小为1的channel，保证第一个写入的goroutine不会阻塞
	errChan := make(chan string, 1)
	var wg sync.WaitGroup

	// 启动3个worker goroutine
	wg.Add(3)
	go func() {
		defer wg.Done()
		one(ctx, errChan, cancel)
	}()
	go func() {
		defer wg.Done()
		two(ctx, errChan, cancel)
	}()
	go func() {
		defer wg.Done()
		three(ctx, errChan, cancel)
	}()

	// 接收第一个写入的值
	err := <-errChan
	fmt.Println("received:", err)
	// 双保险触发取消，通知所有剩余goroutine退出
	cancel()
	// 等待所有worker goroutine完全退出，确保没有写入方残留
	wg.Wait()
	// 所有写入操作结束后再安全关闭channel
	close(errChan)
	log.Println("all goroutines exited, program finished")
}

func one(ctx context.Context, c chan<- string, done context.CancelFunc) {
	select {
	case <-ctx.Done():
		log.Println("other goroutine already sent value, one exited")
		return
	case c <- "from one":
		log.Println("one sent value successfully")
		done()
		return
	}
}

func two(ctx context.Context, c chan<- string, done context.CancelFunc) {
	// 模拟业务处理耗时，方便观察竞争效果
	time.Sleep(1 * time.Second)
	select {
	case <-ctx.Done():
		log.Println("other goroutine already sent value, two exited")
		return
	case c <- "from two":
		log.Println("two sent value successfully")
		done()
		return
	}
}

func three(ctx context.Context, c chan<- string, done context.CancelFunc) {
	// 模拟业务处理耗时
	time.Sleep(2 * time.Second)
	select {
	case <-ctx.Done():
		log.Println("other goroutine already sent value, three exited")
		return
	case c <- "from three":
		log.Println("three sent value successfully")
		done()
		return
	}
}

• 采用缓冲大小为1的channel：第一个完成业务逻辑的goroutine可以直接把值写入channel，不需要等待主协程接收，能第一时间调用cancel通知其他goroutine退出，从根源上减少竞争窗口
• 移除worker函数内部多余的goroutine嵌套和死循环：每个worker只执行一次业务逻辑，直接通过select判断是写入成功还是收到取消信号，没有多余的性能开销
• 双保险取消机制：写入成功的goroutine会调用cancel，主协程收到值后也会调用cancel，确保所有剩余goroutine都能收到退出信号
• 安全的channel关闭逻辑：通过sync.WaitGroup等待所有写入方goroutine完全退出后，主协程再关闭channel，彻底避免向已关闭channel写入导致的panic，也不会出现重复关闭channel的问题
• 彻底避免重复写入：第一个goroutine写入值后，channel缓冲占满，其余goroutine的发送分支会处于阻塞状态，当取消信号触发后，ctx.Done()分支就绪，select会直接命中退出逻辑，不会执行写入操作

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