golang中waitgroup

Go语言中的waitgroup（等待组）是一个用来等待一组线程完成的机制。它类似于信号量，但是比信号量更加灵活，方便地控制线程的并发和同步。在多线程编程中，我们经常需要等待多个线程完成后再进行下一步操作，这时就可以使用waitgroup来实现线程的同步。

一、waitgroup的基本概念

waitgroup是sync包中的一个结构体，它提供了三个方法：Add()、Done()和Wait()。Add()方法用来添加要等待的线程数量，Done()方法用来表示某个线程已经完成，Wait()方法用来阻塞当前的线程，直到所有等待的线程都完成为止。

二、waitgroup的使用方法

在使用waitgroup时，我们需要先创建一个waitgroup对象，然后在每个线程开始执行的位置调用Add()方法，表示要等待的线程数量加1；在每个线程结束时调用Done()方法，表示该线程已经完成。最后，我们在主线程中调用Wait()方法，阻塞主线程，直到所有等待的线程都完成。

三、waitgroup的示例代码

下面我们通过一个示例代码来演示waitgroup的使用：

package main

import (
	"fmt"
	"sync"
	"time"
)

func main() {
	var wg sync.WaitGroup
	for i := 0; i < 5; i++ {
		wg.Add(1)
		go worker(i, &wg)
	}
	wg.Wait()
	fmt.Println("All workers done!")
}

func worker(id int, wg *sync.WaitGroup) {
	defer wg.Done()
	fmt.Printf("Worker %d starting\n", id)
	time.Sleep(time.Second)
	fmt.Printf("Worker %d done\n", id)
}

在上面的代码中，我们创建了一个waitgroup对象wg，并且使用for循环启动了5个worker线程。在每个worker线程的开头调用wg.Add(1)方法，表示要等待的线程数量加1；在每个worker线程的结尾调用wg.Done()方法，表示该线程已经完成。最后，我们在主线程中调用wg.Wait()方法，阻塞主线程，直到所有等待的线程都完成。

通过运行以上代码，我们可以看到每个worker线程按顺序依次启动、执行和完成，同时主线程会等待所有worker线程都执行完毕后才会打印"All workers done!"。

总之，waitgroup是一个非常方便并且强大的机制，可以用来控制线程的并发和同步。它可以帮助我们实现多个线程的同步，提高程序的性能和效率。在实际开发中，只需要简单地按照上面的示例代码使用waitgroup，就可以轻松地实现线程的同步。因此，如果你是一个golang开发者，一定要熟练掌握waitgroup的用法，它会给你的开发工作带来很大的便利。