针对场景
CyclicBarrier允许一组goroutine相互等待,直到所有goroutine都到达栅栏后,才能继续执行
因为CyclicBarrier是可复用的,所以叫做循环栅栏
CyclicBarrier和WaitGroup的选择:
- 如果是一组goroutine相互等待,然后一起继续执行,那么使用CyclicBarrier
- 如果是一个goroutine等待一组goroutine,那么使用WaitGroup
简单使用
CyclicBarrier有两个初始化方法:
New:只需要一个参数,指定参与者数量
NewWithAction:额外提供一个方法参数,当最后一个到达者到达后,其他参与者释放前会执行这个方法;一般用于做放行前的共享变量的更新等
其他方法:
Await(ctx context.Contex):等待其他参与者到达,如果ctx被cancel那么会返回一个ErrBrokenBarrier
源码实现
https://github.com/marusama/cyclicbarrier
并发趣题
H2O制造厂
制造厂生产H2O,需要两个氢原子、一个氧原子,2N条生产线生产氢原子、N条生产线生产氧原子,只有生产出两个氢原子、一个氧原子组合成一个H2O分子所有生产线才能继续执行
import (
"context"
"fmt"
"github.com/marusama/cyclicbarrier"
"golang.org/x/sync/semaphore"
)
type H2O struct {
OSema *semaphore.Weighted
HSema *semaphore.Weighted
Barrier cyclicbarrier.CyclicBarrier
}
func NewH2O() *H2O {
return &H2O{
OSema: semaphore.NewWeighted(1),
HSema: semaphore.NewWeighted(2),
Barrier: cyclicbarrier.New(3),
}
}
func H2OPractive(n int) {
ctx := context.Background()
h2o := NewH2O()
resCh := make(chan string)
for i := 0; i < n; i++ {
// O
go func() {
for {
h2o.OSema.Acquire(ctx, 1)
resCh <- "O"
h2o.Barrier.Await(ctx)
h2o.OSema.Release(1)
}
}()
}
for i := 0; i < 2*n; i++ {
// H
go func() {
for {
h2o.HSema.Acquire(ctx, 1)
resCh <- "H"
h2o.Barrier.Await(ctx)
h2o.HSema.Release(1)
}
}()
}
strRes := make([]string, 3)
for {
strRes[0] = <-resCh
strRes[1] = <-resCh
strRes[2] = <-resCh
fmt.Printf("%v%v%v\n", strRes[0], strRes[1], strRes[2])
}
}