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