經(jīng)?？吹接腥藭柸绾蔚却鲄f(xié)程中創(chuàng)建的協(xié)程執(zhí)行完畢之后再結(jié)束主協(xié)程，例如如下代碼：
package main
import (
"fmt"
)
func main() {
go func() {
fmt.Println("Goroutine 1")
}()
go func() {
fmt.Println("Goroutine 2")
}()
}
執(zhí)行以上代碼很可能看不到輸出，因為有可能這兩個協(xié)程還沒得到執(zhí)行主協(xié)程已經(jīng)結(jié)束了，而主協(xié)程結(jié)束時會結(jié)束所有其他協(xié)程。
解決辦法是可以在main函數(shù)結(jié)尾加上等待：
package main
import (
"fmt"
"time"
)
func main() {
go func() {
fmt.Println("Goroutine 1")
}()
go func() {
fmt.Println("Goroutine 2")
}()
time.Sleep(time.Second * 1) // 睡眠1秒，等待上面兩個協(xié)程結(jié)束
}
這并不是完美的解決方法，如果這兩個協(xié)程中包含復雜的操作，可能很耗時間，就無法確定需要睡眠多久，當然可以用管道實現(xiàn)同步：
package main
import (
"fmt"
)
func main() {
ch := make(chan struct{})
count := 2 // count 表示活動的協(xié)程個數(shù)
go func() {
fmt.Println("Goroutine 1")
ch - struct{}{} // 協(xié)程結(jié)束，發(fā)出信號
}()
go func() {
fmt.Println("Goroutine 2")
ch - struct{}{} // 協(xié)程結(jié)束，發(fā)出信號
}()
for range ch {
// 每次從ch中接收數(shù)據(jù)，表明一個活動的協(xié)程結(jié)束
count--
// 當所有活動的協(xié)程都結(jié)束時，關閉管道
if count == 0 {
close(ch)
}
}
}
上面的解決方案是比較完美的方案，但是Go提供了更簡單的方法——使用sync.WaitGroup。
WaitGroup顧名思義，就是用來等待一組操作完成的。
WaitGroup內(nèi)部實現(xiàn)了一個計數(shù)器，用來記錄未完成的操作個數(shù)，它提供了三個方法，Add()用來添加計數(shù)。
Done()用來在操作結(jié)束時調(diào)用，使計數(shù)減一。
Wait()用來等待所有的操作結(jié)束，即計數(shù)變?yōu)?，該函數(shù)會在計數(shù)不為0時等待，在計數(shù)為0時立即返回。
package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
wg.Add(2) // 因為有兩個動作，所以增加2個計數(shù)
go func() {
fmt.Println("Goroutine 1")
wg.Done() // 操作完成，減少一個計數(shù)
}()
go func() {
fmt.Println("Goroutine 2")
wg.Done() // 操作完成，減少一個計數(shù)
}()
wg.Wait() // 等待，直到計數(shù)為0
}
可見用sync.WaitGroup是最簡單的方式。
補充：Golang 中使用WaitGroup的那點坑
sync.WaitGroup對于Golang開發(fā)者來說并不陌生，其經(jīng)常作為多協(xié)程之間同步的一種機制。用好它勢必會讓你事半功倍，但是一旦錯用將引發(fā)問題。
關于WaitGroup的使用網(wǎng)上有很多例子，在此就不做介紹了，我想說的是我在項目中使用WaitGroup遇到的坑。
在項目中，因為服務器有同步需求， 所以直接使用了WaitGroup，但是未考慮使用場景，結(jié)果在項目上線之后，高峰期的時候客戶端經(jīng)常出現(xiàn)卡頓，經(jīng)過多方查找，才發(fā)現(xiàn)如果使用WaitGroup的時候，未啟動單獨的goroutine，那么極有可能造成主線程的阻塞，
所以我做了下面的測試(測試中，我把WaitGroup置于協(xié)程內(nèi))：
import (
"fmt"
"sync"
"time"
)
func main() {
fmt.Println("main-1")
testW()
fmt.Println("main-2")
time.Sleep(time.Duration(15) * time.Second)
}
func testW() {
fmt.Println("testW-1")
go func() {
var wg sync.WaitGroup
fmt.Println("testW-2")
testW1(wg)
fmt.Println("testW-5")
wg.Wait()
fmt.Println("testW-6")
}()
}
func testW1(wg *sync.WaitGroup) {
wg.Add(1)
fmt.Println("testW-3")
time.AfterFunc(time.Second*5, func() {
wg.Done()
})
fmt.Println("testW-4")
}
輸出為：
main-1
testchan-1
main-2
testchan-2
testchan-3
testchan-4
testchan-5
// 過5秒
testchan-6
總結(jié)：
將WaitGroup用于goroutine內(nèi)，不會導致主線程的阻塞，同樣可以實現(xiàn)同步的效果。
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
您可能感興趣的文章:- Golang 實現(xiàn)分片讀取http超大文件流和并發(fā)控制
- Go 并發(fā)控制context實現(xiàn)原理剖析(小結(jié))
- 解決Golang 中使用WaitGroup的那點坑
- Go語言WaitGroup使用時需要注意的坑
- Golang中的sync.WaitGroup用法實例
- Go并發(fā)控制WaitGroup的使用場景分析
