在Go语言中如何限制协程数量并避免死锁问题？-Golang-PHP中文网

在Go语言中如何限制协程数量并避免死锁问题？

花韻仙語

发布： 2025-03-19 10:56:01

原创

1065人浏览过

在go语言中如何限制协程数量并避免死锁问题？

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语言中的死锁至关重要。

以上就是在Go语言中如何限制协程数量并避免死锁问题？的详细内容，更多请关注php中文网其它相关文章！