优雅地等待子协程执行完毕
参考URL: https://learnku.com/articles/35130
方式1: sync.WaitGroup
sync.WaitGroup 类型是并发安全的,提供了以下三个方法:
func (wg *WaitGroup) Add(delta int)
func (wg *WaitGroup) Done()
func (wg *WaitGroup) Wait()
- Add:WaitGroup 类型有一个计数器,默认值是 0,通常通过个方法来标记需要等待的子协程数量
- Done:当某个子协程执行完毕后,可以通过 Done 方法标记已完成,常用 defer 语句来调用
- Wait 阻塞当前协程,直到对应 WaitGroup 类型实例的计数器值归零
demo使用
package main
import (
"fmt"
"sync"
)
func add_num(a, b int, done func()) {
defer done()
c := a + b
fmt.Printf("%d + %d = %d\n", a, b, c)
}
func main() {
var wg sync.WaitGroup
wg.Add(10)
for i := 0; i < 10; i++ {
go add_num(i, 1, wg.Done)
}
wg.Wait()
}
go 主协程退出其他协程也会退出
package main
import (
"fmt"
"time"
)
func newTask(){
for{
fmt.Println("this is a newTask!!")
time.Sleep(time.Second)//休眠1s
}
}
func main(){
//新建一个协程,新建一个任务
go newTask()
//阻塞运行 不然的newTask还没有运行,main就会运行结束,从而整个结束。
i:=0
for{
i++
if(i==3){
break//结束跳出,newTask也会结束执行
}
fmt.Println("this is main goroutine!!!")
time.Sleep(time.Second)
}
主协程退出,子协程也跟着退出了。 golang main函数结束,所有协程都被结束了