Golang研学：如何掌握并用好defer（延迟执行）

defer：在函数A内用defer关键字调用的函数B会在在函数A return后执行。

先看一个基础的例子，了解一下defer的效果

func main() {
    fmt.Println("in main func:", foo())
}

func foo() int {
    i := 0
    defer fmt.Println("in defer :", i)
    i = 1000
    fmt.Println("in foo:", i)
    return i+24
}

这段代码运行后会打印出

in foo: 1000
in defer : 0
in main func: 1024

变量i初始化为0，defer指定fmt.Println函数延迟到return后执行，最后main函数调用foo打印返回值。

有什么用途?

函数中会申明使用不少变量资源，函数结束时，咱们一般会对它们作一些处理：销毁、释放（例如数据库连接、文件句柄、流）。

通常状况下，咱们会在return语句以前处理这些事情。

可是，若是函数中包含多个return，这些处理咱们须要在每一个return以前都操做一次，实际工做中常常出现遗漏，代码维护时也很麻烦。

例如，在不用defer的时候，代码可能会这样写：

func foo(i int) int {
    if i > 100 {
        fmt.Println("不是期待的数字")
        i = 0
        return i
    }

    if i < 50 {
        fmt.Println("不是期待的数字")
        i = 0
        return i
    }

    return i
}

使用defer后，代码能够这样写

func foo(i int) int {
    defer func() {
        if i == 0 {
            fmt.Println("不是期待的数字")
        }
    }()

    if i > 100 {
        i = 0
        return i
    }

    if i < 50 {
        i = 0
        return i
    }

    return i
}

一个函数中多个defer的执行顺序是什么?

defer在同一个函数中可使用屡次。

多个defer指定的函数执行顺序是"先进后出"。

为何呢 ？

能够这样理解：defer关键字会使其如下的代码先执行后再执行它指定的函数，包括其下的defer语句也会比其先执行，依此类推。

这个顺序很是必要，由于在函数中，后面定义的对象可能依赖前面的对象，不然若是先出现的defer执行了，极可能形成后面的defer执行的时候出现异常。

因此，Go语言设计defer的时候是按先进后出的顺序执行的。

例子：

func foo() {
    i := 0
    defer func() {
        i--
        fmt.Println("第一个defer", i)
    }()

    i++
    fmt.Println("+1后的i：", i)

    defer func() {
        i--
        fmt.Println("第二个defer", i)
    }()

    i++
    fmt.Println("再+1后的i：", i)

    defer func() {
        i--
        fmt.Println("第三个defer", i)
    }()

    i++
    fmt.Println("再+1后的i：", i)
}

运行后能够看到

+1后的i： 1
再+1后的i： 2
再+1后的i： 3
第三个defer 2
第二个defer 1
第一个defer 0

这个过程能够看出函数执行后，先进后出执行defer并逐步处理变量的过程。

当传递参数给defer指定的函数时，函数延迟执行，那么参数值会是多少？

网上有一些总结是说：defer指定的函数的参数在 defer 时肯定，但，这只是一个总结，真正的缘由是， Go语言除了map、slice、chan都是值传递。

改造一下上面这个例子

func foo() {
    i := 0
    defer func(k int) {
        fmt.Println("第一个defer", k)
    }(i)

    i++
    fmt.Println("+1后的i：", i)

    defer func(k int) {
        fmt.Println("第二个defer", k)
    }(i)

    i++
    fmt.Println("再+1后的i：", i)

    defer func(k int) {
        fmt.Println("第三个defer", k)
    }(i)

    i++
    fmt.Println("再+1后的i：", i)
}

获得的结果

+1后的i： 1
再+1后的i： 2
再+1后的i： 3
第三个defer 2
第二个defer 1
第一个defer 0

可能会有人以为有一点出乎预料，i在return时不是已经被计算到3了吗？，为何延迟执行的defer指定的函数里的i不是3呢？

defer关键字指定的函数是在return后执行的，这很容易让人想象在return后调用函数。

可是，defer指定的函数是在当前行就调用了的，只是延迟到return后执行，而不等同于“移动”到return后执行，所以调用时传递的是当前的参数的值。

传递指针参数会是什么状况？

那么若是但愿defer指定的的函数参数的值是通过后面的代码处理过的，能够传递指针参数给defer指定的函数。

改造一下代码：

func foo() {
    i := 0
    defer func(k *int) {
        fmt.Println("第一个defer", *k)
    }(&i)

    i++
    fmt.Println("+1后的i：", i)

    defer func(k *int) {
        fmt.Println("第二个defer", *k)
    }(&i)

    i++
    fmt.Println("再+1后的i：", i)

    defer func(k *int) {
        fmt.Println("第三个defer", *k)
    }(&i)

    i++
    fmt.Println("再+1后的i：", i)
}

运行后获得

+1后的i： 1
再+1后的i： 2
再+1后的i： 3
第三个defer 3
第二个defer 3
第一个defer 3

defer会影响返回值吗？

在开头的第一个例子中能够看到，defer是在foo执行完，main里打印返回值以前执行的，可是没有影响到main里的打印结果。

这仍是由于相同的原则 Go语言除了map、slice、chan都是值传递

比较一下foo1和foo2两个函数的结果：

func main() {

    fmt.Println("foo1 return :", foo1())
    fmt.Println("foot return :", foo2())

}

func foo1() int {

    i := 0

    defer func() {
        i = 1
    }()

    return i
}

func foo2() map[string]string {

    m := map[string]string{}

    defer func() {
        m["a"] = "b"
    }()

    return m
}

运行后，打印出

foo1 return : 0
foot return : map[a:b]

两个函数不一样之处在于的返回值的类型，foo1中，int类型return后，defer不会影响返回结果，可是在foo2中map类型是引用传递，因此defer会改变返回结果。

这说明，在return时，除了map、slice、chan，其余类型return时是将值拷贝到一个临时变量空间，所以，defer指定的函数内对函数内的变量的操做不会影响返回结果的。

还有一种状况，给函数返回值申明变量名，，这时，变量空间是在函数执行前申明出来，return时只是返回这个变量空间的内容，所以defer可以改变返回值。

例如，改造一下foo1函数，给它的返回值申明一个变量名i：

func foo1() (i int) {

    i = 0

    defer func() {
        i = 1
    }()

    return i
}

再运行，能够看到 :

foo1 return : 1

返回值被defer指定的函数修改了。

defer在panic和recover处理上的使用

在Go语言里，defer有一个经典的使用场景就是recover.

在函数执行过程当中，有可能在不少地方都会出现panic，panic后若是不调用recover，程序会退出，为了避免让程序退出，咱们须要在panic后调用recover，但，panic后的代码不会执行，recover是不可能在panic后调用，然而panic所在的函数内defer指定的函数能够执行，因此recover只能在defer指定的函数中被调用，而且只须要在1个defer指定的函数中处理。

例如:

func panicfunc() {
    defer func() {
        fmt.Println("before recover")
        recover()
        fmt.Println("after recover")
    }()

    fmt.Println("before panic")
    panic(0)
    fmt.Println("after panic")
}

运行后，打印出：

before panic
before recover
after recover

总结如下

1. defer语句很是重要，很是经常使用，必须掌握
2. 在统一处理多个return和panic/recover场景下使用defer
3. 谨记“Go语言的函数参数传递的都是值（除了map、slice、chan）”这一重要原则，正确的评估defer指定函数的参数值
4. defer不影响返回值，除非是map、slice和chan，或者返回值定义了变量名
5. 执行顺序：先进后出