Golang函数的错误处理策略

Go 语言提供了以下错误处理机制：直接处理错误：在函数体中检查 err 变量。使用 defer：在函数退出时执行代码，无论函数是否返回正常或错误，用于清理资源或打印错误消息。使用 recover：捕获运行时恐慌，通常是由未处理的错误引起的。自定义类型和错误包装：创建自定义错误类型并使用 errors.Is 和 errors.As 函数包装其他错误，允许类型断言。

Go 函数的错误处理策略

Go 语言提供了强大的错误处理机制，允许开发人员以清晰、统一的方式处理错误。本文将介绍 Go 中错误处理的不同策略，并通过实战案例进行演示。

错误值的定义和返回

在 Go 中，错误值由 error 接口表示，它只是一个实现了 Error() 方法的类型。错误值通常通过 nil 或 errors.New() 函数来创建。以下是如何定义和返回一个错误值：

func myFunction() error {
    return errors.New("error message")
}

直接处理错误

最简单的方法是直接处理错误，方法是在函数体中检查 err 变量：

func main() {
    if err := myFunction(); err != nil {
        log.Println(err)
    }
}

使用 defer

defer 语句允许在函数退出时执行代码，无论该函数是否返回正常或错误。这可以用于清理资源或打印错误消息：

func main() {
    defer func() {
        if err := recover(); err != nil {
            log.Println(err)
        }
    }()

    myFunction()
}

使用 recover

recover 函数可以捕获运行时恐慌，它通常是由未处理的错误引起的。可以通过在 main 函数中调用 recover 来处理未处理的错误：

func main() {
    if err := recover(); err != nil {
        log.Println(err)
    }

    myFunction()
}

自定义类型和错误包装

对于复杂场景，可以创建自定义错误类型并使用 errors.Is 和 errors.As 函数包装其他错误。这允许在错误层次结构中进行类型断言：

type MyError struct {
    error
}

func main() {
    if err := myFunction(); err != nil {
        if myError, ok := err.(MyError); ok {
            log.Println(myError)
        } else {
            log.Println(err)
        }
    }
}

实战案例

考虑一个需要连接到数据库并执行查询的应用程序。应用程序可以按照以下方式处理错误：

func main() {
    db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/database")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    rows, err := db.Query("SELECT * FROM table")
    if err != nil {
        log.Println(err)
        return
    }
    defer rows.Close()

    for rows.Next() {
        var name string
        if err := rows.Scan(&name); err != nil {
            log.Println(err)
            continue
        }
        log.Println(name)
    }
}

应用程序使用 defer 语句来确保在遇到错误时正确关闭数据库和查询句柄。它还使用直接错误处理来打印错误消息并继续处理，以避免因单个错误而中断整个应用程序。