掌握Go语言：Go 并发编程，轻松应对大规模任务处理和高并发请求（34）

package main
 
import (
    "fmt"
    "sync"
)
 
// 模拟耗时任务
func longRunningTask(id int, wg *sync.WaitGroup) {
    defer wg.Done() // 确保WaitGroup的计数器在任务结束时减一
    fmt.Printf("Task %d is running\n", id)
    // 模拟任务执行时间
    for i := 0; i < 5; i++ {
        // 每次循环sleep 0.1秒
        <-time.After(time.Duration(100 * time.Millisecond))
    }
    fmt.Printf("Task %d is done\n", id)
}
 
func main() {
    var wg sync.WaitGroup
    // 创建一个有三个工作者的工作池
    workerPool := make(chan bool, 3)
 
    // 模拟8个任务
    for i := 1; i <= 8; i++ {
        wg.Add(1) // 为每个任务增加计数
        go func(id int) {
            // 通过工作池发起任务
            workerPool <- true
            longRunningTask(id, &wg)
            <-workerPool // 任务完成，从工作池获取权限
        }(i)
    }
 
    // 等待所有任务完成
    wg.Wait()
}

这段代码使用了sync.WaitGroup来同步并发任务的执行，并通过有界缓冲区（工作池）来限制并发的数量。每个任务作为一个goroutine执行，并在开始和结束时分别对sync.WaitGroup的计数加一和减一。工作池通过channel实现，只允许同时运行固定数量的goroutine。这种方式可以有效管理系统资源，避免过度使用系统资源导致的性能问题。