golang并发控制

golang对于并发的支持可以说是最好的,也是我们之所以采用go语言进行编程的一个重要的因素。

golang的并发控制,主要有以下几种方式:

Channel

func Process(ch chan<- int)  {
	// do something
	time.Sleep(time.Second)

	ch <- 1
}

func TestOther(t *testing.T) {
	// 有几个协程,就需要创建包含多少元素的切片
	channels := make([]chan int, 10)

	for i := 0; i < 10; i++ {
		channels[i] = make(chan int)
		go Process(channels[i])
	}

	for i, ch := range channels {
		<-ch
		fmt.Printf("Routine %d quit \n", i)
	}
}

优点是实现简单,缺点是当需要大量创建协程时就需要有相同数量的channel,而且对于子协程继续派生出来的协程不方便控制。

WaitGroup

func Process(wg *sync.WaitGroup, i int)  {
	defer wg.Done()

	// do something
	time.Sleep(time.Second)

	fmt.Printf("Routine %d quit \n", i)
}

func TestOther(t *testing.T) {
	wg := sync.WaitGroup{}

	wg.Add(10)

	for i := 0; i < 10; i++ {
		go Process(&wg, i)
	}
	wg.Wait()
	fmt.Printf("all Goroutine quit \n")
}

WaitGroup是Golang应用开发过程中经常使用的并发控制技术。WaitGroup,可理解为Wait-Goroutine-Group,即等待一组goroutine结束。比如某个goroutine需要等待其他几个goroutine全部完成,那么使用WaitGroup可以轻松实现。

Add()操作必须早于Wait(), 否则会panic
Add()设置的值必须与实际等待的goroutine个数一致,否则会panic

0 评论
最新
最旧 最多投票
内联反馈
查看所有评论
滚动至顶部