当前位置：首页 > 文章列表 > Golang > Go教程 > Golang任务队列实现教程详解

Golang任务队列实现教程详解

2025-09-23 17:17:29 0浏览收藏

**Golang实现简单任务队列教程：轻量高效处理异步任务** 想用Golang轻松搞定异步任务？本教程将手把手教你利用goroutine和channel构建一个简单高效的任务队列。这种方式特别适合处理发送邮件、文件上传、定时任务等场景。我们将详细讲解如何使用channel和goroutine搭建基础任务队列，并通过示例代码展示如何创建、提交和并发执行任务。此外，还将介绍如何添加任务完成通知，以及如何优雅地关闭任务队列，确保程序稳定运行。掌握这些技巧，让你在Golang开发中如虎添翼，轻松应对各种异步处理需求。

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

在Golang中实现一个简单的任务队列，核心思路是利用goroutine和channel来实现并发任务的提交与执行。这种方式轻量、高效，适合处理异步任务，比如发送邮件、处理上传、定时任务等。

使用Channel和Goroutine构建基础任务队列

Go的channel天然适合做任务队列。定义一个任务函数类型，用channel接收任务，多个worker并发处理。

示例代码：

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

return taskChan
}
func main() {
queue := NewTaskQueue(3) // 3个worker
// 提交任务
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)
}

添加任务完成通知和优雅关闭

实际使用中，可能需要知道任务何时完成，或在程序退出时停止队列。

可以通过关闭channel并使用WaitGroup来管理生命周期。

func NewTaskQueueWithClose(workerCount int) (chanfor 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

}

调用close函数后，所有worker会在处理完剩余任务后退出，确保不丢失任务。