Golang中如何使用defer和recover来捕获goroutine中的panic

defer和recover是Golang中处理panic的利器。它们允许你在程序发生崩溃时进行清理工作，并有机会恢复程序的运行。简单来说，defer用于延迟执行函数调用，而recover则用于捕获panic。

解决方案

在Golang中，defer和recover通常一起使用，以优雅地处理goroutine中的panic。

1. defer语句:

   defer

   登录后复制
   关键字用于注册一个函数调用，这个函数会在包含它的函数执行完毕（正常返回或发生panic）之后执行。 这保证了资源清理等操作总能被执行。 可以理解为一种延迟执行的机制，类似于其他语言中的

   finally

   登录后复制
   块，但更加灵活。

   立即学习“go语言免费学习笔记（深入）”；

2. recover函数:

   recover

   登录后复制
   是一个内建函数，它用于重新获得 panic 控制权。 当一个函数调用

   panic

   登录后复制
   时，正常的执行流程会被中断，并沿着调用栈向上寻找

   defer

   登录后复制
   语句。 如果

   defer

   登录后复制
   语句中调用了

   recover

   登录后复制
   ，那么

   panic

   登录后复制
   会被捕获，程序会从

   recover

   登录后复制
   调用的地方继续执行，

   recover

   登录后复制
   函数会返回

   panic

   登录后复制
   的值。 如果没有

   recover

   登录后复制
   被调用，

   panic

   登录后复制
   会继续向上层传递，最终导致程序崩溃。

一个典型的使用场景如下：

package main

import (
    "fmt"
    "runtime"
)

func main() {
    go func() {
        defer func() {
            if r := recover(); r != nil {
                fmt.Println("Recovered from panic:", r)
                // 打印堆栈信息，方便调试
                buf := make([]byte, 2048)
                runtime.Stack(buf, false)
                fmt.Printf("Stack trace:\n%s\n", buf)
            }
        }()

        // 模拟一个panic
        panic("Something went wrong!")
    }()

    // 避免程序立即退出，让goroutine有时间执行
    select {}
}

在这个例子中，我们在goroutine中使用

defer

recover

recover

recover

defer

为什么需要打印堆栈信息？ 仅仅捕获panic是不够的，还需要了解panic发生的原因。 堆栈信息可以帮助你定位到panic发生的具体位置，从而更好地解决问题。

副标题1：如何避免过度使用defer导致性能问题？

defer

defer

defer

避免过度使用

defer

• 只在必要时使用defer： 只在需要确保资源清理等操作必须执行的情况下使用

  defer

  登录后复制
  。 例如，关闭文件、释放锁等。
• 避免在循环中使用defer： 如果需要在循环中执行清理操作，尽量手动执行，而不是使用

  defer

  登录后复制
  。
• 考虑使用资源池： 对于需要频繁创建和销毁的资源，可以使用资源池来减少

  defer

  登录后复制
  的使用。

例如，下面是一个错误的使用

defer

func processData(data []int) {
    for _, value := range data {
        f, err := os.Open("data.txt") // 假设打开文件会panic
        if err != nil {
            panic(err)
        }
        defer f.Close() // 每次循环都defer一个close，效率很低

        // ... 处理文件内容
    }
}

更好的做法是：

func processData(data []int) {
    f, err := os.Open("data.txt")
    if err != nil {
        panic(err)
    }
    defer f.Close() // 只defer一次

    for _, value := range data {
        // ... 处理文件内容
    }
}

副标题2： recover() 返回 nil 的情况有哪些？

recover()

defer

panic

panic

recover()

nil

具体来说，以下情况

recover()

nil

千面视频动捕

千面视频动捕是一个AI视频动捕解决方案，专注于将视频中的人体关节二维信息转化为三维模型动作。

27

查看详情

• 没有发生panic： 如果当前 goroutine 没有发生

  panic

  登录后复制
  ，

  recover()

  登录后复制
  会返回

  nil

  登录后复制
  。
• recover() 不在 defer 函数中调用：

  recover()

  登录后复制
  必须在

  defer

  登录后复制
  函数中调用才有效。如果在

  defer

  登录后复制
  函数之外调用，

  recover()

  登录后复制
  会返回

  nil

  登录后复制
  。
• defer 函数没有被执行： 如果包含

  recover()

  登录后复制
  的

  defer

  登录后复制
  函数没有被执行（例如，函数提前返回），

  recover()

  登录后复制
  也会返回

  nil

  登录后复制
  。
• panic 被其他 recover() 捕获： 如果

  panic

  登录后复制
  已经被其他

  recover()

  登录后复制
  函数捕获，那么后续的

  recover()

  登录后复制
  函数也会返回

  nil

  登录后复制
  。

因此，在使用

recover()

nil

副标题3：如何优雅地处理panic并重启goroutine？

有时候，我们希望在goroutine发生panic后，不仅要捕获panic，还要重启goroutine，以保证程序的稳定性。 一种常见的做法是使用一个循环来不断地启动goroutine，并在goroutine内部使用defer和recover来捕获panic。

package main

import (
    "fmt"
    "time"
)

func worker() {
    defer func() {
        if r := recover(); r != nil {
            fmt.Println("Worker panicked:", r)
            // 重新启动worker goroutine
            go worker()
        }
    }()

    // 模拟worker goroutine的工作
    for i := 0; i < 5; i++ {
        fmt.Println("Worker doing work:", i)
        time.Sleep(time.Second)
        if i == 2 {
            panic("Worker encountered an error!")
        }
    }
}

func main() {
    // 启动worker goroutine
    go worker()

    // 避免程序立即退出
    select {}
}

在这个例子中，

worker

defer

recover

panic

worker

panic

recover

worker

worker

panic

这种重启机制的潜在问题： 如果panic是由于代码中的bug引起的，那么不断重启goroutine可能会导致程序陷入死循环。 因此，在重新启动goroutine之前，应该仔细分析panic的原因，并尝试修复bug。 此外，还可以添加一些限制，例如，限制goroutine重启的次数，以避免程序陷入死循环。

副标题4：defer 语句的执行顺序是什么？

在一个函数中，可以有多个

defer

defer

defer

defer

例如：

package main

import "fmt"

func main() {
    defer fmt.Println("First defer")
    defer fmt.Println("Second defer")
    defer fmt.Println("Third defer")

    fmt.Println("Main function")
}

输出结果：

Main function
Third defer
Second defer
First defer

理解

defer

副标题5：panic 和 os.Exit 的区别？

panic

os.Exit

• panic：

  panic

  登录后复制
  会导致程序中断正常的执行流程，并沿着调用栈向上寻找

  defer

  登录后复制
  语句。

  defer

  登录后复制
  语句会被执行，然后

  panic

  登录后复制
  会继续向上层传递，直到被

  recover

  登录后复制
  捕获，或者程序崩溃。

  panic

  登录后复制
  通常用于处理不可恢复的错误，例如，程序内部的逻辑错误、数据损坏等。
• os.Exit：

  os.Exit

  登录后复制
  会立即终止程序的执行，不会执行任何

  defer

  登录后复制
  语句。

  os.Exit

  登录后复制
  通常用于处理外部错误，例如，命令行参数错误、文件不存在等。

选择使用

panic

os.Exit

panic

os.Exit

总结

defer

recover

panic

defer

recover()

nil

panic

panic

以上就是Golang中如何使用defer和recover来捕获goroutine中的panic的详细内容，更多请关注php中文网其它相关文章！