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)
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容