CyclicBarrier: 循环栅栏

针对场景

CyclicBarrier允许一组goroutine相互等待,直到所有goroutine都到达栅栏后,才能继续执行

因为CyclicBarrier是可复用的,所以叫做循环栅栏

CyclicBarrier和WaitGroup的选择:

  1. 如果是一组goroutine相互等待,然后一起继续执行,那么使用CyclicBarrier
  2. 如果是一个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])
	}
}
上一篇:啊 ♂ Goroutine是个啥


下一篇:golang