首页 > Golang > Go教程

Golang实现简单任务队列方法

时间：2025-10-28 21:18:57 295浏览收藏

今日不肯埋头，明日何以抬头！每日一句努力自己的话哈哈~哈喽，今天我将给大家带来一篇《Golang实现简单任务队列教程》，主要内容是讲解等等，感兴趣的朋友可以收藏或者有更好的建议在评论提出，我都会认真看的！大家一起进步，一起学习！

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

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

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

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

示例代码：

package main
<p>import (
"fmt"
"time"
)</p><p>// 任务函数类型
type Task func()</p><p>// 创建任务队列
func NewTaskQueue(workerCount int) chan<- Task {
taskChan := make(chan Task, 100) // 带缓冲的channel</p><pre class="brush:php;toolbar:false;">// 启动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会在处理完剩余任务后退出，确保不丢失任务。

支持优先级或延迟任务的扩展思路

如果需要更复杂的功能，比如优先级或延迟执行，可以引入heap或time.Timer。

简单延迟任务示例：

func DelayTask(queue chan

这样可以在指定延迟后自动提交任务。

基本上就这些。用channel + goroutine就能快速搭建一个可靠的任务队列，不复杂但容易忽略关闭和缓冲设置。根据实际需求调整worker数量和channel容量即可。

今天关于《Golang实现简单任务队列方法》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！

golang 任务队列 Goroutine channel WaitGroup