go 所有routine完成后才退出

ゝ一纸荒年。 2022-10-28 10:06 90阅读 0赞

### go 等待所有routine完成后才推出 ###

如下面的例子，有10个请求,每个请求需要花费2秒去处理，等待程序处理完成后，退出程序，我们遍历10个请求，然后发起并发请求即可。

需要注意的事项是：

waitGroup.Add(1) //添加waitGroup计算器
    函数结束,Add后立马用defer Done(),如果在程序后面才加defer Done(),程序因为各种原因没走到done挂了,整个程序也会挂掉
    例如下面的num == 8直接return掉了,如果你把defer Done()放在后面程序会挂
    defer waitGroup.Done()
    还有Add和Done必须是成对出现的,一次add多次done会导致还有goroutine没执行完成就退出了

package main
    
    import (
    	"fmt"
    	"sync"
    	"time"
    )
    
    var (
    	//添加一个waitGroup
    	waitGroup = sync.WaitGroup{}
    )
    
    func main() {
    	waitNum := 10
    
    	start := time.Now().Unix()
    	for i := 0; i < waitNum; i++ {
    		//睡眠2秒,模拟程序执行时间
    		go sleep2(2, i)
    	}
    
    	//阻塞起来,等待waitGroup中的计算器为0,则退出阻塞
    	waitGroup.Wait()
    
    	//全部任务完成
    	fmt.Println("使用时间：", time.Now().Unix()-start, "秒")
    }
    func sleep2(sleepSecond int64, snum int) {
    	waitGroup.Add(1) //添加waitGroup计算器
    
    	//函数结束,Add后立马用defer Done(),如果在程序后面才加defer Done(),程序因为各种原因没走到done挂了,整个程序也会挂掉
    
    	//例如下面的num == 8直接return掉了,如果你把defer Done()放在后面程序会挂
    	defer waitGroup.Done()
    	//还有Add和Done必须是成对出现的,一次add多次done会导致还有goroutine没执行完成就退出了
    
    	fmt.Println("sleeping...", snum)
    
    	if snum == 8 {
    		return
    	}
    
    	time.Sleep(time.Duration(sleepSecond) * time.Second)
    	fmt.Println("done...", snum)
    
    }

运行结果：

sleeping... 9
    sleeping... 3
    sleeping... 1
    sleeping... 7
    sleeping... 8
    sleeping... 4
    sleeping... 5
    sleeping... 0
    sleeping... 2
    sleeping... 6
    done... 6
    done... 9
    done... 3
    done... 4
    done... 2
    done... 1
    done... 5
    done... 0
    done... 7
    使用时间： 2 秒
    
    Process finished with exit code 0