文章目录

• Golang等待协程运行结束
• • 方式1: sync.WaitGroup
  • demo使用
• go 主协程退出其他协程也会退出

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函数结束，所有协程都被结束了

标签：wg,WaitGroup,协程,Golang,主协程,func,go,main
来源： https://blog.csdn.net/inthat/article/details/118491901