Go中cancelCtx的done属性如何从atomic.Value转换为chan struct{}？

你观察到的现象是对的：cancelCtx的done字段是atomic.Value类型，初始状态下没有存储有效chan，当调用cancel方法后，它会被设置为chan struct{}（通常是已关闭的）。具体过程如下：

1. cancelCtx的核心结构 #

cancelCtx的关键字段定义：

type cancelCtx struct {
	Context
	mu       sync.Mutex            // 保护后续字段的互斥锁
	done     atomic.Value          // 存储chan struct{}，懒加载创建，首次cancel时关闭
	children map[canceler]struct{} // 子context集合，首次cancel时置为nil
	err      error                 // cancel时设置的错误信息
}

atomic.Value是Go标准库提供的原子类型，支持在多goroutine环境下安全存储、加载任意类型的值，这是实现类型转换的基础。

2. 初始状态：done未绑定chan #

调用newCancelCtx创建cancelCtx时，仅初始化了父Context字段，done是atomic.Value的零值，此时调用done.Load()会返回nil。

3. Done()方法的懒加载逻辑 #

当你调用ctx.Done()时，会触发chan的懒加载：

func (c *cancelCtx) Done() <-chan struct{} {
	d := c.done.Load()
	if d != nil {
		return d.(chan struct{})
	}
	c.mu.Lock()
	defer c.mu.Unlock()
	d = c.done.Load()
	if d == nil {
		d = make(chan struct{})
		c.done.Store(d)
	}
	return d.(chan struct{})
}

• 先尝试原子加载done，若已有值，直接通过类型断言转换为chan struct{}返回。
• 若没有值，加锁后再次检查（双重检查锁），确认后创建新的chan struct{}并存储到done中，再返回。

4. Cancel时的chan设置与关闭 #

调用cancel函数时，关键逻辑如下：

func (c *cancelCtx) cancel(removeFromParent bool, err error) {
	if err == nil {
		panic("context: internal error: missing cancel error")
	}
	c.mu.Lock()
	if c.err != nil {
		c.mu.Unlock()
		return // 已被cancel过，直接返回
	}
	c.err = err
	d, _ := c.done.Load().(chan struct{})
	if d == nil {
		// 若还未创建chan，直接存储全局已关闭的chan
		c.done.Store(closedchan)
	} else {
		// 否则关闭已存在的chan
		close(d)
	}
	// 遍历cancel所有子context
	for child := range c.children {
		child.cancel(false, err)
	}
	c.children = nil
	c.mu.Unlock()

	if removeFromParent {
		removeChild(c.Context, c)
	}
}

• 加锁保证线程安全，先检查是否已被cancel。
• 通过类型断言尝试从done加载chan struct{}：
  • 若返回nil（说明还没调用过Done()），直接将全局的closedchan（提前创建好的已关闭chan struct{}）存储到done中。
  • 若已有chan，直接关闭它。
• 最后处理所有子context的cancel逻辑。

简单来说，atomic.Value作为通用容器存储chan struct{}，通过类型断言实现从atomic.Value到chan struct{}的类型转换，同时结合懒加载和原子操作保证多goroutine下的安全访问。

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