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