Go语言协程并发控制与死锁避免详解
在Go语言中，利用goroutine实现并发任务处理时，常常需要限制协程数量以防止资源耗尽。然而，不当的限制机制可能导致死锁。本文将探讨如何在限制协程数量的同时，有效避免死锁，并确保从协程中顺利接收数据。
问题描述：
使用sync.WaitGroup和通道c来限制同时运行的协程数量，但将数据发送到另一个通道creport时，出现死锁错误：“fatal error: all goroutines are asleep - deadlock!”。
立即学习“go语言免费学习笔记（深入）”；
错误代码示例（简化版）：
（原文提供的代码示例存在逻辑错误，此处提供一个更清晰的简化版，以便说明问题和解决方案）
package main
import (
"fmt"
"sync"
)
func worker(id int, c chan struct{}, creport chan int) {
c <- struct{}{} // 获取令牌，进入处理
defer func() { <-c }() // 释放令牌，离开处理
// 模拟任务处理
result := id * 2
creport <- result
}
func main() {
c := make(chan struct{}, 3) // 限制同时运行3个协程
creport := make(chan int)
var wg sync.WaitGroup
for i := 0; i < 10; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
worker(i, c, creport)
}(i)
}
wg.Wait() // 等待所有协程完成
close(creport) // 关闭creport通道，防止死锁
for result := range creport {
fmt.Println("Result:", result)
}
}
死锁原因分析：
此例中，creport通道未被关闭，main函数中的for...range creport语句试图从一个未关闭的通道中读取数据。如果所有worker goroutine都已完成，但main函数仍在等待creport通道中的数据，就会发生死锁。
解决方案：
为了避免死锁，必须在所有worker goroutine完成任务后，关闭creport通道。 wg.Wait()确保所有协程都已完成，之后再关闭creport。 修改后的代码如下：
package main
import (
"fmt"
"sync"
)
// ... (worker function remains the same) ...
func main() {
// ... (channel creation remains the same) ...
for i := 0; i < 10; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
worker(i, c, creport)
}(i)
}
wg.Wait() // 等待所有协程完成
close(creport) // 关闭creport通道
for result := range creport {
fmt.Println("Result:", result)
}
}
通过在wg.Wait()之后关闭creport通道，for...range creport循环能够正常结束，避免了死锁。 这确保了所有数据都被正确接收，并且不会发生死锁。 记住，正确地管理通道的关闭对于避免Go语言中的死锁至关重要。
