租用问题

质量为本、客户为根、勇于拼搏、务实创新

< 返回租用问题列表

golang异步任务队列怎么实现,gin异步任务

发布时间:2023-10-26 19:49:05

golang异步任务队列怎样实现

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

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