Golang并发调度器：Go WaitGroup的优化实现

package main

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

// 定义任务结构体
type Task struct {
    TaskID int // 任务ID
}

func main() {
    var (
        tasksNumber = 100 // 待执行任务数量
        goroutineNum = 10 // goroutine数量
        wg sync.WaitGroup
        taskQueue = make(chan Task, tasksNumber) // 任务队列
    )
    // 初始化任务队列
    for i := 0; i < tasksNumber; i++ {
        task := Task{
            TaskID: i,
        }
        taskQueue <- task
    }
    close(taskQueue)

    // 启动goroutine
    for i := 0; i < goroutineNum; i++ {
        wg.Add(1)
        go func(id int) {
            defer wg.Done()
            for task := range taskQueue {
                execute(task)
            }
        }(i)
    }
    wg.Wait()
}

// 任务的具体执行函数
func execute(task Task) {
    fmt.Printf("TaskID: %d, Now: %s
", task.TaskID, time.Now().Format("2006-01-02 15:04:05"))
    time.Sleep(1 * time.Second)
    fmt.Printf("TaskID: %d, Finished
", task.TaskID)
}