剖析 Go 协程并发情况下的错误处理策略

2023-11-19 18:44:32

并发中的错误处理：应对协程中通道关闭

在编写 Go 协程应用程序时，错误处理至关重要，以避免程序崩溃。本文探讨了在并发环境中处理协程通道关闭时常见的两种有效策略。

问题：通道关闭后的恐慌

考虑以下代码示例：

package main

import (
	"fmt"
	"time"
)

func main() {
	ch := make(chan int)
	go func() {
		for {
			select {
			case v := <-ch:
				fmt.Println(v)
			}
		}
	}()

	ch <- 1
	time.Sleep(1 * time.Second)
	close(ch)
}

此代码创建一个协程来不断接收来自通道 ch 的值。但是，如果协程在通道关闭后试图从通道中接收值，就会发生恐慌。这是因为通道关闭后接收操作会返回零值，通常不是预期行为。

解决方案 1：使用 select 语句

一种解决方案是使用 select 语句来显式处理通道关闭的情况：

package main

import (
	"fmt"
	"time"
)

func main() {
	ch := make(chan int)
	go func() {
		for {
			select {
			case v, ok := <-ch:
				if !ok {
					fmt.Println("Channel closed")
					return
				}
				fmt.Println(v)
			}
		}
	}()

	ch <- 1
	time.Sleep(1 * time.Second)
	close(ch)
}

在这里，select 语句有一个额外的 default 分支来处理通道关闭的情况。当通道关闭时，ok 标志将变为 false，select 语句将执行 default 分支并打印“Channel closed”消息。

解决方案 2：使用 recover 语句

另一种解决方案是使用 recover 语句来捕获通道关闭后发生的恐慌：

package main

import (
	"fmt"
	"time"
)

func main() {
	ch := make(chan int)
	go func() {
		defer func() {
			if err := recover(); err != nil {
				fmt.Println("Channel closed")
			}
		}()

		for {
			v := <-ch
			fmt.Println(v)
		}
	}()

	ch <- 1
	time.Sleep(1 * time.Second)
	close(ch)
}