[13th][Day10] waiting group

之前有做过一个需求：要打一个 request 一千万次，并收集其 response 以供之後流程处理分析

我的第一个版本写得非常非常纯『蠢』


func main() {
	for i := 1; i < 11; i++ {
		go testInsert(grpcClient, "f"+strconv.Itoa(i)+".txt")
	}
	time.Sleep(3600 * time.Second)

}

func testInsert(grpcClient abc.AbcServiceClient, fileName string) {
	aa1 := getResultWrite(grpcClient, "QWERT", 1000000)
	fmt.Println(aa1)

	f, err := os.Create(fileName)

	defer f.Close()

	if err != nil {
		// fmt.Println(err.Error())
	} else {

		_, err = f.Write([]byte(aa1))
		// fmt.Println(err.Error())
	}
}

缺点：我需要自行调整 time.Sleep 的秒数，否则如果睡完後，func main就直接结束， go routine 还没执行完的话，那本次执行的的任务，收集到的东西会全部消失，浪费 N 分钟

前辈 W大大帮我加入了 waiting group 的概念 ...


func main() {
    wg := sync.WaitGroup{}

	for i := 1; i <= 10; i++ {
		wg.Add(1)
		go conn.testInsert("f"+strconv.Itoa(i)+".txt", &wg)
	}
	wg.Wait()
}

func (conn *Conn) testInsert(fileName string, wg *sync.WaitGroup) {
	aa1 := conn.getResultWrite(conn.param, conn.fileSize)
	// fmt.Println(aa1)

	f, err := os.Create(fileName)
	defer func() {
		f.Close()
		wg.Done()
	}()

	if err != nil {
		fmt.Println(err.Error())
	} else {
		_, err = f.Write([]byte(aa1))
		// fmt.Println(err.Error())
	}
}

func (wg *WaitGroup) Add(delta int)

Add adds delta, which may be negative, to the WaitGroup counter. If the counter becomes zero, all goroutines blocked on Wait are released. If the counter goes negative, Add panics.

https://pkg.go.dev/sync#WaitGroup.Add
delta 只能为正，负的会 panic
增加可以本 wg 等待的 goroutine 数