[golang note] 错误处理

错误处理

• 错误处理的标准模式

√ golang错误处理的标准模式：error接口。

√ golang函数如果要返回错误，规范上是将error作为多返回值中的最后一个，但这并非是强制要求。

▶ error接口

type error interface {
Error() string
}

▶ 内置的error类型使用

▪ 语法如下

func 函数名(参数列表) (返回值列表, err error) {
// 函数体
}

▪ 错误处理

例如我们有一个这样的函数：

func Foo(param int) (n int, err error) {
// 函数体
}

调用函数时建议按如下方式处理错误：

n, err := Foo(0)
if err != nil {
// 错误处理
} else {
// 使用返回值n
}

▪ 示例如下

package main

import (
"errors"
"fmt"
)

func divide(dividend float64, divisor float64) (result float64, err error) {
if divisor == 0 {
return -1, errors.New("除数为0")
}

return dividend / divisor, nil
}

func main() {
result, err := divide(1, 2)
if err != nil {
fmt.Println(err.Error())
} else {
fmt.Println("result =", result)
}

result, err = divide(1, 0)
if err != nil {
fmt.Println(err.Error())
} else {
fmt.Println("result =", result)
}
}

▶ 自定义error类型使用

√ golang错误处理支持自定义的error类型，只需要为自定义error类型实现Error接口即可。

▪ 语法如下

type CustomError struct {
...
}

func (e *CustomError) Error() string {
// 函数体
}

▪ 示例如下

package main

import (
"fmt"
)

type MathError struct {
Op   string
info string
}

func (e *MathError) Error() string {
return "Math operation " + e.Op + " error : " + e.info
}

func divide(dividend float64, divisor float64) (result float64, err error) {
if divisor == 0 {
return -1, &MathError{"division", "divisor is zero"}
}

return dividend / divisor, nil
}

func main() {
result, err := divide(1, 2)
if err != nil {
fmt.Println(err.Error())
} else {
fmt.Println("result =", result)
}

result, err = divide(1, 0)
if err != nil {
fmt.Println(err.Error())
} else {
fmt.Println("result =", result)
}
}

▪ 类型转换

√ 如果处理错误时需要获取详细信息，而不仅仅满足于打印一句错误信息，那就需要用到类型转换。

package main

import (
"fmt"
)

type MathError struct {
Op   string
info string
}

func (e *MathError) Error() string {
return "Math operation " + e.Op + " error : " + e.info
}

func divide(dividend float64, divisor float64) (result float64, err error) {
if divisor == 0 {
return -1, &MathError{"division", "divisor is zero"}
}

return dividend / divisor, nil
}

func main() {
result, err := divide(1, 0)
if err != nil {
// error类型转换为*MathError指针，因为接口定义传入类型对象为*MathError指针
// 如果接口定义时传入类型对象为MathError，那么这里的写法为err.(MathError)
if e, ok := err.(*MathError); ok {
fmt.Println(e.info)
}
} else {
fmt.Println("result =", result)
}
}

资源释放

在c++程序中，经常要注意内存指针、文件句柄、网络套接字等等资源的释放，特别需要注意其释放的时机。而golang使用defer
关键字和背后的内部机制简单地解决了资源释放的问题。

√ defer关键字能保证其后的代码能在函数退出前调用。

√ 一个函数中可以存在多个defer语句，需要注意的是defer语句的调用是遵照先进后出的原则，即最后一个defer语句将最先被执行。

√ 可以在defer后加一个匿名函数来进行复杂的清理工作。

• 简单的清理工作

▶ 语法如下

func 函数名(参数列表) (返回值列表) {
...
// 资源申请
    defer 清理函数
...
}

▶ 示例如下

package main

import (
"io"
"os"
)

func CopyFile(dst, src string) (w int64, err error) {
srcFile, err := os.Open(src)
if err != nil {
return
}
defer srcFile.Close()

dstFile, err := os.Create(dst)
if err != nil {
return
}
defer dstFile.Close()

return io.Copy(dstFile, srcFile)
}

func main() {
CopyFile("D:/2.txt", "D:/1.txt")
}

▶ 先进后出规则

package main

import (
"fmt"
)

func Test() {
defer fmt.Println(1)
defer fmt.Println(2)
defer fmt.Println(3)
}

func main() {
Test()
}

• 复杂的清理工作

▶ 语法如下

func 函数名(参数列表) (返回值列表) {
...
// 资源申请
    defer func() {
// 复杂的清理工作
} ()
...
}

▶ 示例如下

package main

import (
"fmt"
"io"
"os"
)

func CopyFile(dst, src string) (w int64, err error) {
srcFile, err := os.Open(src)
if err != nil {
return
}
defer func() {
fmt.Println("close file :", src)
srcFile.Close()
}()

dstFile, err := os.Create(dst)
if err != nil {
return
}
defer func() {
fmt.Println("close file :", dst)
dstFile.Close()
}()

return io.Copy(dstFile, srcFile)
}

func main() {
CopyFile("D:/2.txt", "D:/1.txt")
}

异常处理

一些高级语言中一般提供类似try...catch...finally...的语法，用于捕获异常。golang提供panic和recover两个关键字用于异常处理。

• panic

panic在golang中是一个内置函数，接收一个interface{}类型的值作为参数：

func panic(interface{}) {
...
}

当一个函数执行过程中调用panic函数时，函数执行流程将立即终止，但panic之前的defer关键字延迟执行的语句将正常执行，之后该函数将返回到上层调用函数，并逐层向上执行panic流程，直至函数所属的goroutine中所有正在执行函数终止。错误信息将被报告，包括在调用panic()函数时传入的参数。下面用一个示例说明：

package main

import (
"fmt"
)

func MyFunc1() {
defer fmt.Println("MyFunc1 defer 1")

panic("MyFunc1 panic test")

defer fmt.Println("MyFunc1 defer 2")
}

func MyFunc2() {
defer fmt.Println("MyFunc2 defer 1")

MyFunc1()

defer fmt.Println("MyFunc2 defer 2")
}

func main() {
MyFunc2()
}

程序输出如下：

• recover

recover在golang中是一个内置函数，返回一个interface{}类型的值作为参数：

func recover() interface{} {
...
}

panic函数触发后不会立即返回，而是先defer，再返回。如果defer的时候，有办法将panic捕获到，然后及时进行异常处理，并阻止panic传递，那处理机制就完善了。因此golang提供了recover内置函数，用于捕获panic并阻止其向上传递。需要注意的是，recover之后，逻辑并不会恢复到panic处，函数还是会在defer之后返回，但是所属goroutine将不会退出。

▶ 本层函数处理

package main

import (
"fmt"
)

func MyFunc1() {
defer func() {
fmt.Println("MyFunc1 defer 1")
if r := recover(); r != nil {
fmt.Println("Runtime error caught :", r)
}
}()

panic("MyFunc1 panic test")

fmt.Println("MyFunc1 defer 2")
}

func MyFunc2() {
defer fmt.Println("MyFunc2 defer 1")

MyFunc1()

defer fmt.Println("MyFunc2 defer 2")
}

func main() {
MyFunc2()
}

程序输出如下：

▶ 上层函数处理

package main

import (
"fmt"
)

func MyFunc1() {
defer fmt.Println("MyFunc1 defer 1")

panic("MyFunc1 panic test")

fmt.Println("MyFunc1 defer 2")
}

func MyFunc2() {
defer func() {
fmt.Println("MyFunc1 defer 2")
if r := recover(); r != nil {
fmt.Println("Runtime error caught :", r)
}
}()

MyFunc1()

defer fmt.Println("MyFunc2 defer 2")
}

func main() {
MyFunc2()
}

程序输出如下：

• 模拟try...catch...语法

▶ 语法如下

func Try(f func(), handler func(interface{})) {
defer func() {
if err := recover(); err != nil {
handler(err)
}
}()

f()
}

▶ 示例如下

package main

import (
"fmt"
)

func Try(f func(), handler func(interface{})) {
defer func() {
if err := recover(); err != nil {
handler(err)
}
}()

f()
}

func main() {
Try(func() {
panic("main panic")
}, func(e interface{}) {
fmt.Println(e)
})
}