在Go语言中,可以使用goroutine和channel来实现异步任务队列。下面是一个简单的示例代码:
package main
import (
"fmt"
"sync"
"time"
)
func main() {
// 创建一个无缓冲的channel,用于任务队列
taskQueue := make(chan func())
// 使用sync.WaitGroup来等待所有任务完成
var wg sync.WaitGroup
// 启动消费者goroutine,不断从任务队列中取出任务并执行
go func() {
for task := range taskQueue {
task()
wg.Done()
}
}()
// 添加一些任务到任务队列
for i := 0; i < 10; i++ {
// 增加等待的任务数量
wg.Add(1)
// 创建一个任务,并将其添加到任务队列
task := func() {
fmt.Println("执行任务", i)
time.Sleep(1 * time.Second) // 模拟任务执行时间
}
taskQueue <- task
}
// 等待所有任务完成
wg.Wait()
// 关闭任务队列
close(taskQueue)
fmt.Println("所有任务已完成")
}
在上面的示例代码中,我们创建了一个无缓冲的channel `taskQueue` 作为任务队列,通过 `taskQueue <- task` 将任务添加到队列中。然后,我们启动一个消费者goroutine来不断从任务队列中取出任务并执行,直到任务队列被关闭。
通过使用`sync.WaitGroup`来等待所有任务完成,我们可以确保在所有任务执行完之前,程序不会退出。当所有任务都完成后,我们关闭任务队列并结束程序。
这样,我们就实现了一个简单的异步任务队列。