Go waitgroup with channel (worker)

package main

import (
    "fmt"
    "sync"
)

func worker(id int, tasks <-chan int, results chan<- int, wg *sync.WaitGroup) {
    defer wg.Done()

    for task := range tasks {
        // 执行任务
        result := task * 2

        // 将结果发送到结果通道
        results <- result
        fmt.Printf("Worker %d processed task %d\n", id, task)
    }
}

func main() {
    numTasks := 10
    numWorkers := 3

    // 创建任务通道和结果通道
    tasks := make(chan int, numTasks)
    results := make(chan int, numTasks)

    // 创建WaitGroup
    var wg sync.WaitGroup

    // 启动多个goroutine进行任务处理
    for i := 1; i <= numWorkers; i++ {
        wg.Add(1)
        go worker(i, tasks, results, &wg)
    }

    // 发送任务到任务通道
    for i := 1; i <= numTasks; i++ {
        tasks <- i
    }
    close(tasks)

    // 等待所有goroutine完成任务
    wg.Wait()

    // 关闭结果通道
    close(results)

    // 输出结果
    for result := range results {
        fmt.Println("Result:", result)
    }
}