Golang如何实现goroutine池-Golang-PHP中文网

Golang如何实现goroutine池

P粉602998670

发布： 2025-10-03 10:53:02

原创

859人浏览过

答案：Go语言中通过goroutine池可控制并发、复用资源，核心为channel传递任务与固定worker消费。示例实现包含任务队列、worker数量控制、启动关闭机制，使用有缓冲channel避免阻塞，select监听done信号实现优雅退出，具备并发安全与资源控制特性，适用于限制并发场景。

golang如何实现goroutine池

Go语言中的goroutine虽然轻量，但如果频繁创建大量goroutine，仍可能带来调度开销和资源浪费。使用goroutine池可以复用已创建的goroutine，控制并发数量，提升性能。实现一个简单的goroutine池并不复杂，核心思路是利用channel接收任务，由固定数量的worker持续从channel中取任务执行。

基本结构设计

一个基础的goroutine池通常包含以下几个部分：

任务队列：用channel传递待执行的函数
Worker数量：启动固定数量的goroutine来消费任务
启动与关闭机制：安全地启动和停止整个池

以下是一个简单但实用的实现示例：

package main

type Task func()

type Pool struct {
    tasks chan Task
    done  chan struct{}
}

// NewPool 创建一个新的goroutine池，指定worker数量和任务队列大小
func NewPool(workers, queueSize int) *Pool {
    pool := &Pool{
        tasks: make(chan Task, queueSize),
        done:  make(chan struct{}),
    }

    // 启动指定数量的worker
    for i := 0; i < workers; i++ {
        go func() {
            for {
                select {
                case task, ok := <-pool.tasks:
                    if !ok {
                        return
                    }
                    task()
                case <-pool.done:
                    return
                }
            }
        }()
    }

    return pool
}

// Submit 提交任务到池中
func (p *Pool) Submit(task Task) {
    p.tasks <- task
}

// Stop 停止所有worker，关闭任务队列
func (p *Pool) Stop() {
    close(p.done)
    close(p.tasks)
}

登录后复制

使用示例

下面是如何使用上面定义的Pool：

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

ViiTor实时翻译

AI实时多语言翻译专家！强大的语音识别、AR翻译功能。

116

查看详情

package main

import (
    "fmt"
    "time"
)

func main() {
    // 创建一个有3个worker，任务队列最多10个任务的池
    pool := NewPool(3, 10)

    // 提交一些任务
    for i := 0; i < 5; i++ {
        pool.Submit(func() {
            fmt.Printf("处理任务中...\n")
            time.Sleep(time.Second)
        })
    }

    // 模拟运行一段时间后关闭
    time.Sleep(2 * time.Second)
    pool.Stop()

    fmt.Println("任务池已停止")
}

登录后复制