Golang goroutine池实现与管理示例-Golang-PHP中文网

Golang goroutine池实现与管理示例

P粉602998670

发布： 2025-10-05 19:34:02

原创

599人浏览过

答案：Go语言通过goroutine池控制并发，避免资源耗尽。核心组件包括任务队列、worker、协程池和关闭信号，使用channel实现任务分发。示例中NewPool创建池，Start启动worker，Submit提交任务，Stop优雅关闭。关键点为非阻塞提交、及时释放资源和扩展性设计。

golang goroutine池实现与管理示例

Go语言的goroutine轻量且高效，但无限制地创建可能导致内存暴涨或调度开销过大。使用goroutine池能有效控制并发数量，提升系统稳定性。下面介绍一种简单实用的goroutine池实现方式。

基本结构设计

一个典型的goroutine池包含以下几个核心组件：

任务队列：存放待执行的任务（通常为函数）
工作者（worker）：从队列中取出任务并执行
协程池（Pool）：管理worker数量和生命周期
控制信号：用于优雅关闭所有worker

通过channel实现任务分发和同步，是Go中最自然的方式。

简易goroutine池实现

以下是一个可复用的goroutine池示例：

立即学习“go语言免费学习笔记（深入）”；

乾坤圈新媒体矩阵管家

新媒体账号、门店矩阵智能管理系统

查看详情

package main

import (
    "fmt"
    "sync"
    "time"
)

// Task 表示一个可执行的任务
type Task func()

// Pool 协程池结构体
type Pool struct {
    workers   int           // 工作协程数
    taskQueue chan Task     // 任务队列
    wg        sync.WaitGroup
    closeChan chan struct{} // 关闭信号
}

// NewPool 创建新的协程池
func NewPool(workers, queueSize int) *Pool {
    return &Pool{
        workers:   workers,
        taskQueue: make(chan Task, queueSize),
        closeChan: make(chan struct{}),
    }
}

// Start 启动协程池
func (p *Pool) Start() {
    for i := 0; i < p.workers; i++ {
        p.wg.Add(1)
        go func() {
            defer p.wg.Done()
            for {
                select {
                case task, ok := <-p.taskQueue:
                    if !ok {
                        return // 通道已关闭
                    }
                    task()
                case <-p.closeChan:
                    return
                }
            }
        }()
    }
}

// Submit 提交任务到池中
func (p *Pool) Submit(task Task) bool {
    select {
    case p.taskQueue <- task:
        return true
    case <-p.closeChan:
        return false
    }
}

// Stop 停止协程池
func (p *Pool) Stop() {
    close(p.closeChan)
    close(p.taskQueue)
    p.wg.Wait()
}

登录后复制

使用示例

下面演示如何使用上述协程池处理一批任务：

func main() {
    pool := NewPool(3, 10) // 3个worker，最多缓存10个任务
    pool.Start()

    // 提交20个任务
    for i := 0; i < 20; i++ {
        id := i
        task := func() {
            fmt.Printf("执行任务 %d，运行于协程: %d\n", id, id%3)
            time.Sleep(500 * time.Millisecond) // 模拟耗时操作
        }
        pool.Submit(task)
    }

    // 等待一段时间后停止池
    time.Sleep(2 * time.Second)
    pool.Stop()
    fmt.Println("协程池已停止")
}

登录后复制

输出会显示任务被3个worker轮流执行，总耗时远小于串行执行时间。