GO Barrier

Barrier 用于多个线程完成状态的同步。在linux的系统库中有相关的实现,在 pthread.h 中有函数声明。在golang的标准库中还没有相关的实现。下面利用sync标准库实现个简单的barrier。

实现

----------$GOPATH/src/barrier/barrier.go-----------
package barrier
import (
 "sync"
)
type Barrier struct {
 curCnt int
 maxCnt int
 cond *sync.Cond
}
func NewBarrier(maxCnt int) *Barrier {
 mutex := new(sync.Mutex)
 cond := sync.NewCond(mutex)
 return &Barrier{curCnt: maxCnt, maxCnt: maxCnt, cond: cond}
}
func (barrier *Barrier) BarrierWait() {
 barrier.cond.L.Lock()
 if barrier.curCnt--; barrier.curCnt > 0 {
 barrier.cond.Wait()
 } else {
 barrier.cond.Broadcast()
 barrier.curCnt = barrier.maxCnt
 }
 barrier.cond.L.Unlock()
}

测试

package main
import (
 "fmt"
 "sync"
 "barrier"
)
func main() {
 fmt.Println("Hello World!")
 barrier := barrier.NewBarrier(3)
 var wg sync.WaitGroup
 for i := 0; i < 3; i++ {
 wg.Add(1)
 go func() {
 defer wg.Done()
 fmt.Println("A")
 barrier.BarrierWait()
 fmt.Println("B")
 barrier.BarrierWait()
 fmt.Println("C")
 }()
 }
 wg.Wait()
}

版权声明:本文为博主原创文章,未经博主允许不得转载。

有疑问加站长微信联系(非本文作者)

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传