深入理解Go语言中defer、panic与recover的错误处理机制

package main

import (
    "errors"
    "fmt"
    "runtime/debug" // 用于在 panic 发生时打印堆栈信息
)

// report 结构体用于示例
type report struct {
    data map[string]float64
}

// generateReport 尝试生成报告，并演示如何从 panic 中恢复并返回 error
// 注意：使用命名返回值 (rep *report, err error) 是关键
func generateReport(filename string) (rep *report, err error) {
    // 初始化 report，如果 panic 发生，可能需要将其置为 nil
    rep = &report{
        data: make(string)float64),
    }

    // defer 函数将在 generateReport 返回前执行
    defer func() {
        if r := recover(); r != nil {
            // 捕获到 panic，打印堆栈信息有助于调试
            fmt.Printf("在 generateReport 中捕获到 panic: %v\n", r)
            debug.PrintStack() // 打印完整的堆栈信息

            // 根据 panic 的类型设置 err
            switch x := r.(type) {
            case string:
                // 如果 panic 的值是字符串，将其包装成 error
                err = errors.New(fmt.Sprintf("报告处理错误: %s", x))
            case error:
                // 如果 panic 的值已经是 error 类型，直接赋值
                err = x
            default:
                // 处理其他未知类型的 panic
                err = errors.New(fmt.Sprintf("未知错误类型: %v", x))
            }
            // 如果发生 panic 并返回错误，通常需要将成功的返回值置为零值或 nil
            // 这里将 rep 置为 nil，表示报告生成失败
            rep = nil
        }
    }()

    // 模拟可能导致 panic 的情况
    if filename == "bad_format.txt" {
        panic("报告格式无法识别。") // 模拟字符串类型的 panic
    }
    if filename == "runtime_error.txt" {
        // 模拟一个运行时错误，例如越界访问
        var s []int
        _ = s[0] // 这会引发运行时 panic
    }
    if filename == "custom_error_panic.txt" {
        // 模拟一个自定义 error 类型的 panic
        panic(errors.New("自定义报告解析失败"))
    }

    // 正常业务逻辑
    rep.data["metric1"] = 100.5
    rep.data["metric2"] = 200.3

    fmt.Printf("报告 '%s' 生成成功。\n", filename)
    return rep, nil // 正常返回
}

func main() {
    // 示例1: 正常情况
    rep1, err1 := generateReport("good_report.txt")
    if err1 != nil {
        fmt.Printf("处理 good_report.txt 失败: %v\n", err1)
    } else {
        fmt.Printf("成功处理 good_report.txt, 报告数据: %v\n", rep1.data)
    }
    fmt.Println("---")

    // 示例2: 模拟字符串 panic
    rep2, err2 := generateReport("bad_format.txt")
    if err2 != nil {
        fmt.Printf("处理 bad_format.txt 失败: %v\n", err2)
    } else {
        fmt.Printf("成功处理 bad_format.txt, 报告数据: %v\n", rep2.data)
    }
    fmt.Println("---")

    // 示例3: 模拟运行时 panic (error 类型)
    rep3, err3 := generateReport("runtime_error.txt")
    if err3 != nil {
        fmt.Printf("处理 runtime_error.txt 失败: %v\n", err3)
    } else {
        fmt.Printf("成功处理 runtime_error.txt, 报告数据: %v\n", rep3.data)
    }
    fmt.Println("---")

    // 示例4: 模拟自定义 error panic
    rep4, err4 := generateReport("custom_error_panic.txt")
    if err4 != nil {
        fmt.Printf("处理 custom_error_panic.txt 失败: %v\n", err4)
    } else {
        fmt.Printf("成功处理 custom_error_panic.txt, 报告数据: %v\n", rep4.data)
    }
    fmt.Println("---")
}