1

首发地址:https://mp.weixin.qq.com/s/FRa0A51DGQ6MiKO6PUu6wQ

Go 语言中的错误处理不仅仅只有 if err != nildeferpanicrecover 这三个相对来说不不如 if err != nil 有名气的控制流语句,也与错误处理息息相关。本文就来讲解下这三者在 Go 语言中的应用。

Defer

defer 是一个 Go 中的关键字,通常用于简化执行各种清理操作的函数。defer 后跟一个函数(或方法)调用,该函数(或方法)的执行会被推迟到外层函数返回的那一刻,即函数(或方法)要么遇到了 return,要么遇到了 panic

语法

defer 功能使用语法如下:

defer Expression

其中 Expression 必须是函数或方法的调用。

defer 使用示例如下:

func f() {
    defer fmt.Println("deferred in f")
    fmt.Println("calling f")
}

func main() {
    f()
}

执行示例代码,得到输出如下:

$ go run main.go            
calling f
deferred in f

根据输出可以发现,被 defer 修饰的 fmt.Println("deferred in f") 调用并没有立即执行,而是先执行了 fmt.Println("calling f"),然后才会执行 defer 修饰的函数调用语句。

执行顺序

一个函数中可以写多个 defer 语句:

func f() {
    defer fmt.Println("deferred in f 1")
    defer fmt.Println("deferred in f 2")
    defer fmt.Println("deferred in f 3")
    fmt.Println("calling f")
}

执行示例代码,得到输出如下:

$ go run main.go
calling f
deferred in f 3
deferred in f 2
deferred in f 1

defer 修饰的函数调用,在外层函数返回后按后进先出顺序执行,即 Last In First Out(LIFO)。

不仅如此,defer 可以写在任意位置,并且还可以嵌套,即在被 defer 修饰的函数中再次使用 defer

示例如下:

func f() {
    fmt.Println("1")

    defer func() {
        fmt.Println("2")
        defer fmt.Println("3")
        fmt.Println("4")
    }()

    fmt.Println("5")

    defer fmt.Println("6")

    fmt.Println("7")
}

执行示例代码,得到输出如下:

$ go run main.go
1
5
7
6
2
4
3

这个输出结果符合你的预期吗?

先看外层函数 f 的代码逻辑,有两个 defer 语句,无论位置在哪,defer 都会使函数调用延迟执行,所以先输出了 157

然后根据 LIFO 原则,先执行第 2 个 defer 语句所修饰的函数调用,所以输出 6

接着执行第 1 个 defer 语句所修饰的函数调用,其内部同样会按顺序执行没有被 defer 语句修饰的代码,所以先输出 24,然后执行 defer 语句所修饰的函数调用,输出 3

读写函数返回值

有时候,我们可以使用 defer 语句来读取或修改函数的返回值。

有如下示例,试图在 defer 中修改函数的返回值:

func f() int {
    r := 2
    defer func() {
        fmt.Println("r:", r)
        r *= 3
    }()
    return r
}

func main() {
    fmt.Println(f())
}

执行示例代码,得到输出如下:

$ go run main.go
r: 2
2

看来没有成功。

函数使用具名返回值再来看看:

func f() (r int) {
    r = 2
    defer func() {
        fmt.Println("r:", r)
        r *= 3
    }()
    return r
}

执行示例代码,得到输出如下:

$ go run main.go
r: 2
6

这次成功了。

如果改成这样呢:

func f() (r int) {
    defer func() {
        fmt.Println("r:", r)
        r *= 3
    }()
    return 2
}

现在,返回值直接写成了 2,而非变量 r

执行示例代码,得到输出如下:

$ go run main.go
r: 2
6

这次返回值依然修改成功了。

前面几个示例,其实都算使用了闭包。因为被 defer 修饰的函数内部都引用了外部变量 r

我们再看一个不使用闭包的示例:

func f() (r int) {
    defer func(r int) {
        fmt.Println("r:", r)
        r *= 3
    }(r)
    return 2
}

执行示例代码,得到输出如下:

$ go run main.go
r: 0
2

这次返回值没有修改成功,并且被 defer 修饰的函数内部读到的 r 值为 0,并不是前面示例中的 2

也就是说,实际上虽然被 defer 修饰的函数调用会延迟执行,但是我们传递给函数的参数,会被立即求值

我们接着看下面这个示例:

func f() (r int) {
    x := 2
    defer func() {
        fmt.Println("r:", r)
        fmt.Println("x:", x)
        x *= 3
    }()
    return x
}

执行示例代码,得到输出如下:

$ go run main.go
r: 2
x: 2
2

当代码执行到 return x 时,r 值也会被赋值为 2,这没什么好解释的。

然后在 defer 所修饰的函数内部,我们只修改了 x 变量,这对返回结果 r 没有影响。

把函数返回值类型改成指针试试呢:

func f() (r *int) {
    x := 2
    defer func() {
        fmt.Println("r:", *r)
        fmt.Println("x:", x)
        x *= 3
    }()
    return &x
}

func main() {
    fmt.Println(*f())
}

执行示例代码,得到输出如下:

$ go run main.go
r: 2
x: 2
6

这次返回值又成功被修改了。

看到这里,你是不是对 defer 语句的效果有点懵,没关系,我们再来梳理下 defer 执行时机。

defer 语句的行为其实是可预测的,我们可以记住这三条规则

  1. 在计算 defer 语句时,将立即计算被 defer 修饰的函数参数。
  2. defer 修饰的函数,在外层函数返回后按后进先出的顺序(LIFO)执行。
  3. 延迟函数可以读取或赋值给外层函数的具名返回值。

现在,你再翻回去重新看看上面的几个示例程序,是不是都能理解了呢?

释放资源

defer 还常被用来释放资源,比如关闭文件对象。

这里有个示例程序,可以将一个文件内容复制到另外一个文件中:

func CopyFile(dstName, srcName string) (written int64, err error) {
    src, err := os.Open(srcName)
    if err != nil {
        return
    }

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

    written, err = io.Copy(dst, src)
    dst.Close()
    src.Close()
    return
}

不过这个程序存在 bug,如果 os.Create 执行失败,函数返回后 src 并没有被关闭。

而这种场景刚好适用 defer,示例如下:

func CopyFile(dstName, srcName string) (written int64, err error) {
    src, err := os.Open(srcName)
    if err != nil {
        return
    }
    defer src.Close()

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

    return io.Copy(dst, src)
}

此时如果 os.Create 执行失败,函数返回后 defer src.Close() 将会被执行,文件资源得以释放。

切记,不要在 if err != nil 之前调用 defer 释放资源,这很可能会触发 panic

src, err := os.Open(srcName)
defer src.Close()
if err != nil {
    return
}

因为,如果调用 os.Open 报错,src 值将为 nil,而 nil.Close() 会触发 panic,导致程序意外终止而退出。

此外,在处理释放资源的情况,你可能写出如下代码:

type fakeFile struct {
    name string
}

func (f *fakeFile) Close() error {
    fmt.Println("close:", f)
    return nil
}

// 错误写法:f 变量的值最终是 f2,所以 f2 会被关闭两次,f1 没关闭
func processFile() {
    f := fakeFile{name: "f1"}
    defer f.Close()

    f = fakeFile{name: "f2"}
    defer f.Close()

    fmt.Println("calling processFile")
    return
}

func main() {
    processFile()
}

执行示例代码,得到输出如下:

$ go run main.go
calling processFile
close: &{f2}
close: &{f2}

可以发现,在函数 processFile 中,因为 f 被重复赋值,导致 f 变量的值最终是 f2,所以 f2 会被关闭两次,f1 并没有被关闭。

还记得我们前面讲过的规则吗:在计算 defer 语句时,将立即计算被 defer 修饰的函数参数

所以,我们可以在 defer 处让变量 f 先被计算出来:

func processFile1() {
    f := fakeFile{name: "f1"}
    defer func(f fakeFile) {
        f.Close()
    }(f)

    f = fakeFile{name: "f2"}
    defer func(f fakeFile) {
        f.Close()
    }(f)

    fmt.Println("calling processFile1")
    return
}

这样就解决了问题。

当然,更简单的方式是我们压根就不要使用同一个变量来表示不同的文件对象:

func processFile2() {
    f1 := fakeFile{name: "f1"}
    defer f1.Close()

    f2 := fakeFile{name: "f2"}
    defer f2.Close()

    fmt.Println("calling processFile2")
    return
}

不过,有时候在在 for 循环中,就是会出现 f 被重复赋值的情况,在 for 循环中使用 defer 语句,我们可能还会踩到类似的坑,所以你一定要小心。

WithClose

文章读到这里,想必你也看出来了,defer 功能正是对标了 Python 中的 try...finally 或者 with 语句的效果。

Python 的 with 语法非常优雅,如何使用 defer 实现近似效果呢?

你可以在我的另一篇文章《在 Go 中如何实现类似 Python 中的 with 上下文管理器》中找到答案。

篇幅所限,我就不在这里再废话连篇的讲一遍了。

如果你想用下面这种单独的代码块作用域来实现:

func f() {
    {
        // defer 函数一定是在函数退出时才会执行,而不是代码块退出时执行
        defer fmt.Println("defer done")
        fmt.Println("code block")
    }

    fmt.Println("calling f")
}

很遗憾的告诉你,这并不能达到想要的效果,你可以思考后再点击我的另一篇文章来对比下你我二人的实现是否相同。

结构体方法是否使用指针接收者

当结构体方法使用指针作为接收者时,也要小心。

示例如下:

type User struct {
    name string
}

func (u User) Name() {
    fmt.Println("Name:", u.name)
}

func (u *User) PointName() {
    fmt.Println("PointName:", u.name)
}

func printUser() {
    u := User{name: "user1"}

    defer u.Name()
    defer u.PointName()

    u.name = "user2"
}

func main() {
    printUser()
}

执行示例代码,得到输出如下:

$ go run main.go
PointName: user2
Name: user1

User.Name 方法接收者为结构体,在 defer 中被调用,最终输出结果为初始 nameuser1

User.PointName 方法接收者为指针,在 defer 中被调用,最终输出结果为修改后的 nameuser2

可见,defer 处不仅会计算函数参数,其实它会对其后面的表达式求值,并计算出最终将要执行的函数或方法。

也就是说,代码执行到 defer u.Name() 时,变量 u 的值就已经计算出来了,相当于“复制”了一个新的变量,后面再通过 u.name = "user2" 修改其属性,二者已经不是同一个变量了。

而代码执行到 defer u.PointName() 时,其实这里的 u 是指针类型,即使“复制”了一个新的变量,其内部保存的指针依然相等,所以可以被修改。

如果将代码修改成如下这样,执行结果又会怎样呢?

func printUser() {
    u := User{name: "user1"}

    defer func() {
        u.Name()
        u.PointName()
    }()

    u.name = "user2"
}

这个就交给你自己去实验了。

当 defer 遇到 os.Exit

defer 遇到 os.Exit 时会怎样呢?

func f() {
    defer fmt.Println("deferred in f")
    fmt.Println("calling f")
    os.Exit(0)
}

func main() {
    f()
}

执行示例代码,得到输出如下:

$ go run main.go
calling f

可见,当遇到 os.Exit 时,程序直接退出,defer 并不会被执行,这一点平时开发过程中要格外注意。

一个过时的面试题

前几年,有一个考察 defer 的面试题经常在网上出现:

func f() {
    for i := 0; i < 3; i++ {
        defer func() {
            fmt.Println(i)
        }()
    }
}

问执行 f 以后,输出什么?

既然会成为面试题,执行结果就肯定有猫腻。

如果你使用 Go 1.22 以前的版本执行示例代码,将得到如下结果:

$ go run main.go
3
3
3

而如果你使用 Go 1.22 及以后的版本执行示例代码,将得到如下结果:

$ go run main.go
2
1
0

这是由于,在 Go 1.22 以前,由 for 循环声明的变量只会被创建一次,并在每次迭代时更新。在 Go 1.22 中,循环的每次迭代都会创建新的变量,以避免意外的共享错误

这在 Go 1.22 Release Notes 中有说明。

在旧版本的 Go 中要修复这个问题,只需要这样写即可:

func f() {
    for i := 0; i < 3; i++ {
        defer fmt.Println(i)
    }
}

直接把 defer 放在外面,不要构成闭包。

又或者为 defer 函数增加参数:

func f() {
    for i := 0; i < 3; i++ {
        defer func(i int) {
            fmt.Println(i)
        }(i)
    }
}

总之,解决方案就是不要出现闭包。

不要出现 defer nil 的情况

前文说过,defer 后面支持函数或方法的调用。

但是,如果计算 defer 后的表达式出现 nil 的情况,则会触发 panic

func deferNil() {
    var f func()
    defer f()
    fmt.Println("calling deferNil")
}

func main() {
    deferNil()
}

执行示例代码,得到输出如下:

calling deferNil
panic: runtime error: invalid memory address or nil pointer dereference
[signal SIGSEGV: segmentation violation code=0x2 addr=0x0 pc=0x10264f88c]

goroutine 1 [running]:
main.deferNil()
        /go/blog-go-example/error/defer-panic-recover/defer/main.go:363 +0x6c
main.main()
        /go/blog-go-example/error/defer-panic-recover/defer/main.go:384 +0x1c
exit status 2

因为 nil 不可被调用。

至于到底什么是 panic,咱们往下看。

Panic

在 Go 中,error 表示一个错误,错误通常会返给调用方,交由调用方来决定如何处理。而 panic 则表示一个无法挽回的异常,panic 会直接终止当前执行的控制流。

panic 是一个内置函数,它会停止程序的正常控制流并输出 panic 相关信息。

有两种方式可以触发 panic,一种是非法操作导致运行时错误,比如访问数组索引越界,此时会触发运行时 panic。另一种是主动调用 panic 函数。

当在函数 F 中调用了 panic 后,程序执行流程如下:

函数 F 调用 panic 时,F 的执行会被停止,接下来会执行 F 中调用 panic 之前的所有 defer 函数,然后 F 返回给调用者。

接着,对于 F 的调用方 G 的行为也类似于对 panic 的调用。

该过程继续向上返回,直到当前 goroutine 中的所有函数都返回,此时程序崩溃。

最后,你将在执行 Go 程序的控制台看到程序执行异常的堆栈信息。

使用

panic 使用示例如下:

func f() {
    defer fmt.Println("defer 1")
    fmt.Println(1)
    panic("woah")
    defer fmt.Println("defer 2")
    fmt.Println(2)
}

func main() {
    f()
}

执行示例代码,得到输出如下:

$ go run main.go
1
defer 1
panic: woah

goroutine 1 [running]:
main.f()
        /go/blog-go-example/error/defer-panic-recover/panic/main.go:10 +0xa0
main.main()
        /go/blog-go-example/error/defer-panic-recover/panic/main.go:29 +0x1c
exit status 2

可以发现,panic 会输出异常堆栈信息。

并且 1defer 1 都被输出了,而 2defer 2 没有输出,说明 panic 调用之后的代码不会执行,但它不影响 panic 之前 defer 函数的执行。

此外,如果你足够细心,还可以发现 panic 后程序的退出码为 2

子 Goroutine 中 panic

如果在子 goroutine 中发生 panic,也会导致主 goroutine 立即退出:

func g() {
    fmt.Println("calling g")
    // 子 goroutine 中发生 panic,主 goroutine 也会退出
    go f(0)
    fmt.Println("called g")
}

func f(i int) {
    fmt.Println("panicking!")
    panic(fmt.Sprintf("i=%v", i))
    fmt.Println("printing in f", i) // 不会被执行
}

func main() {
    g()
    time.Sleep(10 * time.Second)
}

执行示例代码,程序并不会等待 10s 后才退出,而是立即 panic 并退出,得到输出如下:

$ go run main.go
calling g
called g
panicking!
panic: i=0

goroutine 3 [running]:
main.f(0x0)
        /go/blog-go-example/error/defer-panic-recover/panic/main.go:25 +0xa0
created by main.g in goroutine 1
        /go/blog-go-example/error/defer-panic-recover/panic/main.go:19 +0x5c
exit status 2

panic 和 os.Exit

虽然 panicos.Exit 都能使程序终止并退出,但它们有着显著的区别,尤其在触发时的行为和对程序流程的影响上。

panic 用于在程序中出现异常情况时引发一个运行时错误,通常会导致程序崩溃(除非被 recover 恢复)。当触发 panic 时,defer 语句仍然会执行。panic 还会打印详细的堆栈信息,显示引发错误的调用链。panic 退出状态码固定为 2

os.Exit 会立即终止程序,并返回指定的状态码给操作系统。当执行 os.Exit 时,defer 语句不会执行。os.Exit 直接通知操作系统退出程序,它不会返回给调用者,也不会引发运行时堆栈追踪,所以也就不会打印堆栈信息。os.Exit 可以设置程序退出状态码。

因为 panic 比较暴力,所以一般只建议在 main 函数中使用,比如应用的数据库初始化失败后直接 panic,因为程序无法连接数据库,程序继续执行意义不大。而普通函数中推荐尽量返回 error 而不是直接 panic

不过 panic 也不是没有挽救的余地,recover 就是来恢复 panic 的。

Recover

recover 也是一个函数,用来从 panic 所导致的程序崩溃中恢复执行。

使用

recover 使用示例如下:

func f() {
    defer func() {
        recover()
    }()
    
    defer fmt.Println("defer 1")
    fmt.Println(1)
    panic("woah")
    defer fmt.Println("defer 2")
    fmt.Println(2)
}

func main() {
    f()
}

执行示例代码,得到输出如下:

$ go run main.go
1
defer 1

recover() 的调用捕获了 panic 触发的异常,并且程序正常退出。

recover 函数只在 defer 语句的上下文中才有效,直接调用的话,只会返回 nil

如下两种方式都是错误的用法:

recover()
defer recover()

可见,recover 必须与 defer 一同使用,来从 panic 中恢复程序。不过 panic 之后的代码依旧不会执行,recover() 调用后只会执行 defer 语句中的剩余代码。

下面这个例子将会捕获到 panic,并且输出 panic 信息:

func f() {
    defer func() {
        if r := recover(); r != nil {
            fmt.Println("recover:", r)
        }
    }()
    panic("woah")
}

执行示例代码,得到输出如下:

$ go run main.go
recover: woah

可以发现,recover 函数的返回值,正是 panic 函数的参数。

不要在 defer 中出现 panic

为了避免不必要的麻烦,defer 函数中最好不要有能够引起 panic 的代码。

正常来说,defer 用来释放资源,不会出现大量代码。如果 defer 函数中逻辑过多,则需要斟酌下有没有更优解。

如下示例将输出什么?

func f() {
    defer func() {
        if r := recover(); r != nil {
            fmt.Println("recover:", r)
        }
    }()

    defer func() {
        panic("woah 1")
    }()
    panic("woah 2")
}

执行示例代码,得到输出如下:

$ go run main.go
recover: woah 1

看来,defer 中的 panic("woah 1") 覆盖了程序正常控制流中的 panic("woah 2")

如果我们将代码顺序稍作修改:

func f() {
    defer func() {
        panic("woah 1")
    }()

    defer func() {
        if r := recover(); r != nil {
            fmt.Println("recover:", r)
        }
    }()

    panic("woah 2")
}

执行示例代码,得到输出如下:

$ go run main.go
recover: woah 2
panic: woah 1

goroutine 1 [running]:
main.f.func1()
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:68 +0x2c
main.f()
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:77 +0x68
main.main()
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:142 +0x1c
exit status 2

看来,调用 recoverdefer 应该放在函数的入口处,成为第一个 defer

recover 只能捕获当前 Goroutine 中的 panic

需要额外注意的一点是,recover 只会捕获当前 goroutine 所触发的 panic

示例如下:

func f() {
    defer func() {
        if r := recover(); r != nil {
            fmt.Println("recover:", r)
        }
    }()

    go func() {
        panic("woah")
    }()
    time.Sleep(1 * time.Second)
}

执行示例代码,得到输出如下:

$ go run main.go
panic: woah

goroutine 18 [running]:
main.f.func2()
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:91 +0x2c
created by main.f in goroutine 1
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:90 +0x40
exit status 2

goroutine 中触发的 panic 并没有被 recover 捕获。

所以,如果你认为代码中需要捕获 panic 时,就需要在每个 goroutine 中都执行 recover

将 panic 转换成 error 返回

有时候,我们可能需要将 panic 转换成 error 并返回,防止当前函数调用他人提供的不可控代码时出现意外的 panic

func g(i int) (number int, err error) {
    defer func() {
        if r := recover(); r != nil {
            var ok bool
            err, ok = r.(error)
            if !ok {
                err = fmt.Errorf("f returns err: %v", r)
            }
        }
    }()

    number, err = f(i)
    return number, err
}

func f(i int) (int, error) {
    if i == 0 {
        panic("i=0")
    }
    return i * i, nil
}

func main() {
    fmt.Println(g(1))
    fmt.Println(g(0))
}

执行示例代码,得到输出如下:

$ go run main.go
1 <nil>
0 f returns err: i=0

net/http 使用 recover 优雅处理 panic

我们在开发 HTTP Server 程序时,即使某个请求遇到了 panic 也不应该使整个程序退出。所以,就需要使用 recover 来处理 panic

来看一个使用 net/http 创建的 HTTP Server 程序示例:

package main

import (
    "fmt"
    "log"
    "net/http"
    "os"
)

func handler(w http.ResponseWriter, r *http.Request) {
    if r.URL.Path == "/panic" {
        panic("url is error")
    }
    // 打印请求的路径
    fmt.Fprintf(w, "Hello, you've requested: %s\n", r.URL.Path)
}

func main() {
    // 创建一个日志实例,写到标准输出
    logger := log.New(os.Stdout, "http: ", log.LstdFlags)

    // 自定义 HTTP Server
    server := &http.Server{
        Addr:     ":8080",
        ErrorLog: logger, // 设置日志记录器
    }

    // 注册处理函数
    http.HandleFunc("/", handler)

    // 启动服务器
    fmt.Println("Starting server on :8080")
    if err := server.ListenAndServe(); err != nil {
        logger.Println("Server failed to start:", err)
    }
}

启动示例,程序会阻塞在这里等待请求进来:

$ go run main.go
Starting server on :8080

使用 curl 命令分别对 HTTP Server 发送三次请求:

$ curl localhost:8080
Hello, you've requested: /
$ curl localhost:8080/panic
curl: (52) Empty reply from server
$ curl localhost:8080/hello
Hello, you've requested: /hello

可以发现,在请求 /panic 路由时,HTTP Server 触发了 panic 并返回了空内容,然后第三个请求依然能够得到正确的响应。

可见 HTTP Server 并没有退出。

现在回去看一下执行 HTTP Server 的控制台日志:

Starting server on :8080
http: 2024/10/13 23:08:28 http: panic serving [::1]:50547: url is error
goroutine 34 [running]:
net/http.(*conn).serve.func1()
        /go/pkg/mod/golang.org/toolchain@v0.0.1-go1.23.1.darwin-arm64/src/net/http/server.go:1947 +0xb0
panic({0x10114c000?, 0x1011a4ba8?})
        /go/pkg/mod/golang.org/toolchain@v0.0.1-go1.23.1.darwin-arm64/src/runtime/panic.go:785 +0x124
main.handler({0x1011a8178?, 0x140001440e0?}, 0x1400010bb28?)
        /workspace/projects/go/blog-go-example/error/defer-panic-recover/recover/http/main.go:12 +0x130
net/http.HandlerFunc.ServeHTTP(0x101348320?, {0x1011a8178?, 0x140001440e0?}, 0x1010999e4?)
        /go/pkg/mod/golang.org/toolchain@v0.0.1-go1.23.1.darwin-arm64/src/net/http/server.go:2220 +0x38
net/http.(*ServeMux).ServeHTTP(0x0?, {0x1011a8178, 0x140001440e0}, 0x14000154140)
        /go/pkg/mod/golang.org/toolchain@v0.0.1-go1.23.1.darwin-arm64/src/net/http/server.go:2747 +0x1b4
net/http.serverHandler.ServeHTTP({0x1400011ade0?}, {0x1011a8178?, 0x140001440e0?}, 0x6?)
        /go/pkg/mod/golang.org/toolchain@v0.0.1-go1.23.1.darwin-arm64/src/net/http/server.go:3210 +0xbc
net/http.(*conn).serve(0x140000a4120, {0x1011a8678, 0x1400011acf0})
        /go/pkg/mod/golang.org/toolchain@v0.0.1-go1.23.1.darwin-arm64/src/net/http/server.go:2092 +0x4fc
created by net/http.(*Server).Serve in goroutine 1
        /go/pkg/mod/golang.org/toolchain@v0.0.1-go1.23.1.darwin-arm64/src/net/http/server.go:3360 +0x3dc

panic 信息 url is error 被输出了,并且打印了堆栈信息。

不过这 HTTP Server 依然在运行,并能提供服务。

这其实就是在 net/http 中使用了 recover 来处理 panic

我们可以看下 http.Server.Serve 的源码:

func (srv *Server) Serve(l net.Listener) error {
    ...

    ctx := context.WithValue(baseCtx, ServerContextKey, srv)
    for {
        rw, err := l.Accept()
        if err != nil {
            ...
            return err
        }
        connCtx := ctx
        if cc := srv.ConnContext; cc != nil {
            connCtx = cc(connCtx, rw)
            if connCtx == nil {
                panic("ConnContext returned nil")
            }
        }
        tempDelay = 0
        c := srv.newConn(rw)
        c.setState(c.rwc, StateNew, runHooks) // before Serve can return
        go c.serve(connCtx)
    }
}

可以发现,在 for 循环中,每接收到一个请求都会交给 go c.serve(connCtx) 开启一个新的 goroutine 来处理。

那么在 serve 方法中就一定会有 recover 语句:

// Serve a new connection.
func (c *conn) serve(ctx context.Context) {
    if ra := c.rwc.RemoteAddr(); ra != nil {
        c.remoteAddr = ra.String()
    }
    ctx = context.WithValue(ctx, LocalAddrContextKey, c.rwc.LocalAddr())
    var inFlightResponse *response
    defer func() {
        if err := recover(); err != nil && err != ErrAbortHandler {
            const size = 64 << 10
            buf := make([]byte, size)
            buf = buf[:runtime.Stack(buf, false)]
            c.server.logf("http: panic serving %v: %v\n%s", c.remoteAddr, err, buf)
        }
        if inFlightResponse != nil {
            inFlightResponse.cancelCtx()
            inFlightResponse.disableWriteContinue()
        }
        if !c.hijacked() {
            if inFlightResponse != nil {
                inFlightResponse.conn.r.abortPendingRead()
                inFlightResponse.reqBody.Close()
            }
            c.close()
            c.setState(c.rwc, StateClosed, runHooks)
        }
    }()

    ...
}

果然,在 serve 方法源码中发现了 defer + recover 的组合。

并且这行代码:

c.server.logf("http: panic serving %v: %v\n%s", c.remoteAddr, err, buf)

可以在执行 HTTP Server 的控制台日志中得到印证:

http: 2024/10/13 23:08:28 http: panic serving [::1]:50547: url is error

panic(nil)

panic 函数签名如下:

func panic(v any)

既然 panic 参数是 any 类型,那么 nil 当然也可以作为参数。

可以写出 panic(nil) 程序示例代码如下:

func f() {
    defer func() {
        if err := recover(); err != nil {
            fmt.Println(err)
        }
    }()
    panic(nil)
}

执行示例代码,得到输出如下:

$ go run main.go
panic called with nil argument

这没什么问题。

但是在 Go 1.21 版本以前,执行上述代码,将得到如下结果:

$ go run main.go

你没看错,我也没写错误,这里什么都没输出。

在旧版本的 Go 中,panic(nil) 并不能被 recover 捕获,recover() 调用结果将返回 nil

你可以在 issues/25448 中找到关于此问题的讨论。

幸运的是,在 Go 1.21 发布时,这个问题得以解决。

不过,这就破坏了 Go 官方承诺的 Go1 兼容性保障。因此,Go 团队又提供了 GODEBUG=panicnil=1 标识来恢复旧版本中的 panic 行为。

使用方式如下:

$ GODEBUG=panicnil=1 go run main.go

其实,根据 panic 声明中的注释我们也能够观察到 Go 1.21 后 panic(nil) 行为有所改变:

// Starting in Go 1.21, calling panic with a nil interface value or an
// untyped nil causes a run-time error (a different panic).
// The GODEBUG setting panicnil=1 disables the run-time error.
func panic(v any)

panic 相关源码实现如下:

// The implementation of the predeclared function panic.
func gopanic(e any) {
    if e == nil {
        if debug.panicnil.Load() != 1 {
            e = new(PanicNilError)
        } else {
            panicnil.IncNonDefault()
        }
    }
...
}

在没有指定 GODEBUG=panicnil=1 情况下,panic(nil) 调用等价于 panic(new(runtime.PanicNilError))

数据库事务

使用 defer + recover 来处理数据库事务,也是比较常用的做法。

这里有一个来自 GORM 官方文档中的 示例程序

type Animal struct {
    Name string
}

func CreateAnimals(db *gorm.DB) error {
    tx := db.Begin()
    defer func() {
        if r := recover(); r != nil {
            tx.Rollback()
        }
    }()

    if err := tx.Error; err != nil {
        return err
    }

    if err := tx.Create(&Animal{Name: "Giraffe"}).Error; err != nil {
        tx.Rollback()
        return err
    }

    if err := tx.Create(&Animal{Name: "Lion"}).Error; err != nil {
        tx.Rollback()
        return err
    }

    return tx.Commit().Error
}

在函数最开始开启了一个事务,接着使用 defer + recover 来确保程序执行中间过程遇到 panic 时能够回滚事务。

程序执行过程中使用 tx.Create 创建了两条 Animal 数据,并且如果输出,都将回滚事务。

如果没有错误,最终调用 tx.Commit() 提交事务,并将其错误结果返回。

这个函数实现逻辑非常严谨,没什么问题。

但是这个示例代码写的过于啰嗦,还有优化的空间,可以写成这样:

func CreateAnimals(db *gorm.DB) error {
    tx := db.Begin()
    defer tx.Rollback()

    if err := tx.Error; err != nil {
        return err
    }

    if err := tx.Create(&Animal{Name: "Giraffe"}).Error; err != nil {
        return err
    }

    if err := tx.Create(&Animal{Name: "Lion"}).Error; err != nil {
        return err
    }

    return tx.Commit().Error
}

这里在 defer 中直接去掉了 recover 的判断,所以无论如何程序最终都会执行 tx.Rollback()

之所以可以这样写,是因为调用 tx.Commit() 时事务已经被提交成功,之后执行 tx.Rollback() 并不会影响已经提交事务。

这段代码看上去要简洁不少,不必在每次出现 error 时都想着调用 tx.Rollback() 回滚事务。

你可能认为这样写有损代码性能,但其实绝大多数场景下我们不需要担心。我更愿意用一点点可以忽略不计的性能损失,换来一段清晰的代码,毕竟可读性很重要。

panic 并不是都可以被 recover 捕获

最后,咱们再来看一个并发写 map 的场景,如果触发 panic 结果将会怎样?

示例如下:

func f() {
    m := map[int]struct{}{}

    go func() {
        defer func() {
            if err := recover(); err != nil {
                fmt.Println("goroutine 1", err)
            }
        }()
        for {
            m[1] = struct{}{}
        }
    }()

    go func() {
        defer func() {
            if err := recover(); err != nil {
                fmt.Println("goroutine 2", err)
            }
        }()
        for {
            m[1] = struct{}{}
        }
    }()

    select {}
}

这里启动两个 goroutine 来并发的对 map 进行写操作,并且每个 goroutine 中都使用 defer + recover 来保证能够正常处理 panic 发生。

最后使用 select {} 阻塞主 goroutine 防止程序退出。

执行示例代码,得到输出如下:

$ go run main.go
fatal error: concurrent map writes

goroutine 3 [running]:
main.f.func1()
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:156 +0x4c
created by main.f in goroutine 1
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:149 +0x50

goroutine 1 [select (no cases)]:
main.f()
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:171 +0x84
main.main()
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:204 +0x1c

goroutine 4 [runnable]:
main.f.func2()
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:167 +0x4c
created by main.f in goroutine 1
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:160 +0x80
exit status 2

然而程序还是输出 panic 信息 fatal error: concurrent map writes 并退出了。

但是根据输出信息,我们无法知道具体原因。

Go 1.19 Release Notes 中有提到,从 Go 1.19 版本开始程序遇到不可恢复的致命错误(例如并发写入 map,或解锁未锁定的互斥锁)只会打印一个简化的堆栈信息,不包含运行时元数据。不过这可以通过将环境变量 GOTRACEBACK 被设置为 systemcrash 来解决。

所以我们可以使用如下两种方式来输出更详细的堆栈信息:

$ GOTRACEBACK=system go run main.go
$ GOTRACEBACK=crash go run main.go

再次执行示例代码,得到输出如下:

$  GOTRACEBACK=system go run main.go
fatal error: concurrent map writes

goroutine 4 gp=0x14000003180 m=3 mp=0x14000057008 [running]:
runtime.fatal({0x104904795?, 0x0?})
        /go/pkg/mod/golang.org/toolchain@v0.0.1-go1.23.1.darwin-arm64/src/runtime/panic.go:1088 +0x38 fp=0x14000051750 sp=0x14000051720 pc=0x104898a28
runtime.mapassign_fast64(0x104938ee0, 0x1400007a0c0, 0x1)
        /go/pkg/mod/golang.org/toolchain@v0.0.1-go1.23.1.darwin-arm64/src/runtime/map_fast64.go:122 +0x40 fp=0x14000051790 sp=0x14000051750 pc=0x1048cb5d0
main.f.func1()
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:156 +0x4c fp=0x140000517d0 sp=0x14000051790 pc=0x1049017bc
runtime.goexit({})
        /go/pkg/mod/golang.org/toolchain@v0.0.1-go1.23.1.darwin-arm64/src/runtime/asm_arm64.s:1223 +0x4 fp=0x140000517d0 sp=0x140000517d0 pc=0x1048d4694
created by main.f in goroutine 1
        /go/blog-go-example/error/defer-panic-recover/recover/main.go:149 +0x50
...
exit status 2

这里省略了大部分堆栈输出,只保留了重要部分。根据堆栈信息可以发现在 runtime/map_fast64.go:122 处发生了 panic

相关源码内容如下:

func mapassign_fast64(t *maptype, h *hmap, key uint64) unsafe.Pointer {
    if h == nil {
        panic(plainError("assignment to entry in nil map"))
    }
    if raceenabled {
        callerpc := getcallerpc()
        racewritepc(unsafe.Pointer(h), callerpc, abi.FuncPCABIInternal(mapassign_fast64))
    }
    if h.flags&hashWriting != 0 {
        fatal("concurrent map writes") // 第 122 行
    }
    ...
    return elem
}

显然是第 122 行代码 fatal("concurrent map writes") 触发了 panic,并且其参数内容 concurrent map writes 也正是输出结果。

fatal 函数源码如下:

// fatal triggers a fatal error that dumps a stack trace and exits.
//
// fatal is equivalent to throw, but is used when user code is expected to be
// at fault for the failure, such as racing map writes.
//
// fatal does not include runtime frames, system goroutines, or frame metadata
// (fp, sp, pc) in the stack trace unless GOTRACEBACK=system or higher.
//
//go:nosplit
func fatal(s string) {
    // Everything fatal does should be recursively nosplit so it
    // can be called even when it's unsafe to grow the stack.
    systemstack(func() {
        print("fatal error: ")
        printindented(s) // logically printpanicval(s), but avoids convTstring write barrier
        print("\n")
    })

    fatalthrow(throwTypeUser)
}

fatal 内部调用了 fatalthrow 来触发 panic。看来由 fatalthrow 所触发的 panic 无法被 recover 捕获。

我们开发时要切记:并发读写 map 触发 panic,无法被 recover 捕获。

并发操作 map 一定要小心,这是一个比较危险的行为,在 Web 开发中,如果在某个接口 handler 方法中触发了 panic,整个 http Server 会直接挂掉。

涉及并发操作 map,我们应该使用 sync.Map 来代替:

func f() {
    m := sync.Map{}

    go func() {
        defer func() {
            if err := recover(); err != nil {
                fmt.Println("goroutine 1", err)
            }
        }()
        for {
            m.Store(1, struct{}{})
        }
    }()

    go func() {
        defer func() {
            if err := recover(); err != nil {
                fmt.Println("goroutine 2", err)
            }
        }()
        for {
            m.Store(1, struct{}{})
        }
    }()

    select {}
}

这个示例就不会 panic 了。

总结

本文对错误处理三剑客 deferpanicrecover 进行了讲解梳理,虽然这三者并不是 error,但它们与错误处理息息相关。

defer 可以推迟一个函数或方法的调用,通常用于简化执行各种清理操作的函数。

panic 是一个内置函数,它会停止程序的正常控制流并输出 panic 相关信息。相比于 errorpanic 更加暴力,谨慎使用。

recover 用来从 panic 所导致的程序崩溃中恢复执行,并且要与 defer 一起使用。

本文示例源码我都放在了 GitHub 中,欢迎点击查看。

希望此文能对你有所启发。

联系我


江湖十年
67 声望15 粉丝