使用Defer的几个场景

Go 语言中的 defer 语句是 UNIX 之父 Ken Thompson 大神发明的,是完全正交的设计.

也正因为 Go 语言遵循的是正交的设计,所以才有了: "少是指数级的多/Less is exponentially more" 的说法. 因为是正交的设计,最终得到的组合形式是指数级的组合形式.

相反,C++的特性虽然很多,但是很多不是正交的设计,而只是简单的特性罗列,所以C++的很多地方是无法达到指数级的多的组合方式的. 但是学习成本却非常高.

简单的例子就是C++的构造函数和析构函数和C语言的函数和struct完全是互斥的. 具体的例子可以参考: C++去掉构造函数会怎么样?

关于 Go 语言中 defer 语句的详细介绍请参考: Defer,Panic,and Recover .

C++ 中模拟的 defer 实现请参考: C++版的defer语句 .

这里主要是总结 defer 语句的一些使用场景.

1. 简化资源的回收

这是最常见的 defer 用法. 比如:

mu.Lock()
defer mu.Unlock()

当然,defer 也有一定的开销,也有为了节省性能而回避使用的 defer 的:

mu.Lock()
count++
mu.Unlock()

从简化资源的释放角度看,defer 类似一个语法糖,好像不是必须的.

2. panic异常的捕获

defer 除了用于简化资源的释放外,还是Go语言异常框架的一个组成部分.

Go语言中,panic用于抛出异常,recover用于捕获异常. recover只能在defer语句中使用,直接调用recover是无效的.

比如:

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语言中函数的异常,就离不开defer语句了.

3. 修改返回值

defer 除了用于配合 recover,用于捕获 panic 异常外,还可以用于在 return 之后修改函数的返回值.

比如:

func doubleSum(a,b int) (sum int) {
	defer func() {
		sum *= 2
	}()
	sum = a + b
}

当然,这个特性应该只是 defer 的副作用,具体在什么场景使用就要由开发者自己决定了.

4. 安全的回收资源

前面第一点提到,defer 最常见的用法是简化资源的回收. 而且,从资源回收角度看,defer 只是一个语法糖.

其实,也不完全是这样,特别是在涉及到第二点提到的panic异常等因素导致goroutine提前退出时.

比如,有一个线程安全的slice修改函数,为了性能没有使用defer语句:

func set(mu *sync.Mutex,arr []int,i,v int) {
	mu.Lock()
	arr[i] = v
	mu.Unlock()
}

但是,如果 i >= len(arr)的话,runtime就会抛出切片越界的异常(这里只是举例,实际开发中不应该出现切片越界异常). 这样的话,mu.Unlock() 就没有机会被执行了.

如果用defer的话,即使出现异常也能保证mu.Unlock()被调用:

func set(mu *sync.Mutex,v int) {
	mu.Lock()
	defer mu.Unlock()
	arr[i] = v
}

当然,Go语言约定异常不会跨越package边界. 因此,调用一般函数的时候不用担心goroutine异常退出的情况.

不过对于一些比较特殊的package,比如go test依赖的testing包,包中的t.Fatal就是依赖了Go中类似异常的特性(准确的说是调用了runtime.Goexit()).

比如有以下的测试函数(详情请参考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 = %vn",id)
			wg.Done()
		}(i)
	}
	wg.Wait()
}

当测试失败的时候,wg.Done()将没有机会执行,最终导致wg.Wait()死锁.

对于这个例子,安全的做法是使用defer语句保证wg.Done()始终会被执行.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!