Go语言中使用Defer几个场景
关于 <font face="NSimsun">defer</font> 的详细介绍请参考: Defer, Panic, and Recover .
C++ 中模拟的 <font face="NSimsun">defer</font> 实现请参考: C++版的defer语句
1. 简化资源的回收
这是最常见的 <font face="NSimsun">defer</font> 用法. 比如:
mu.Lock() defer mu.Unlock()
当然, <font face="NSimsun">defer</font> 也有一定的开销, 也有为了节省性能而回避使用的 <font face="NSimsun">defer</font> 的:
mu.Lock() count++ mu.Unlock()
从简化资源的释放角度看, <font face="NSimsun">defer</font> 类似一个语法糖, 好像不是必须的.
2. panic异常的捕获
<font face="NSimsun">defer</font> 除了用于简化资源的释放外, 还是Go语言异常框架的一个组成部分.
Go语言中, <font face="NSimsun">panic</font>用于抛出异常, <font face="NSimsun">recover</font>用于捕获异常. <font face="NSimsun">recover</font>只能在<font face="NSimsun">defer</font>语句中使用, 直接调用<font face="NSimsun">recover</font>是无效的.
比如:
func main() {
f()
fmt.Println("Returned normally from f.")
}
func f() {
defer func() {
if r := recover(); r != nil {
fmt.Println("Recovered in f", r)
}
}()
fmt.Println("Calling g.")
g()
fmt.Println("Returned normally from g.")
}
func g() {
panic("ERROR")
} 因此, 如果要捕获Go语言中函数的异常, 就离不开<font face="NSimsun">defer</font>语句了.
3. 修改返回值
<font face="NSimsun">defer</font> 除了用于配合 <font face="NSimsun">recover</font>, 用于捕获 <font face="NSimsun">panic</font> 异常外, 还可以用于在 <font face="NSimsun">return</font> 之后修改函数的返回值.
比如:
func doubleSum(a, b int) (sum int) {
defer func() {
sum *= 2
}()
sum = a + b
} 当然, 这个特性应该只是 <font face="NSimsun">defer</font> 的副作用, 具体在什么场景使用就要由开发者自己决定了.
4. 安全的回收资源
前面第一点提到, <font face="NSimsun">defer</font> 最常见的用法是简化资源的回收. 而且, 从资源回收角度看, <font face="NSimsun">defer</font> 只是一个语法糖.
其实, 也不完全是这样, 特别是在涉及到第二点提到的<font face="NSimsun">panic</font>异常等因素导致<font face="NSimsun">goroutine</font>提前退出时.
比如, 有一个线程安全的slice修改函数, 为了性能没有使用<font face="NSimsun">defer</font>语句:
func set(mu *sync.Mutex, arr []int, i, v int) {
mu.Lock()
arr[i] = v
mu.Unlock()
} 但是, 如果 <font face="NSimsun">i >= len(arr)</font>的话, <font face="NSimsun">runtime</font>就会抛出切片越界的异常(这里只是举例, 实际开发中不应该出现切片越界异常). 这样的话, <font face="NSimsun">mu.Unlock()</font> 就没有机会被执行了.
如果用<font face="NSimsun">defer</font>的话, 即使出现异常也能保证<font face="NSimsun">mu.Unlock()</font>被调用:
func set(mu *sync.Mutex, arr []int, i, v int) {
mu.Lock()
defer mu.Unlock()
arr[i] = v
} 当然, Go语言约定异常不会跨越<font face="NSimsun">package</font>边界. 因此, 调用一般函数的时候不用担心<font face="NSimsun">goroutine</font>异常退出的情况.
不过对于一些比较特殊的<font face="NSimsun">package</font>, 比如<font face="NSimsun">go test</font>依赖的<font face="NSimsun">testing</font>包, 包中的<font face="NSimsun">t.Fatal</font>就是依赖了Go中类似异常的特性(准确的说是调用了<font face="NSimsun">runtime.Goexit()</font>).
比如有以下的测试函数(详情请参考Issue5746):
func TestFailed(t *testing.T) {
var wg sync.WaitGroup
for i := 0; i < 2; i++ {
wg.Add(1)
go func(id int) {
// defer wg.Done()
t.Fatalf("TestFailed: id = %v\n", id)
wg.Done()
}(i)
}
wg.Wait()
} 当测试失败的时候, <font face="NSimsun">wg.Done()</font>将没有机会执行, 最终导致<font face="NSimsun">wg.Wait()</font>死锁.