GoLong的学习之路，进阶，语法之并发（并发错误处理）补充并发三部曲

news2024/9/23 15:24:20

这篇文章主要讲的是如何去处理并发的错误。

在Go语言中十分便捷地开启goroutine去并发地执行任务，但是如何有效的处理并发过程中的错误则是一个很棘手的问题。

文章目录

recover
errgroup

recover

哦对，似乎没写错误处理的文章。后面补上。

首先，这里的recover通常用来错误处理。

我们可以在代码中使用 recover 来会恢复程序中弹出的 panic，而 panic 只会触发当前 goroutine 中的 defer 操作。

func f1() {
	defer func() {
		if e := recover(); e != nil {
			fmt.Printf("recover panic:%v\n", e)
		}
	}()
	// 开启一个goroutine执行任务
	go func() {
		fmt.Println("in goroutine....")
		// 只能触发当前goroutine中的defer
		panic("panic in goroutine")
	}()

	time.Sleep(time.Second)
	fmt.Println("exit")
}
func main() {
	f1()
}

在这里插入图片描述
从输出结果可以看到程序并没有正常退出，而是由于 panic 异常退出了（exit code 2）。

正如上面示例演示的那样，在启用 goroutine 去执行任务的场景下，如果想要 recover goroutine中可能出现的 panic 就需要在 goroutine 中使用 recover。

在这里插入图片描述
程序中的 panic 被 recover 成功捕获，程序最终正常退出。

errgroup

我们通常在使用goroutine是调用一个函数或匿名函数

go func(){
  // ...
}

go foo()

并发的那些函数，其实很难拿的准。错误信息只多不少。

当我们想要将一个任务拆分成多个子任务交给多个 goroutine 去运行，这时我们该如何获取到子任务可能返回的错误呢？

// fetchUrlDemo 并发获取url内容
func FetchUrlDemo() {
	wg := sync.WaitGroup{}
	var urls = []string{
		"http://www.baidu.com",
		"http://www.qq.com",
		"http://www.nihaosfasdfasdf.com",
	}

	for _, url := range urls {
		wg.Add(1)
		go func(url string) {
			defer wg.Done()
			resp, err := http.Get(url)
			if err == nil {
				fmt.Printf("获取%s成功\n", url)

				resp.Body.Close()
			}
			return // 如何将错误返回呢？
		}(url)
	}
	wg.Wait()
	// 如何获取goroutine中可能出现的错误呢？
}

上面的示例代码中，我们开启了 3 个 goroutine 分别去获取3个 url 的内容。

类似这种将任务分为若干个子任务的场景会有很多，那么我们如何获取子任务中可能出现的错误呢？

errgroup 包就是为了解决这类问题而开发的.

它能为处理公共任务的子任务而开启的一组 goroutine 提供同步、error 传播和基于context 的取消功能。

errgroup 包中定义了一个 Group 类型，它包含了若干个不可导出的字段。

type Group struct {
	cancel func()

	wg sync.WaitGroup

	errOnce sync.Once
	err     error
}

errgroup.Group 提供了Go和Wait两个方法。

Go : func (g *Group) Go(f func() error)

Go 函数会在新的 goroutine 中调用传入的函数f。
第一个返回非零错误的调用将取消该Group；下面的Wait方法会返回该错误

Wait：func (g *Group) Wait() error

Wait 会阻塞直至由上述 Go 方法调用的所有函数都返回，然后从它们返回第一个非nil的错误（如果有）

// fetchUrlDemo2 使用errgroup并发获取url内容
func fetchUrlDemo2() error {

	g := new(errgroup.Group) // 创建等待组（类似sync.WaitGroup）
	var urls = []string{
		"http://www.4399.com",
		"http://www.baidu.com",
		"http://www.sdhfjoahoesrh.com",
	}
	for _, url := range urls {
		url := url // 注意此处声明新的变量
		// 启动一个goroutine去获取url内容
		g.Go(func() error {
			resp, err := http.Get(url)
			if err == nil {
				fmt.Printf("获取%s成功\n", url)
				resp.Body.Close()
			}
			return err // 返回错误
		})
	}
	if err := g.Wait(); err != nil {
		// 处理可能出现的错误
		fmt.Println(err)
		return err
	}
	fmt.Println("所有goroutine均成功")
	return nil
}

func main() {
	fetchUrlDemo2()

}