Golang并发工具包之信号量(Semaphore)

Go虽然天生的支持高并发,但是有些场景下我们还是需要控制协程同时并发处理的数量,在Java的juc包中已经提供了类似功能的工具类-信号量(Semaphore),它是基于AQS实现的。Go的SDK中并没有提供类似的API,我们通过goroutine和channel实现一个简单的Semaphore,并提供:获取许可(Acquire())、指定时间内获取许可(TryAcquireOnTime)、释放许可(Release())等方法,具体实现如下:

type Semaphore struct {
 permits int // 许可数量
 channel chan int // 通道
}
/* 创建信号量 */
func NewSemaphore(permits int) *Semaphore {
 return &Semaphore{channel: make(chan int, permits), permits: permits}
}
/* 获取许可 */
func (s *Semaphore) Acquire() {
 s.channel <- 0
}
/* 释放许可 */
func (s *Semaphore) Release() {
 <-s.channel
}
/* 尝试获取许可 */
func (s *Semaphore) TryAcquire() bool {
 select {
 case s.channel <- 0:
 return true
 default:
 return false
 }
}
/* 尝试指定时间内获取许可 */
func (s *Semaphore) TryAcquireOnTime(timeout time.Duration) bool {
 for {
 select {
 case s.channel <- 0:
 return true
 case <-time.After(timeout):
 return false
 }
 }
}
/* 当前可用的许可数 */
func (s *Semaphore) AvailablePermits() int {
 return s.permits - len(s.channel)
}

本文来自:简书

感谢作者:_男猪脚

查看原文:Golang并发工具包之信号量(Semaphore)

• 编辑
• 预览