首页 > 代码库 > Golang中多用途的defer
Golang中多用途的defer
defer顾名思义就是延迟执行,那么defer在Golang中该如何使用以及何时使用呢?
A "defer" statement invokes a function whose executionis deferred to the moment the surrounding function returns,
Golang的官方时这么定义的。
1.那么在什么情况下会调用defer延迟过的函数呢?
从文档中可以知道主要有两种情况:
当函数执行了return 语句后
当函数处于panicing状态,也就是程序中panic回溯上来到当前函数范围内时
2. 如何使用defer呢?
defer可以当作一个修饰符,其不是用来修饰函数定义的,而是用来修饰函数调用的。当你调用一个函数时,在正常的书写方式的前面加上 defer
之后,该函数调用就被延迟了,具体我们来看个例子:
func TestPanic(){ defer func (){fmt.Println("defer function be invoked")}() fmt.Println("The first one to be invoke") return }
运行后会发现结果为:
The first one to be invoke defer function be invoked
这里我们用到了一个匿名函数,但是其原型就是 fun() 形式的调用。这里符合上面的在return 语句之后执行了我们defer过的函数。
3.defer主要用于什么场景
defer最主要的使用场景有两个,一个时资源的释放,如关闭打开的文件,另一个是和panic
以及 recover
组合起来模拟try...catch 功能;。除此之外defer还可以用来对return命名返回值作修改。
资源释放当我们打开文件作操作后,很容易忘记释放文件资源。假设现在B资源要依赖A资源。如下:
resA, err:= getA() if err!=nil { return } resB,err := getB(resA) if err != nile { return }
这里我们就容易产生bug,在获取B资源出错时,没有释放获取成功的A资源。这里我们添加:
resA, err:= getA() defer ReleaseA() if err!=nil { return } resB,err := getB(resA) defer ReleaseB() if err != nile { return }
就可以保证,无论时在哪个位置返回,其上的资源都会得到释放。
异常处理defer还有最大的用处就是和panic以及recover组合成try...catch结构。
看个示例,来自官网的:
func protect(g func()) { defer func() { log.Println("done") // Println executes normally even if there is a panic if x := recover(); x != nil { log.Printf("run time panic: %v", x) } }() log.Println("start") g() }
当g()中通过panic抛出错误时,会在defer中用recover进行捕获。也就是在子函数中的panic触发了其处于panicing状态,从而当panic回溯到当前函数时调用本函数的defer修饰的函数。
当然这里把g()替换成panic()也是可行的,就没有panic回溯了,直接时本函数中的panic触发使其处在panicing状态,从而调用defer修饰的函数。
修改返回值
如同上面的例子,如果对于有异常和没有异常返回不同的值,那么该如何操作。传统的方法是:
try { ... } catch (exception1){ return 1; } return 2 ...
由于Golang中不能在defer里面返回值,所以我们不能用上面的逻辑,在主函数和defer里面返回不同的结果。这个时候我们可以借助命名返回值来帮忙。在Golang中使用
func InvokeF()(rst int){ defer func(){ if err:=recover();err!=nil { if err== exception1 { rst = -1 } } }() rst = f() return }
当正常时,返回的rst为f()调用结果,当出现异常时,rst的值会被修改,从而达到目的。
4.defer中的坑
栈式调用
defer的调用顺序时栈式的,也就是后修饰的函数会被先调用。如果熟悉进程和线程API的化,这里defer的调用就和at_exit()有点类似,在退出时栈式调用注册过的退出函数。在使用中我们会发现这样的方式是符合我们代码原意的。假设A资源依赖B资源。那么我们这样:
resA ,err := getA() defer releaseA() ... resB,err := getB(resA) defer releaseB()
一来,我们在获取资源时就注册defer,语义上跟符合人类语言。二来,当A资源获取失败时仅释放A的环境;当B资源获取失败时,释放B和A的资源,且先释放依赖A的B。这样代码更为整洁易读。
出现时计算
首先要注意的是,defer修饰的函数调用中如果有参数,那么参数取的是当前计算时的值,而不是在return或者panic时该变量的值。
func main() { i := 2; deferFun := func (i int){fmt.Printf("i in panic is %d \n",i)} defer deferFun(i+1) i += 12 fmt.Printf("i befor return is %d \n",i) return }
得到结果为:
ibeforreturnis14iinpanicis3
注意,这里只是执行了参数的计算,而并没有执行函数体,函数体时在return之后执行的。
丢弃返回值
defer修饰过的函数调用的返回值时丢弃的,因此不要想着再去使用其返回值。本来也没有地方去使用。
本文出自 “Done_in_72_hours” 博客,请务必保留此出处http://gotaly.blog.51cto.com/8861157/1413402