什么是defer
defer语句用于golang程序中延迟函数的调用, 每次defer都会把一个函数压入栈中, 函数返回前再把延迟的函数取出并执行。
为了方便描述, 我们把创建defer的函数称为主函数, defer语句后面的函数称为延迟函数。延迟函数可能有输入参数, 这些参数可能来源于定义defer的函数, 延迟函数也可能引用主函数用于返回的变量, 也就是说延迟函数可能会影响主函数的一些行为。
defer的规则
规则一:延迟函数的参数在defer语句出现时就已经确定
package main
import "fmt"
func main() {
deferFuncParameter()
}
func deferFuncParameter() {
var aInt = 1
defer fmt.Println(aInt)
aInt = 2
return
}
结果:
代码说明: 函数deferFuncParameter()定义一个整型变量并初始化为1,然后使用defer语句打印出变量值, 最后修改变量值为2。
参考答案: 输出1。 延迟函数fmt.Println(aInt)的参数在defer语句出现时就已经确定了, 后面修改的aInt变量实际上是拷贝了一份。所以无论后面如何修改aInt变量都不会影响延迟函数的执行。
注意: 对于指针类型参数, 规则仍然适用, 只不过延迟函数的参数是一个地址值, 这种情况下,defer后面的语句对变量的修改可能会影响延迟函数。
package main
import "fmt"
func main() {
deferArray()
}
func printArray(array *[3]int) {
for i := range array {
fmt.Println(array[i])
}
}
func deferArray() {
var aArray = [3]int{1, 2, 3}
defer printArray(&aArray)
aArray[0] = 10
return
}
结果:
函数说明: 函数deferFuncParameter()定义一个数组, 通过defer延迟函数printArray()的调用, 最后修改数组第一个元素。 printArray()函数接受数组的指针并把数组全部打印出来。
参考答案: 输出10、 2、 3三个值。 延迟函数printArray()的参数在defer语句出现时就已经确定了, 即数组的地址, 由于延迟函数执行时机是在return语句之前, 所以对数组的最终修改值会被打印出来。
规则二:defer延迟函数执行按后进先出顺序执行, 即先出现的defer最后执行
定义defer类似于入栈操作, 执行defer类似于出栈操作。
设计defer的初衷是简化函数返回时资源清理的动作, 资源往往有依赖顺序, 比如先申请A资源, 再跟据A资源申请B资源, 跟据B资源申请C资源, 即申请顺序是:A—>B—>C, 释放时往往又要反向进行。 这就是把deffer设计成FIFO的原因。每申请到一个用完需要释放的资源时, 立即定义一个defer来释放资源是个很好的习惯。
规则三: 延迟函数可能操作主函数的具名返回值
定义defer的函数, 即主函数可能有返回值, 返回值有没有名字没有关系, defer所作用的函数, 即延迟函数可能会影响到返回值。
package main
import "fmt"
func main() {
fmt.Println(test())
}
func test() (res int) {
a := 1
defer func() {
res ++
}()
return a
}
结果:
函数说明: 函数拥有一个具名返回值result, 函数内部声明一个变量a, defer指定一个延迟函数, 最后返回变量a。延迟函数中递增res。
参考答案: 函数输出2。 函数的return语句并不是原子的, 实际执行分为设置返回值—>ret, defer语句实际执行在返回前, 即拥有defer的函数返回过程是: 设置返回值—>执行defer—>res。 所以return语句先把res设置为a的值, 即1, defer语句中又把res递增1, 所以最终返回2。
return 返回值解析:
该函数的return语句可以拆分成下面两行:
result = i
return
而延迟函数的执行正是在return之前, 即加入defer后的执行过程如下:
result = i
result++
return
一个主函数拥有一个匿名的返回值, 返回时使用字面值, 比如返回”1”、 ”2”、 ”Hello”这样的值, 这种情况下defer语句是无法操作返回值的。
另外返回值是匿名类型的值,这种情况下defer语句可以引用到返回值, 但不会改变返回值。
package main
import "fmt"
func main() {
fmt.Println(test())
}
func test() int {
a := 1
defer func() {
a ++
}()
return a
}
func printArray(array *[3]int) {
for i := range array {
fmt.Println(array[i])
}
}
结果:
上面的函数, 返回一个局部变量, 同时defer函数也会操作这个局部变量。 对于匿名返回值来说, 可以假定仍然有一个变量存储返回值, 假定返回值变量为”anony”, 上面的返回语句可以拆分成以下过程:
anony = a
a++
return
由于a是整型, 会将值拷贝给anony, 所以defer语句中修改i值, 对函数返回值不造成影响。
总结
- defer定义的延迟函数参数在defer语句出时就已经确定下来了
- defer定义顺序与实际执行顺序相反
- return不是原子操作,执行过程是: 保存返回值(若有)—>执行defer( 若有) —>执行ret跳转
- 申请资源后立即使用defer关闭资源是好习惯