Golang用channel实现信号量控制并发

本文深入浅出地讲解了如何巧妙利用 Go 语言中零内存开销的 `chan struct{}` 构建轻量、高效且线程安全的信号量，以精确控制并发 goroutine 数量——无需依赖第三方库或复杂锁机制，仅通过带缓冲通道的发送（占用槽位）和接收（释放槽位）操作，即可实现类似传统信号量的资源配额管理，特别适合限流、任务调度和临界区保护等场景，是 Go 并发编程中简洁而强大的实践范式。

用 chan struct{} 实现固定容量的信号量

Go 没有内置 Semaphore 类型，但用带缓冲的 chan struct{} 能精准模拟：容量即并发上限，send 占用、recv 释放。关键不是“锁”，而是“资源槽位”的原子占与放。

实操建议：

• 初始化：sem := make(chan struct{}, 3) 表示最多 3 个 goroutine 同时执行
• 进入临界区前必须 sem （阻塞直到有空槽）
• 退出后必须 <-sem（释放一个槽，注意不能写成 sem <- struct{}{}）
• 务必用 defer 保证释放，尤其在函数含多个 return 或 panic 可能时

为什么不用 sync.Mutex 或 sync.WaitGroup？

Mutex 是互斥，只允许 1 个；WaitGroup 是计数等待，不阻塞新协程启动。信号量核心是「限流」而非「同步」或「等待完成」——你要的是「超了就等」，不是「等别人做完」。

常见错误现象：

• 误用 sync.WaitGroup 控制并发：它无法让第 4 个 goroutine 主动暂停，只会无节制启动然后干等
• 用 time.Sleep 代替阻塞获取：破坏了资源竞争的确定性，且无法响应槽位释放
• 忘记 close channel 或重复 close：导致 panic；信号量 channel 永远不该 close

封装成可复用的 Semaphore 类型

直接裸用 chan struct{} 容易漏掉 defer 或写反方向。封装一层能防错、加注释、也方便后续扩展（比如带超时的 TryAcquire）。

最小可行封装示例：

type Semaphore struct {
    c chan struct{}
}
<p>func NewSemaphore(n int) *Semaphore {
return &Semaphore{c: make(chan struct{}, n)}
}</p><p>func (s <em>Semaphore) Acquire() { s.c <- struct{}{} }
func (s </em>Semaphore) Release() { <-s.c }
</p>

使用时更清晰：

sem := NewSemaphore(5)
for i := 0; i <h3>和 <code>golang.org/x/sync/semaphore</code> 的实际差异</h3><p>标准库外的 <code>semaphore</code> 包（如 <code>x/sync/semaphore</code>）底层仍是 channel，但它支持带上下文的 <code>Acquire(ctx, n)</code> 和可中断等待，适合长时 IO 场景。如果你只需要固定粒度（n=1）、无超时控制，手写 <code>chan struct{}</code> 更轻量、无依赖、行为完全透明。</p><p>注意点：</p>

• x/sync/semaphore 的 Acquire 返回 error，需检查是否被 ctx cancel；裸 channel 版本不会返回 error，但会永久阻塞（除非你手动加 select + timeout）
• 它的 WeightedSemaphore 支持非单位占用（比如某任务占 3 个槽），裸 channel 难以简洁表达
• 生产环境若需 cancel、timeout、metrics 上报，优先用官方维护的包；否则，make(chan struct{}, N) 就是最简解

真正容易被忽略的是：信号量保护的不是代码段，而是**下游资源**（如数据库连接池、HTTP 客户端限流、文件句柄）。别让它变成“为了限而限”——得清楚每个槽对应什么物理约束。

今天关于《Golang用channel实现信号量控制并发》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！