如何在Golang中使用channel实现任务协调_Golang channel任务协调实践

channel可用于任务协调,如通过无缓冲channel实现完成信号,带缓冲channel控制并发数,结合context与select实现超时控制,并用close通知接收端退出,从而有效管理goroutine同步与资源清理。

在Go语言中,channel不仅是数据传递的工具,更是实现goroutine间任务协调的核心机制。通过合理使用channel,可以有效控制并发流程、同步执行顺序、避免竞态条件。下面结合常见场景说明如何用channel进行任务协调。

1. 等待任务完成(信号同步)

最简单的协调方式是使用无缓冲channel作为“完成信号”。主goroutine启动子任务后,等待其通过channel发送完成通知。

done := make(chan bool)

go func() { // 模拟耗时任务 time.Sleep(2 * time.Second) done <- true // 任务完成,发送信号 }()

fmt.Println("任务开始...") <-done // 阻塞等待任务结束 fmt.Println("任务完成")

这种方式替代了WaitGroup,在简单场景下更直观。

2. 控制并发数(工作池模式)

当需要限制同时运行的goroutine数量时,可用带缓冲的channel作为“许可证”控制并发。

semaphore := make(chan struct{}, 3) // 最多3个并发

for i := 0; i < 10; i++ { go func(id int) { semaphore <- struct{}{} // 获取许可 defer func() { <-semaphore }() // 释放许可

    fmt.Printf("任务 %d 开始\n", id)
    time.Sleep(time.Second)
    fmt.Printf("任务 %d 完成\n", id)
}(i)

}

该模式适用于爬虫、批量请求等需限流的场景。

3. 多任务协同与超时控制

多个任务并行执行,主逻辑等待全部完成或超时退出,可结合select和context实现。

ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
defer cancel()

tasks := 3 done := make(chan bool, tasks)

for i := 0; i < tasks; i++ { go func(id int) { select { case <-ctx.Done(): fmt.Printf("任务 %d 被取消\n", id) default: time.Sleep(time.Duration(rand.Intn(1500)) * time.Millisecond) done <- true } }(i) }

// 等待所有任务完成或超时 for i := 0; i < tasks; i++ { select { case <-done: fmt.Println("一个任务完成") case <-ctx.Done(): fmt.Println("等待超时,停止等待") break } }

这种组合方式能优雅处理失败和延迟任务。

4. 主动关闭与资源清理

使用close(channel)通知所有监听者“不再有数据”,配合range自动退出循环。

dataCh := make(chan int)
done := make(chan bool)

go func() { for num := range dataCh { // 自动检测channel关闭 fmt.Println("处理数据:", num) } done <- true }()

// 发送部分数据 for i := 0; i < 5; i++ { dataCh <- i } close(dataCh) // 关闭channel,触发接收端退出

<-done fmt.Println("所有数据处理完毕")

这是实现生产者-消费者模型的标准做法。

基本上就这些。channel的本质是同步点,合理设计它的容量和使用方式,就能灵活协调各种并发任务。关键在于明确谁发信号、谁接收、何时关闭,避免死锁和泄露。