golang异步任务队列怎么实现

avatar
作者
猴君
阅读量: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来等待所有任务完成,我们可以确保在所有任务执行完之前,程序不会退出。当所有任务都完成后,我们关闭任务队列并结束程序。

这样,我们就实现了一个简单的异步任务队列。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!