Go语言中“Must”模式的实现与应用：利用泛型简化错误处理

// Must 是一个泛型辅助函数，用于处理返回 (T, error) 签名的函数。
// 如果传入的 error 不为 nil，它将触发 panic。
// 否则，它返回 T 类型的值。
func Must[T any](obj T, err error) T {
    if err != nil {
        panic(err)
    }
    return obj
}

package main

import (
    "fmt"
    "errors"
)

// Must 是一个泛型辅助函数，用于处理返回 (T, error) 签名的函数。
// 如果传入的 error 不为 nil，它将触发 panic。
// 否则，它返回 T 类型的值。
func Must[T any](obj T, err error) T {
    if err != nil {
        panic(err)
    }
    return obj
}

// success 模拟一个总是成功的函数
func success() (int, error) {
    return 0, nil
}

// fail 模拟一个总是失败的函数
func fail() (int, error) {
    return -1, errors.New("操作失败：这是一个模拟错误")
}

func main() {
    // 成功案例：n1 将被赋值为 0
    n1 := Must(success())
    fmt.Printf("成功执行结果: %d\n", n1)

    // 失败案例：当 fail() 返回非 nil 错误时，Must 会触发 panic
    // 这行代码将导致程序中断，下面的 fmt.Println(n2) 不会被执行
    // 为了演示，我们将其放在 try-catch 模拟块中，但在实际Go代码中，通常不会这样捕获
    // 而是让 panic 终止程序或由 recover 机制处理（例如在 defer 函数中）
    fmt.Println("尝试执行失败操作...")
    func() {
        defer func() {
            if r := recover(); r != nil {
                fmt.Printf("捕获到 panic: %v\n", r)
            }
        }()
        var n2 int = Must(fail()) // 这里会 panic
        fmt.Printf("失败执行结果 (此行不会被打印): %d\n", n2)
    }()

    fmt.Println("程序继续执行 (如果 panic 被 recover)")
}

// Must2 是一个泛型辅助函数，用于处理返回 (T1, T2, error) 签名的函数。
// 如果传入的 error 不为 nil，它将触发 panic。
// 否则，它返回 T1 和 T2 类型的值。
func Must2[T1 any, T2 any](obj1 T1, obj2 T2, err error) (T1, T2) {
    if err != nil {
        panic(err)
    }
    return obj1, obj2
}

// 模拟一个返回两个值和一个错误的函数
func fetchData() (string, int, error) {
    // 假设这里进行了一些数据获取操作
    return "Go语言", 2023, nil
}

// 模拟一个会失败的 fetchData 版本
func fetchDataFail() (string, int, error) {
    return "", 0, errors.New("数据获取失败")
}

func main() {
    // ... (之前的 Must 函数和 main 函数内容) ...

    fmt.Println("\n--- 演示 Must2 函数 ---")
    // 成功案例
    data, year := Must2(fetchData())
    fmt.Printf("获取到数据: %s, 年份: %d\n", data, year)

    // 失败案例
    fmt.Println("尝试执行失败的数据获取操作...")
    func() {
        defer func() {
            if r := recover(); r != nil {
                fmt.Printf("捕获到 panic (Must2): %v\n", r)
            }
        }()
        _, _ = Must2(fetchDataFail()) // 这里会 panic
        fmt.Println("此行不会被打印")
    }()
}