如何获取与WaitGroup关联的运行中goroutine的数量？

核心结论 #

原生Go标准库的sync.WaitGroup没有对外暴露可安全调用的计数查询接口，无法直接通过它获取剩余运行的goroutine数量。最稳妥的实现方式是对sync.WaitGroup做轻量封装，自己维护并发安全的计数器，实现成本极低。

代码实现 #

import "sync"

type CountableWG struct {
    wg  sync.WaitGroup
    mu  sync.Mutex
    cnt int64
}

// Add 增加待等待的goroutine数量，delta可以为负数，用法和原生WaitGroup完全一致
func (c *CountableWG) Add(delta int) {
    c.mu.Lock()
    c.cnt += int64(delta)
    c.mu.Unlock()
    c.wg.Add(delta)
}

// Done 标记单个goroutine执行完成
func (c *CountableWG) Done() {
    c.mu.Lock()
    c.cnt -= 1
    c.mu.Unlock()
    c.wg.Done()
}

// Wait 阻塞等待所有关联goroutine执行完成
func (c *CountableWG) Wait() {
    c.wg.Wait()
}

// RunningCount 获取当前仍在运行的关联goroutine数量，并发安全
func (c *CountableWG) RunningCount() int64 {
    c.mu.Lock()
    defer c.mu.Unlock()
    return c.cnt
}

使用说明 #

你可以直接替换原有代码中的sync.WaitGroup为上述CountableWG结构体，原有调用Add、Done、Wait的逻辑不需要做任何修改，需要统计数量时调用RunningCount()方法即可拿到准确的实时数值。

不推荐的实现方式 #

• 不要通过反射读取sync.WaitGroup的私有字段获取计数：不同Go版本的WaitGroup内部结构可能调整，反射代码随时会失效，且读取未加锁的私有字段会拿到脏数据，结果不可靠。
• 不要自行通过零散的全局变量计数：单独维护的计数变量如果没有和WaitGroup的Add/Done操作绑定，很容易出现漏加、漏减的逻辑错误，不如上述封装方案可靠。

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