GO 利用 sync 库实现 Barrier

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