使用 Go 中的封装器处理恐慌

本文讨论了在 Go 中使用封装器处理恐慌。封装器是一种管理恐慌恢复的模式，它允许开发人员捕获和处理恐慌，而不会中断程序的执行。文章介绍了两种封装器：`Wrap` 和 `WrapWithSignal`。`Wrap` 封装器用于在发生恐慌时捕获和打印错误消息，而 `WrapWithSignal` 封装器则用于向给定的布尔通道发送信号，即使发生恐慌也是如此。

我遇到了以下代码，它使用包装器管理从恐慌中恢复。我知道恐慌不会在 goroutine 之间传播，它们必须独立管理，因此需要在发生恐慌的 goroutine 中进行处理。

package main

import (
    "fmt"
    "time"
)

func main() {
    go Wrap(test)()
    time.Sleep(time.Second)
    fmt.Println("HELLO")
}

func test() {
    panic("PANIC")
}

func Wrap(f func()) func() {
    return func() {
        defer func() {
            if r := recover(); r != nil {
                fmt.Printf("RECOVERED - %v\r\n", r)
            }
        }()
        f()
    }
}

func WrapWithSignal(f func(chan bool), signal chan bool) func() {
    return func() {
        defer func() {
            if r := recover(); r != nil {
                fmt.Printf("RECOVERED - %v\r\n", r)
                signal <- false
            }
        }()
        f(signal)
    }
}

1. wrapwithsignal 函数有什么用途，如何使用它？

2. 是否有另一种方法来包装函数以便恢复不那么冗长？

解决方案

在实现长时间运行的进程时，某些代码路径可能会导致恐慌。对于未初始化的映射和指针，以及在用户输入验证不力的情况下除以零的问题，这通常很常见。

在这些情况下程序完全崩溃通常比恐慌本身更糟糕，因此捕获和处理恐慌会很有帮助。

在大多数 web 应用程序中，捕获恐慌并在发生恐慌时发出 http.internalservererror 消息是很常见的。

在恢复过程中，您基本上可以做任何您想做的事情，尽管发出日志很常见。

1) wrap with signal 可用于从恐慌中恢复并向给定的布尔通道发送信号。

带信号的扭曲函数实现了包装模式，可用于包装信号通道并向通道发出事件，即使发生紧急情况也是如此。在这种情况下，如果不调用恢复，就会陷入死锁。

package main

import (
    "errors"
    "fmt"
    "math/rand"
    "time"
)

func main() {
    signal:= make(chan bool)
    go WrapWithSignal(play, signal)()

    if <-signal{
        fmt.Println("Congratulations, you win!")
        return
    }

    fmt.Println("You died.")
}

func play(signal chan bool){
    fmt.Println("Playing russian roulette")
    rand.Seed(time.Now().UnixNano())

    if rand.Intn(2) == 1{
        panic(errors.New("got a bullet"))
    }

    signal <- true
}

2）是否有另一种方法来包装函数以便恢复不那么冗长？

简短的回答：不，除非您使用软件包。理论上，包总是比较简洁。

长答案：根据具体情况，您可以创建一个 goroutine 包装器并基于等待组创建您自己的恐慌组。它在验证用户输入或执行并发操作并在最后收集所有错误时非常有用。将其拆分为一个包，您将得到冗长的内容。

只是一个示例：https://play.golang.org/p/on9AwZkvyIZ

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。