golang中的defer

一、defer的做用

defer 语句会将函数推迟到外层函数返回以后执行。

即defer后面的函数在defer语句所在的函数执行结束的时候会被调用

 

二、defer的语法

defer后面必须是函数调用语句，不能是其余语句，不然编译器会出错

2.一、简单的使用

package main

import "fmt"

func main() {

	defer fmt.Println("world")
	fmt.Println("hello")
}

　　

输出结果

hello
world

　　

3. defer规则

Golang官方博客里总结了defer的行为规则，只有三条，咱们围绕这三条进行说明。

3.1 规则一：延迟函数的参数在defer语句出现时就已经肯定下来了

官方给出一个例子，以下所示：

func a() {    i := 0
    defer fmt.Println(i)
    i++
    return
}

　　

defer语句中的fmt.Println()参数i值在defer出现时就已经肯定下来，其实是拷贝了一份。后面对变量i的修改不会影响fmt.Println()函数的执行，仍然打印"0"。

注意：对于指针类型参数，规则仍然适用，只不过延迟函数的参数是一个地址值，这种状况下，defer后面的语句对变量的修改可能会影响延迟函数。

3.2 规则二：延迟函数执行按后进先出顺序执行，即先出现的defer最后执行

这个规则很好理解，定义defer相似于入栈操做，执行defer相似于出栈操做。

设计defer的初衷是简化函数返回时资源清理的动做，资源每每有依赖顺序，好比先申请A资源，再跟据A资源申请B资源，跟据B资源申请C资源，即申请顺序是:A-->B-->C，释放时每每又要反向进行。这就是把deffer设计成FIFO的缘由。

每申请到一个用完须要释放的资源时，当即定义一个defer来释放资源是个很好的习惯。

3.3 规则三：延迟函数可能操做主函数的具名返回值

定义defer的函数，即主函数可能有返回值，返回值有没有名字没有关系，defer所做用的函数，即延迟函数可能会影响到返回值。

若要理解延迟函数是如何影响主函数返回值的，只要明白函数是如何返回的就足够了。

3.3.1 函数返回过程

有一个事实必需要了解，关键字return不是一个原子操做，实际上return只代理汇编指令ret，即将跳转程序执行。好比语句return i，实际上分两步进行，即将i值存入栈中做为返回值，而后执行跳转，而defer的执行时机正是跳转前，因此说defer执行时仍是有机会操做返回值的。

举个实际的例子进行说明这个过程：

func deferFuncReturn() (result int) {    i := 1

    defer func() {
       result++
    }()    return i}

　　

该函数的return语句能够拆分红下面两行：

result = i
return

　　

而延迟函数的执行正是在return以前，即加入defer后的执行过程以下：

result = i
result++
return

　　

因此上面函数实际返回i++值。

关于主函数有不一样的返回方式，但返回机制就如上机介绍所说，只要把return语句拆开均可以很好的理解，下面分别举例说明

3.3.1 主函数拥有匿名返回值，返回字面值

一个主函数拥有一个匿名的返回值，返回时使用字面值，好比返回"1"、"2"、"Hello"这样的值，这种状况下defer语句是没法操做返回值的。

一个返回字面值的函数，以下所示：

func foo() int {    var i int

    defer func() {
        i++
    }()    return 1}

　　

上面的return语句，直接把1写入栈中做为返回值，延迟函数没法操做该返回值，因此就没法影响返回值。

3.3.2 主函数拥有匿名返回值，返回变量

一个主函数拥有一个匿名的返回值，返回使用本地或全局变量，这种状况下defer语句能够引用到返回值，但不会改变返回值。

一个返回本地变量的函数，以下所示：

func foo() int {    var i int

    defer func() {
        i++
    }()    return i
}

　　

上面的函数，返回一个局部变量，同时defer函数也会操做这个局部变量。对于匿名返回值来讲，能够假定仍然有一个变量存储返回值，假定返回值变量为"anony"，上面的返回语句能够拆分红如下过程：

anony = i
i++
return

　　

因为i是整型，会将值拷贝给anony，因此defer语句中修改i值，对函数返回值不形成影响。

3.3.3 主函数拥有具名返回值

主函声明语句中带名字的返回值，会被初始化成一个局部变量，函数内部能够像使用局部变量同样使用该返回值。若是defer语句操做该返回值，可能会改变返回结果。

一个影响函返回值的例子：

func foo() (ret int) {    defer func() {
        ret++
    }()    return 0}

　　

上面的函数拆解出来，以下所示：

ret = 0
ret++
return

　　

函数真正返回前，在defer中对返回值作了+1操做，因此函数最终返回1。

4. defer实现原理

本节咱们尝试了解一些defer的实现机制。

4.1 defer数据结构

源码包src/src/runtime/runtime2.go:_defer定义了defer的数据结构：

type _defer struct {
    sp      uintptr   //函数栈指针
    pc      uintptr   //程序计数器
    fn      *funcval  //函数地址
    link    *_defer   //指向自身结构的指针，用于连接多个defer}

　　

咱们知道defer后面必定要接一个函数的，因此defer的数据结构跟通常函数相似，也有栈地址、程序计数器、函数地址等等。

与函数不一样的一点是它含有一个指针，可用于指向另外一个defer，每一个goroutine数据结构中实际上也有一个defer指针，该指针指向一个defer的单链表，每次声明一个defer时就将defer插入到单链表表头，每次执行defer时就从单链表表头取出一个defer执行。

下图展现一个goroutine定义多个defer时的场景： 

从上图能够看到，新声明的defer老是添加到链表头部。

函数返回前执行defer则是从链表首部依次取出执行，再也不赘述。

一个goroutine可能连续调用多个函数，defer添加过程跟上述流程一致，进入函数时添加defer，离开函数时取出defer，因此即使调用多个函数，也老是能保证defer是按FIFO方式执行的。

4.2 defer的建立和执行

源码包src/runtime/panic.go定义了两个方法分别用于建立defer和执行defer。

• deferproc()： 在声明defer处调用，其将defer函数存入goroutine的链表中；

• deferreturn()：在return指令，准确的讲是在ret指令前调用，其将defer从goroutine链表中取出并执行。

能够简单这么理解，在编译在阶段，声明defer处插入了函数deferproc()，在函数return前插入了函数deferreturn()。

 

5. 总结

• defer定义的延迟函数参数在defer语句出时就已经肯定下来了

• defer定义顺序与实际执行顺序相反

• return不是原子操做，执行过程是: 保存返回值(如有)-->执行defer（如有）-->执行ret跳转

• 申请资源后当即使用defer关闭资源是好习惯