深入理解Go语言并发：select{}行为、死锁避免与工作池模式

package main

import (
    "fmt"
    "math/rand"
    "sync" // 导入sync包
    "time"
)

func runTask(t string, wg *sync.WaitGroup) {
    defer wg.Done() // 任务完成后通知WaitGroup
    start := time.Now()
    fmt.Println("starting task", t)
    time.Sleep(time.Millisecond * time.Duration(rand.Int31n(1500))) // 模拟处理时间
    fmt.Println("done running task", t, "in", time.Since(start))
}

func main() {
    numWorkers := 3 // 此处为示例，实际并发数由WaitGroup控制
    files := []string{"a", "b", "c", "d", "e", "f", "g", "h", "i", "j"}

    var wg sync.WaitGroup // 声明一个WaitGroup

    // activeWorkers := make(chan bool, numWorkers) // 不再需要此通道来限制并发数

    for _, f := range files {
        wg.Add(1) // 为每个任务增加计数器
        // activeWorkers <- true // 原始代码中用于限制并发的逻辑，此处不再适用
        fmt.Printf("scheduling task %s\n", f) // 提示正在调度任务
        go runTask(f, &wg)
    }

    wg.Wait() // 阻塞main goroutine，直到所有任务完成
    fmt.Println("All tasks completed.")
}

package main

import (
    "fmt"
    "math/rand"
    "time"
)

// runTask 模拟一个耗时任务，并返回任务标识
func runTask(t string) string {
    start := time.Now()
    fmt.Println("starting task", t)
    time.Sleep(time.Millisecond * time.Duration(rand.Int31n(1500))) // 模拟处理时间
    fmt.Println("done running task", t, "in", time.Since(start))
    return t // 返回任务标识作为结果
}

// worker goroutine 从输入通道接收任务，处理后将结果发送到输出通道
func worker(id int, in chan string, out chan string) {
    for task := range in {
        fmt.Printf("Worker %d processing task %s\n", id, task)
        result := runTask(task)
        out <- result // 将结果发送到输出通道
    }
    fmt.Printf("Worker %d exiting.\n", id)
}

func main() {
    numWorkers := 3 // 限制并发的worker数量
    files := []string{"a", "b", "c", "d", "e", "f", "g", "h", "i", "j"}

    // 创建输入和输出通道
    in := make(chan string)
    out := make(chan string)

    // 启动固定数量的worker goroutine
    for i := 0; i < numWorkers; i++ {
        go worker(i+1, in, out)
    }

    // 启动一个goroutine来调度所有任务到输入通道
    go func() {
        for _, f := range files {
            in <- f // 提交任务
        }
        close(in) // 所有任务提交完毕后关闭输入通道
    }()

    // 从输出通道收集所有任务的结果
    // 循环次数等于任务总数，确保收集所有结果
    for i := 0; i < len(files); i++ {
        completedTask := <-out
        fmt.Printf("Received result for task: %s\n", completedTask)
    }
    close(out) // 所有结果收集完毕后关闭输出通道

    fmt.Println("All tasks processed and results collected.")
}