阅读量:1
在Go语言中，可以使用sync包中的WaitGroup类型来优化计算密集型任务的性能。
WaitGroup是一个计数器，用于等待一组goroutine的结束。当我们启动一组goroutine来执行计算密集型任务时，我们可以使用WaitGroup来等待所有的goroutine完成，然后继续执行后续的操作。
下面是一个使用WaitGroup优化性能的示例代码：
package main import ( "fmt" "sync" "time" ) func main() { var wg sync.WaitGroup // 设置要启动的goroutine数量 wg.Add(3) // 启动goroutine执行计算密集型任务 go calculate(1, &wg) go calculate(2, &wg) go calculate(3, &wg) // 等待所有的goroutine完成 wg.Wait() // 继续执行后续的操作 fmt.Println("All goroutines completed") } func calculate(id int, wg *sync.WaitGroup) { defer wg.Done() fmt.Printf("Start goroutine %d\n", id) // 模拟计算密集型任务 time.Sleep(time.Second * time.Duration(id)) fmt.Printf("End goroutine %d\n", id) } 在上面的代码中，我们首先创建了一个WaitGroup变量wg，并通过wg.Add(3)设置要启动的goroutine数量为3。
然后，我们使用go calculate(1, &wg)的方式启动了3个goroutine来执行计算密集型任务。在每个goroutine中，我们使用defer wg.Done()来标记该goroutine的结束。
最后，我们调用wg.Wait()来等待所有的goroutine完成，然后继续执行后续的操作。
通过使用WaitGroup，我们可以确保所有的goroutine都完成后再继续执行后续的操作，从而达到优化性能的目的。
