如何限制 Go 中并发任务的数量-Golang-PHP中文网

如何限制 Go 中并发任务的数量

在处理大量数据（如 10 万行文本写入数据库）时，盲目启动等量 goroutine 会导致资源耗尽；应采用固定工作池模式，通过 channel 分发任务并控制并发数（如 100），兼顾性能与稳定性。

在处理大量数据（如 10 万行文本写入数据库）时，盲目启动等量 goroutine 会导致资源耗尽；应采用固定工作池模式，通过 channel 分发任务并控制并发数（如 100），兼顾性能与稳定性。

Go 语言中，无节制地启动数万个 goroutine（例如对每行文本启动一个 go saveToDB(l)）看似简单，实则存在严重隐患：它会迅速耗尽内存、压垮数据库连接池、触发操作系统级调度开销，甚至导致程序崩溃或服务不可用。真正的并发控制不在于“多”，而在于“稳”——即维持合理、可控的并发度。

推荐方案是构建一个 固定大小的工作池（Worker Pool），其核心思想是：预启动 N 个长期运行的 worker goroutine，所有任务通过 channel 统一分发，worker 持续从 channel 中接收并执行任务，直到 channel 关闭。

以下是完整、可直接运行的实现示例：

使用Go语言搭建家庭相册系统-相关课件

下载

package main

import (
    "fmt"
    "strings"
    "sync"
)

// 模拟实际数据库保存逻辑（请替换为真实 DB 调用）
func actuallySaveToDB(line string) error {
    // 示例：简单打印 + 模拟 I/O 延迟
    fmt.Printf("Saving: %s\n", line)
    return nil
}

// Worker 函数：持续从 channel 读取任务并执行
func saveToDB(ch <p>✅ <strong>关键设计说明：</strong>  </p>

ch 是一个 无缓冲或带缓冲的 channel，作为任务队列；带缓冲（如 make(chan string, 1000)）可缓解生产者/消费者速度差异，避免主 goroutine 阻塞。
saveToDB 使用 for line := range ch 语义：channel 关闭后循环自动终止，无需额外标志位或 select{default:} 判断。
defer wg.Done() 确保无论函数如何退出（含 panic），worker 都能正确通知 WaitGroup。
主流程清晰分离：启动池 → 投递任务 → 关闭通道 → 等待完成，逻辑内聚、易于维护。

⚠️ 注意事项：