用 channel 实现类似 semaphore 的限流器

舞姬之光

发布时间：2026-02-14 16:24:10

279人浏览过

来源于php中文网

原创

不能直接用 chan struct{} 做信号量，因其无计数能力，仅能表达“有/无”，无法安全支持多次 acquire 后统一 release；需结合 sync.mutex 管理可用数量，或直接使用 golang.org/x/sync/semaphore。

用 channel 实现类似 semaphore 的限流器

为什么不能直接用 `chan struct{}` 做信号量

Go 的 channel 本身没有“计数”能力，chan struct{} 只能表达“有/无”，无法安全支持多次 Acquire 后再统一 Release —— 比如你想允许最多 5 个并发，但某个协程可能一次拿走 2 个许可，另一个只还 1 个，这时候光靠 channel 关闭/阻塞逻辑就乱了。标准库没有 semaphore，得自己包一层状态控制。

用带缓冲的 channel + 互斥锁实现可重入、可批量操作的限流器

核心思路：用 chan struct{} 管理“空闲许可”的分发，用 sync.Mutex 保护剩余数量（available），这样既能支持 Acquire(n) 和 Release(n)，又能避免 channel 容量被误读（比如 len(ch) 不是实时可用数，因为有 goroutine 正在 recv 但还没完成）。

实操建议：

迷你天猫商城

迷你天猫商城是一个基于Spring Boot的综合性B2C电商平台，需求设计主要参考天猫商城的购物流程：用户从注册开始，到完成登录，浏览商品，加入购物车，进行下单，确认收货，评价等一系列操作。作为迷你天猫商城的核心组成部分之一，天猫数据管理后台包含商品管理，订单管理，类别管理，用户管理和交易额统计等模块，实现了对整个商城的一站式管理和维护。所有页面均兼容IE10及以上现代浏览器。部署方式1、项目

下载

初始化时往带缓冲 channel 写入 n 个 struct{}，代表初始许可数
Acquire(n) 先加锁检查 available >= n，满足才从 channel 读取 n 次；不满足就解锁后阻塞在 channel 上（或用 select 加超时）
Release(n) 加锁更新 available += n，然后往 channel 写入 n 次 —— 注意这里必须确保写入成功，所以 channel 缓冲区大小要 ≥ 最大许可数
不要依赖 cap(ch) 或 len(ch) 判断状态，它们和业务语义不一致

一个轻量但线程安全的实现示例

type Semaphore struct {
    ch        chan struct{}
    mu        sync.Mutex
    available int
    limit     int
}
<p>func NewSemaphore(limit int) *Semaphore {
ch := make(chan struct{}, limit)
for i := 0; i < limit; i++ {
ch <- struct{}{}
}
return &Semaphore{
ch:        ch,
available: limit,
limit:     limit,
}
}</p><p>func (s *Semaphore) Acquire(n int) bool {
s.mu.Lock()
if s.available >= n {
s.available -= n
s.mu.Unlock()
for i := 0; i < n; i++ {
<-s.ch
}
return true
}
s.mu.Unlock()
// 阻塞式获取：也可改用 select + timeout
for i := 0; i < n; i++ {
<-s.ch
}
s.mu.Lock()
s.available -= n
s.mu.Unlock()
return true
}</p><p>func (s *Semaphore) Release(n int) {
s.mu.Lock()
s.available += n
s.mu.Unlock()
for i := 0; i < n; i++ {
s.ch <- struct{}{}
}
}

注意：Acquire 中两次加锁/解锁之间不能有 channel 操作，否则可能死锁；Release 的写入必须发生在加锁之后更新 available，否则并发 Acquire 可能读到过期值。