前言

错误处理是所有编程语言都需要面对的一个问题,golang中,对待普通的错误通常使用error类型+返回值进行处理,对待更严重的错误,通常使用panic和recover进行处理。在本文中,着重于error类型的相关处理操作。

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来进行处理,下篇文章会进行分析。