Golang研学:如何掌握并用好defer(延迟执行)

栏目: Go · 发布时间: 5年前

内容简介:先看一个基础的例子,了解一下defer的效果

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

Golang研学:如何掌握并用好defer(延迟执行)

先看一个基础的例子,了解一下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 初始化为 0defer 指定 fmt.Println 函数延迟到 return 后执行,最后 main 函数调用 foo 打印返回值。

有什么用途?

函数中会申明使用很多变量资源,函数结束时,我们通常会对它们做一些处理:销毁、释放(例如数据库链接、文件句柄、流)。

一般情况下,我们会在 return 语句之前处理这些事情。

但是,如果函数中包含多个 return ,这些处理我们需要在每个 return 之前都操作一次,实际工作中经常出现遗漏,代码维护时也很麻烦。

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

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

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

    return i
}

使用defer后,代码可以这样写

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

    if i > 100 {
        return 0
    }

    if i < 50 {
        return 0
    }

    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都是值传递

比较一下 foo1foo2 两个函数的结果:

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 .

在函数执行过程中,有可能在很多地方都会出现 panicpanic 后如果不调用 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. 在统一处理多个 returnpanic/recover 场景下使用defer
  3. 谨记“Go语言的函数参数传递的都是值(除了map、slice、chan)”这一重要原则,正确的评估defer指定函数的参数值
  4. defer不影响返回值,除非是map、slice和chan,或者返回值定义了变量名
  5. 执行顺序:先进后出

以上所述就是小编给大家介绍的《Golang研学:如何掌握并用好defer(延迟执行)》,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对 码农网 的支持!

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

算法设计与分析导论

算法设计与分析导论

R.C.T.Lee (李家同)、S.S.Tseng、R.C.Chang、Y.T.Tsai / 王卫东 / 机械工业 / 2008-1 / 49.00元

本书在介绍算法时,重点介绍用干设计算法的策略.非常与众不同。书中介绍了剪枝搜索、分摊分析、随机算法、在线算法以及多项式近似方案等相对较新的思想和众多基于分摊分析新开发的算法,每个算法都与实例一起加以介绍,而且每个例子都利用图进行详细解释。此外,本书还提供了超过400幅图来帮助初学者理解。本书适合作为高等院校算法设计与分析课程的高年级本科生和低年级研究生的教材,也可供相美科技人员和专业人七参考使用。一起来看看 《算法设计与分析导论》 这本书的介绍吧!

正则表达式在线测试
正则表达式在线测试

正则表达式在线测试

RGB CMYK 转换工具
RGB CMYK 转换工具

RGB CMYK 互转工具