Golang如何实现任务并行处理

Go语言通过goroutine和channel实现高效并发编程,核心是用goroutine执行任务、channel协调数据。02. 启动goroutine只需go关键字调用函数,但需注意main函数退出会终止所有协程。03. 使用sync.WaitGroup可可靠等待所有任务完成:启动前wg.Add(1),结束时wg.Done(),主协程wg.Wait()阻塞直至完成。

Go语言通过goroutinechannel天然支持并发编程,实现任务并行处理非常高效且简洁。核心思路是将多个任务交给不同的goroutine执行,通过channel协调数据传递与同步。

使用Goroutine启动并行任务

每个goroutine是一个轻量级线程,由Go运行时调度。你只需用go关键字启动函数即可:

  • 定义一个任务函数,比如处理数据或调用API
  • 在主逻辑中用go task()并发执行多个实例
  • 注意:main函数退出时所有goroutine会强制终止,需确保它们能正常完成

示例:

func doTask(id int) {
    fmt.Printf("任务 %d 开始\n", id)
    time.Sleep(1 * time.Second)
    fmt.Printf("任务 %d 完成\n", id)
}

func main() { for i := 0; i < 5; i++ { go doTask(i) } time.Sleep(2 * time.Second) // 等待任务完成(临时方案) }

使用WaitGroup等待所有任务完成

sync.WaitGroup用于等待一组并发任务结束,比sleep更可靠。

  • 在启动每个goroutine前调用wg.Add(1)
  • 在每个任务结尾调用wg.Done()
  • 主协程调用wg.Wait()阻塞直到所有任务完成

改进后的例子:

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 5; i++ {
        wg.Add(1)
        go func(id int) {
            defer wg.Done()
            doTask(id)
        }(i)
    }
    wg.Wait()
    fmt.Println("所有任务已完成")
}

通过Channel传递结果与控制并发

channel不仅用于通信,还能有效控制资源使用和收集返回值。

  • 创建带缓冲的channel作为任务队列
  • 多个worker goroutine从channel读取任务并处理
  • 主线程发送任务后关闭channel,等待结果返回

典型工作池模式:

func worker(id int, jobs <-chan int, results chan<- int) {
    for job := range jobs {
        fmt.Printf("工人 %d 处理任务 %d\n", id, job)
        time.Sleep(time.Second)
        results <- job * 2
    }
}

func main() { jobs := make(chan int, 10) results := make(chan int, 10)

// 启动3个worker
for w := 1; w <= 3; w++ {
    go worker(w, jobs, results)
}

// 发送5个任务
for j := 1; j <= 5; j++ {
    jobs <- j
}
close(jobs)

// 收集结果
for r := 1; r <= 5; r++ {
    <-results
}

}

基本上就这些。Go的并发模型简单但强大,合理使用goroutine、channel和WaitGroup就能实现高效的任务并行处理,不复杂但容易忽略细节如资源释放和错误处理。