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

_男猪脚 · · 3722 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

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)
}

有疑问加站长微信联系（非本文作者）