阅读量:2
在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
来等待所有任务完成,我们可以确保在所有任务执行完之前,程序不会退出。当所有任务都完成后,我们关闭任务队列并结束程序。
这样,我们就实现了一个简单的异步任务队列。