解决 Go 并发求和中的通道死锁：Range 与计数器方案

1. Go 并发求和场景与初始实现

在 go 语言中，利用 goroutine 和 channel 实现并发任务处理是常见的模式。例如，将一个大型数组分成多个部分，由不同的 goroutine 并发计算各部分的和，最后通过 channel 汇总结果。

考虑以下场景：一个整数数组 a 需要计算总和。我们将其分成两部分，并启动两个 Goroutine 分别计算这两部分的总和，然后将结果发送到一个共享的 Channel 中。主 Goroutine 从 Channel 接收这两个结果并相加，得到最终的总和。

初始实现代码如下：

package main

import (
    "fmt"
)

// Add 函数计算切片a中所有元素的和，并将结果发送到res通道。
func Add(a []int, res chan<- int) {
    sum := 0
    for _, v := range a {
        sum += v
    }
    res <- sum // 将计算结果发送到通道
}

func main() {
    a := []int{1, 2, 3, 4, 5, 6, 7}
    n := len(a)
    ch := make(chan int) // 创建一个无缓冲通道

    // 启动两个Goroutine并发计算
    go Add(a[:n/2], ch)
    go Add(a[n/2:], ch)

    sum := 0
    // 尝试使用range循环从通道接收数据
    for s := range ch {
        sum += s
    }
    // close(ch) // 初始代码中此处被注释或缺失

    fmt.Println(sum)
}

2. 死锁问题分析：Range 循环与通道关闭

上述代码在运行时会发生死锁。其根本原因在于主 Goroutine 中 for s := range ch 语句的阻塞行为，以及通道 ch 从未被关闭。

• for range 循环的工作机制： 当使用 for range 循环从一个 Channel 接收数据时，它会持续尝试接收，直到 Channel 被关闭。一旦 Channel 被关闭，for range 循环就会退出。
• 死锁的发生： 在上述代码中，两个 Add Goroutine 完成计算并将结果发送到 ch 后，它们会自然退出。然而，ch 通道并没有被任何 Goroutine 关闭。主 Goroutine 在接收到两个结果后，for s := range ch 会继续等待第三个值。由于没有任何 Goroutine 会再向 ch 发送数据，并且 ch 也未被关闭，主 Goroutine 将无限期地等待下去，导致程序死锁。

为了解决这个问题，通常需要确保在所有发送操作完成后，通道会被关闭。然而，在有多个发送方的情况下，确定由哪个发送方来关闭通道是一个复杂且容易出错的问题（例如，过早关闭或重复关闭会导致 panic）。

3. 解决方案：基于计数器的通道接收

当不确定何时关闭通道，或者有多个发送方时，一种更健壮的方法是不关闭通道，而是通过其他机制来判断何时停止接收。这里介绍一种基于计数器的解决方案，它通过跟踪已完成的 Goroutine 数量来管理接收过程。

核心思路是：

1. 明确知道有多少个 Goroutine 会向通道发送数据。
2. 在主 Goroutine 中，使用一个循环，迭代固定次数（即发送方的数量）从通道接收数据。
3. 每次成功接收一个值，就递增一个计数器，直到计数器达到预设的发送方数量。

修正后的代码示例：

package main

import (
    "fmt"
)

// Add 函数计算切片a中所有元素的和，并将结果发送到res通道。
func Add(a []int, res chan<- int) {
    sum := 0
    for _, v := range a {
        sum += v
    }
    res <- sum // 将计算结果发送到通道
}

func main() {
    a := []int{1, 2, 3, 4, 5, 6, 7}
    n := len(a)
    ch := make(chan int) // 创建一个无缓冲通道

    // 启动两个Goroutine并发计算
    go Add(a[:n/2], ch)
    go Add(a[n/2:], ch)

    sum := 0
    count := 0 // 初始化计数器，用于跟踪已接收的结果数量

    // 循环接收数据，直到接收到预期的所有结果（这里是2个）
    for count < 2 {
        s := <-ch // 从通道接收一个值
        sum += s
        count++ // 递增计数器
    }
    // 当count达到2时，循环结束，所有预期结果都已接收

    fmt.Println(sum)
}

4. 代码解析与运行结果

在修正后的 main 函数中：

MindShow

MindShow官网 | AI生成PPT，快速演示你的想法

1492

查看详情

• 我们不再使用 for s := range ch 循环。
• 引入了一个整型变量 count，初始化为 0。
• 使用 for count < 2 循环来控制接收过程。由于我们明确知道有两个 Add Goroutine 会向 ch 发送数据，所以当 count 达到 2 时，意味着所有预期的结果都已接收完毕。
• 在循环体内部，s := <-ch 会阻塞直到有数据可读。一旦接收到一个值，sum 会更新，并且 count 会递增。
• 当 count 达到 2 后，循环终止，程序继续执行 fmt.Println(sum) 打印最终结果，而不会发生死锁。

运行此代码，将正确输出 28 (1+2+3+4+5+6+7)。

5. 注意事项与最佳实践

• 选择合适的通道接收方式：

  • 当只有一个发送方，并且发送方明确知道何时完成所有发送时，close(channel) 后使用 for range channel 是简洁有效的。
  • 当有多个发送方，或者发送方不应负责关闭通道时，应避免使用 for range 循环，转而使用计数器、sync.WaitGroup 或其他同步机制来协调接收。

• 通道关闭的风险：

  • 向已关闭的通道发送数据会导致 panic。
  • 关闭一个已关闭的通道会导致 panic。
  • 从已关闭的通道接收数据不会阻塞，而是立即返回零值和 false（表示通道已关闭）。

• sync.WaitGroup 的应用： 对于更复杂的并发场景，sync.WaitGroup 是一个更通用的同步原语，用于等待一组 Goroutine 完成。它可以与计数器方法结合使用，或单独用于确保所有工作 Goroutine 都已完成，然后再进行最终结果的汇总或通道关闭（如果需要）。例如：

  // ... (Add 函数不变)
  func main() {
      a := []int{1, 2, 3, 4, 5, 6, 7}
      n := len(a)
      ch := make(chan int)
      var wg sync.WaitGroup // 引入WaitGroup
  
      wg.Add(2) // 告知WaitGroup有两个Goroutine要等待
  
      go func() {
          defer wg.Done() // Goroutine完成时调用Done
          Add(a[:n/2], ch)
      }()
      go func() {
          defer wg.Done() // Goroutine完成时调用Done
          Add(a[n/2:], ch)
      }()
  
      // 启动一个Goroutine来关闭通道，避免主Goroutine阻塞
      go func() {
          wg.Wait() // 等待所有Add Goroutine完成
          close(ch) // 所有发送方完成后关闭通道
      }()
  
      sum := 0
      for s := range ch { // 现在可以安全地使用range循环
          sum += s
      }
      fmt.Println(sum)
  }

  登录后复制

  这种 sync.WaitGroup 配合 close(ch) 的模式在多发送方场景中更为常见，它将关闭通道的责任从发送方转移到一个专门的 Goroutine，并在所有发送方完成后执行关闭。

6. 总结

在 Go 语言并发编程中，理解 Channel 的工作原理，特别是 for range 循环对通道关闭的依赖，对于避免死锁至关重要。当有多个发送方时，直接在发送方中判断并关闭通道是困难且危险的。此时，采用基于计数器或 sync.WaitGroup 的策略来协调 Goroutine 的完成和通道数据的接收，是更安全和健壮的实践。这确保了程序在所有并发任务完成后能够正确地汇总结果并优雅地终止。

以上就是解决 Go 并发求和中的通道死锁：Range 与计数器方案的详细内容，更多请关注php中文网其它相关文章！