package main

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

// 当我们知道需要等待的具体任务数量时，可以一次性添加所有任务。
// WaitGroup 批量添加示例: `sync.WaitGroup`可以在启动goroutine前批量添加任务，避免在循环中调用`Add`方法
func batchAddExample() {
	fmt.Println("=== WaitGroup 批量添加示例 ===")

	var wg sync.WaitGroup
	tasks := []string{"下载文件", "处理图片", "发送邮件", "备份数据"}

	// 一次性添加所有任务（推荐方式）
	wg.Add(len(tasks))

	// 启动所有goroutine
	for i, task := range tasks {
		go func(id int, taskName string) {
			defer wg.Done()

			fmt.Printf("Worker %d: 开始执行 %s\n", id, taskName)
			time.Sleep(time.Duration(id+1) * 500 * time.Millisecond)
			fmt.Printf("Worker %d: 完成 %s\n", id, taskName)
		}(i, task)
	}

	fmt.Println("主goroutine: 等待所有任务完成...")
	wg.Wait()
	fmt.Println("所有任务已完成")
}

func main() {
	batchAddExample()
}