Golang等待协程运行结束

优雅地等待子协程执行完毕
参考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函数结束,所有协程都被结束了