如何在Golang中实现一个简单的任务队列

package main
<p>import (
"fmt"
"time"
)</p><p>// 任务函数类型
type Task func()</p><p>// 创建任务队列
func NewTaskQueue(workerCount int) chan<- Task {
taskChan := make(chan Task, 100) // 带缓冲的channel</p><pre class='brush:php;toolbar:false;'>// 启动worker
for i := 0; i < workerCount; i++ {
    go func() {
        for task := range taskChan {
            task() // 执行任务
        }
    }()
}

return taskChan

// 提交任务
for i := 1; i <= 5; i++ {
    i := i
    task := func() {
        fmt.Printf("处理任务 %d\n", i)
        time.Sleep(time.Second)
    }
    queue <- task
}

// 等待一段时间让任务执行完
time.Sleep(6 * time.Second)

func NewTaskQueueWithClose(workerCount int) (chan<- Task, func()) {
    taskChan := make(chan Task, 100)
    var wg sync.WaitGroup
<pre class='brush:php;toolbar:false;'>for i := 0; i < workerCount; i++ {
    wg.Add(1)
    go func() {
        defer wg.Done()
        for task := range taskChan {
            task()
        }
    }()
}

// 返回关闭函数
closeFunc := func() {
    close(taskChan)
    wg.Wait() // 等待所有worker完成
}

return taskChan, closeFunc

func DelayTask(queue chan<- Task, delay time.Duration, task Task) {
    time.AfterFunc(delay, func() {
        queue <- task
    })
}