Golang等待子协程执行完毕\go 主协程退出其他协程也会退出

刺骨的言语ヽ痛彻心扉 2022-10-12 15:58 439阅读 0赞

### 文章目录 ###

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