错误处理是所有编程语言都需要面对的一个问题,golang中,对待普通的错误通常使用error类型+返回值进行处理,对待更严重的错误,通常使用panic和recover进行处理。在本文中,着重于error类型的相关处理操作。
error类型是Golang内置类型之一,其本质上只是一个接口,所以只要实现了这个接口,就可以是error类型了,后续自定义错误类型就是这个原理:
type error interface {
Error() string
}
Go 语言内置了一个 errors 包,可以用来创建和处理错误。可以使用 errors.New() 函数创建一个简单的错误,如下所示:
package main
import (
"errors"
"fmt"
)
func divide(a, b int) (int, error) {
if b == 0 {
return 0, errors.New("division by zero")
}
return a / b, nil
}
func main() {
result, err := divide(10, 0)
if err != nil {
fmt.Println(err)
return
}
fmt.Println(result)
}
在上面的例子中,定义了一个函数 divide(),它接受两个int类型,并返回计算结果。如果 b 等于零,则函数使用errors.New()
新建一个错误,其中包含字符串 “division by zero”。在调用此函数时,判断err是否为空,来确定程序是否报错了。
有时候,errors.New()
并不能满足复杂的业务环境,所以需要自定义错误类型,原理就是实现error的接口即可,如下所示:
package main
import (
"fmt"
)
type DivideError struct {
dividend int
divisor int
}
func NewDivideError(a,b int)*DivideError {
return &DivideError{dividend: a,divisor: b}
}
func (e *DivideError) Error() string {
return fmt.Sprintf("cannot divide %d by %d", e.dividend, e.divisor)
}
func divide(a, b int) (int, error) {
if b == 0 {
return 0, NewDivideError(a,b)
}
return a / b, nil
}
func main() {
result, err := divide(10, 0)
if err != nil {
if e, ok := err.(*DivideError); ok {
fmt.Println("Divide error:", e)
return
}
fmt.Println(err)
return
}
fmt.Println(result)
}
上述代码中,定义了一个自定义错误类型 DivideError
,并实现了Error方法,所以此结构体可以直接作为error类型进行返回。在主函数中,也可以正常进行处理。
常用的错误处理方式在上文的例子中已经提到了,即 使用if
来判断错误是否为空,如果不为空,就会做一些处理,这也是最常用的一种方式。
我们还可以使用defer来进行辅助进行错误的处理及清理资源,如在退出前关闭文件、数据库操作、锁等等,关于文件的操作例子如下:
func readFile(filename string) (string, error) {
f, err := os.Open(filename)
if err != nil {
return "", err
}
defer f.Close()
content, err := ioutil.ReadAll(f)
if err != nil {
return "", err
}
return string(content), nil
}
注意,defer的运行顺序是类似栈的顺序,即先进后出。
总的来说,GO的错误处理是比较轻量的,使用error类型即可处理大部分的错误了。但是还有一些极端严重的错误需要使用panic和recover来进行处理,下篇文章会进行分析。