Go 并发可视化解释 - Semaphore

在这个系列的最后两篇文章中，我们讨论了来自sync包的Mutex和RWMutex。当我们希望只有一个Goroutine能够独占地访问共享数据时，这两个结构非常有用。

然而，在现实生活中，有些用例需要允许多个用户同时访问共享资源。这个数量可以很大，也可以很小，但必须始终是有限的。例如，一个容纳60000人的体育场在任何时候都不应容纳超过这个数量的人。或者，在机场，无论乘客队列有多长，同时允许办理登机手续的最大乘客数量就是开放柜台的数量。在计算机科学中，这种并发访问的用例是用信号量（Semaphore）来建模的。在本文中，我将通过可视化的方式解释信号量（Semaphore）的工作原理。我还将与您分享如何在Golang中使用通道来简单实现信号量（Semaphore）的方法。

场景

有4个Gopher想去游泳。然而，只有2个游泳道。每个泳道最多允许1个Gopher在任何时候游泳，但他们可以轮流游泳。

Candier想去游泳。因为两个泳道都可用，所以她可以立即获得一个泳道。

1*rNpdTpCmhFyRV0FULxkBvQ.png

片刻之后，Partier和Swimmer也想去游泳。此时只有一个泳道可用。只有其中一个可以获得泳道，另一个必须等待。当M个Goroutine竞争N（N < M）个槽位时，我们不能保证谁会赢得这个“竞争”。在这种情况下，当M=2和N=1时，假设Swimmer赢了。

1*hMzIXmAgnnqLXUhD8LTIOQ.png

Swimmer非常擅长这项运动。他迅速完成了他的轮次，并迅速释放了泳道，使其可用于Partier。与此同时，Candier仍然在她的泳道上游泳。

1*wtiF0BMWk8KY144wVXSgOg.png

Stringer想去游泳，但两个泳道目前都被占用。他别无选择，只能等待。他不知道也不关心哪一条泳道会先可用。

1*jxskbQzVSTQLLp6Lo3nGnQ.png

假设Partier在这项运动中也比Candier更有天赋。尽管在Candier之后开始，但仍然比她早完成。Partier释放了他的泳道，使其可用于Stringer。

1*kiRY3yaZ1GUjxu75DLGP5g.png

不久后，Candier完成并释放了她的泳道。泳道变得可用，但没有人试图占用它，它仍然可用。

最后，Stringer完成了他的轮次。他释放了他的泳道，使两个泳道都可用。

1*Xc6fPx1gWLDIQYLv_yL9ww.png

面试

在技术面试中，Semaphore这个术语可能听起来有点吓人。然而，正如你在上面的插图中所看到的，它是非常容易理解的。事实上，我曾几次在面试中问过我的面试者关于Semaphore的问题。例如，设计一个在黑色星期五上线的虚拟商店。有很多顾客想进去，但商店最多只能容纳N个顾客。每个顾客进去后没有时间限制，他/她可以一整天待在店里，也可以一进去就离开。当已经有N个顾客在里面时，后来的顾客必须排队等候，直到有人离开。当然，有很多解决这个问题的方法，其中没有一个是对或错的。我希望这篇文章为您提供了处理N个并发访问问题的另一种工具。

真实世界的例子

办理登机手续柜台、体育场的座位以及计算机资源，如CPU、内存和网络，有一个共同点：它们都是有限的。通常，控制应用程序资源使用是一个好主意。我曾在我的一个应用程序中使用Semaphore来限制一次只能有限数量的并发资源密集型Goroutine。这也可以通过一个包含N个Goroutine的池来解决。然而，由于我们的并发工作负载不是均匀分布在时间上的，它可能在0和N之间的任何位置，所以我们发现Semaphore是一个更好的选择。

展示你的代码！

请注意，Semaphore并不像sync.Mutex一样作为内置组件提供。相反，Go团队将其作为扩展提供。将其添加到项目中非常简单：go get golang.org/x/sync。

package main


import (
    "context"
    "golang.org/x/sync/semaphore"
    "log"
    "time"
)


func main() {
    pool := semaphore.NewWeighted(2)


    go swim("Candier", pool)
    go swim("Swimmer", pool)
    go swim("Partier", pool)
    go swim("Stringer", pool)


    time.Sleep(5 * time.Second) // For brevity, better use sync.WaitGroup


    log.Println("Main: Done, shutting down")
}


func swim(name string, pool *semaphore.Weighted) {
    log.Printf("%v: I want to swim\n", name)


    // In real applications, pass in your context such as HTTP request context
    ctx := context.Background()


    // We can also Acquire/Release more than 1
    // when the workloads consume different amount of resources
    if err := pool.Acquire(ctx, 1); err != nil {
        log.Printf("%v: Ops, something went wrong! I cannot acquire a lane\n", name)
        return
    }


    log.Printf("%v: I got a lane, I'm swimming\n", name)
    time.Sleep(time.Second)
    log.Printf("%v: I'm done. Releasing my lane\n", name)
    pool.Release(1)
}

实现自己的Semaphore

如果您不想将golang.org/x/sync/semaphore添加到项目中，使用通道自己实现Semaphore也相当简单。

type Semaphore struct {
    ch chan bool
}


func NewSemaphore(weight int) *Semaphore {
    return &Semaphore{
        ch: make(chan bool, weight),
    }
}


func (s *Semaphore) Acquire() {
    s.ch <- true
}


func (s *Semaphore) Release() {
    <-s.ch
}}