Go语言无内置信号量,推荐使用golang.org/x/sync/semaphore包实现并发控制,支持Acquire/Release操作与上下文取消;也可用带缓冲channel模拟二进制信号量。
Go 语言本身没有内置的信号量(semaphore)类型,但可以通过 sync.WaitGroup + chan struct{} 或更推荐的方式 —— 使用标准库中的 golang.org/x/sync/semaphore 包来实现精确的并发数限制。
这是最简洁、线程安全且语义清晰的做法。该包提供了一个带权重的信号量,支持 Acquire/Release 操作,适合控制同时运行的 goroutine 数量。
go get golang.org/x/sync/semaphore
sem.Acquire(ctx, 1) 获取一个许可;完成后必须调用 sem.Release(1) 归还Acquire 是阻塞操作,若无可用许可会等待;支持上下文取消,可及时退出示例代码:
package mainimport ( "context" "fmt" "log" "time" "golang.org/x/sync/semaphore" )
func main() { sem := semaphore.NewWeighted(3) // 最多 3 个并发 ctx := context.Background()
for i := 0; i < 10; i++ { i := i go func() { if err := sem.Acquire(ctx, 1); err != nil { log.Printf("acquire failed: %v", err) return } defer sem.Release(1) fmt.Printf("task %d started\n", i) time.Sleep(time.Second * 2) // 模拟耗时操作 fmt.Printf("task %d done\n", i) }() } // 等待所有 goroutine 完成(实际项目中建议用 WaitGroup 配合) time.Sleep(time.Second * 8)
}
立即学习“go语言免费学习笔记(深入)”;
用 channel 模拟二进制信号量(简单场景)
如果不想引入外部依赖,可以用带缓冲的
chan struct{} 实现简易版信号量,适用于固定权重(每次申请/释放 1 单位)的场景。
sem := make(chan struct{}, maxConcurrency)
sem (满时阻塞)
示例片段:
sem := make(chan struct{}, 2)
for i := 0; i < 6; i++ {
go func(id int) {
sem <- struct{}{} // 获取许可
defer func() { <-sem }() // 释放许可
fmt.Printf("working on %d\n", id)
time.Sleep(time.Second)
}(i)}
立即学习“go语言免费学习笔记(深入)”;
常见陷阱与注意事项
信号量只是并发控制工具,不解决数据竞争或任务协调问题,使用时需格外注意:
Acquire 都有对应的 Release,否则会导致许可泄露、后续任务永久阻塞
defer 是最稳妥的方式两者适用不同模式:
多数 API 限流、下游调用节流场景,x/sync/semaphore 更轻量直接;后台批处理、消息消费等长期运行任务,worker pool 更合适。