go语言自定义的例子（Go语言之defer）

go语言自定义的例子（Go语言之defer）编译阶段在堆上分配的 defer例如，在一个执行次数不确定的 for 循环中，defer 的执行次数是随机的：1func randomDefers {2 rand.Seed(time.Now.UnixNano)3 for rand.Intn(100) > 42 {4 defer func {5 println("changkun.de/golang")6 }7 }8}因而 defer 并不是免费的午餐，在一个复杂的调用中，当无法直接确定需要的产生的延迟调用的数量时，延迟语句将导致运行性能的下降。本文我们来讨论 defer 的实现本质及其对症下药的相关性能优化手段。defer 的类型

作者 | 欧长坤

来源 | 码农桃花源

延迟语句 defer 在最早期的 Go 语言设计中并不存在，后来才单独增加了这一特性，由 Robert Griesemer 完成语言规范的编写 [Griesemer 2009]， 并由 Ken Thompson 完成最早期的实现 [Thompson 2009]，两人合作完成这一语言特性。

defer 的语义表明，它会在函数返回、产生恐慌或者 runtime.Goexit 时被调用。直觉上看，defer 应该由编译器直接将需要的函数调用插入到该调用的地方，似乎是一个编译期特性，不应该存在运行时性能问题，非常类似于 C 的 RAII 范式（当离开资源的作用域时，自动执行析构函数）。但实际情况是，由于 defer 并没有与其依赖资源挂钩，也允许在条件、循环语句中出现，从而不再是一个作用域相关的概念，这就是使得 defer 的语义变得相对复杂。在一些复杂情况下，无法在编译期决定存在多少个 defer 调用。

例如，在一个执行次数不确定的 for 循环中，defer 的执行次数是随机的：

1func randomDefers {
2 rand.Seed(time.Now.UnixNano)
3 for rand.Intn(100) > 42 {
4 defer func {
5 println("changkun.de/golang")
6 }
7 }
8}


因而 defer 并不是免费的午餐，在一个复杂的调用中，当无法直接确定需要的产生的延迟调用的数量时，延迟语句将导致运行性能的下降。本文我们来讨论 defer 的实现本质及其对症下药的相关性能优化手段。

• defer 的类型

• 在堆上分配的 defer

• 编译阶段

• 运行阶段

• 在栈上创建 defer

• 开放编码式 defer

• 产生条件

• 延迟比特

• defer 的优化之路

• 小结

• 进一步阅读的参考文献

defer 的类型

延迟语句的文法产生式 DeferStmt -> "defer" Expression 的描述非常的简单，因而也很容易将其处理为语法树的形式，但我们这里更关心的其实是它语义背后的中间和目标代码的形式。

在 《Go 语言原本》Go 程序编译流程 一节中我们提到过，在进行中间代码生成阶段，会通过 compileSSA 先调用 buildssa 为函数体生成 SSA 形式的函数，而后调用 genssa 将函数的 SSA 中间表示转换为具体的指令。

Go 语言的语句在执行 buildssa 阶段中，会由 state.stmt 完成函数中各个语句的 SSA 处理。

1// src/cmd/compile/internal/gc/ssa.go
2func buildssa(fn *Node worker int) *ssa.Func {
3 var s state
4 ...
5 s.stmtList(fn.Nbody)
6 ...
7}
8func (s *state) stmtList(l Nodes) {
9 for _ n := range l.Slice { s.stmt(n) }
10}


对于延迟语句而言，其中间表示会产生三种不同的延迟形式， 第一种是最一般情况下的在堆上分配的延迟语句，第二种是允许在栈上分配的延迟语句，最后一种则是**开放编码式（Open-coded）**的延迟语句。

1// src/cmd/compile/internal/gc/ssa.go
2func (s *state) stmt(n *Node) {
3 ...
4 switch n.Op {
5 case ODEFER:
6 // 开放编码式 defer
7 if s.hasOpenDefers {
8 s.openDeferRecord(n.Left)
9 } else {
10 // 堆上分配的 defer
11 d := callDefer
12 if n.Esc == EscNever {
13 // 栈上分配的 defer
14 d = callDeferStack
15 }
16 s.call(n.Left d)
17 }
18 case ...
19 }
20 ...
21}


在堆上分配的 defer

我们先来讨论最简单的在堆上分配的 defer 这种形式。在堆上分配的原因是 defer 语句出现在了循环语句里，或者无法执行更高阶的编译器优化导致的。

如果一个与 defer 出现在循环语句中，则可执行的次数可能无法在编译期决定；如果一个调用中 defer 由于数量过多等原因，不能被编译器进行开放编码，则也会在堆上分配 defer。

总之，由于这种不确定性的存在，在堆上分配的 defer 需要最多的运行时支持，因而产生的运行时开销也最大。

编译阶段

为了使延迟语句的功能满足语言规范，该语句在编译的 SSA 阶段会被翻译为两个主体，其中第一个主体是被延迟的函数本身，另一个主体则是函数结束时需要执行所记录 defer 的代码块。

state.call 调用会生成用于记录延迟调用参数的指令，并创建一个 deferproc 的调用指令；而后 state.exit 调用在函数返回前插入 deferreturn 调用的指令。

1// src/cmd/compile/internal/gc/ssa.go
2func (s *state) call(n *Node k callKind) *ssa.Value {
3 ...
4 var call *ssa.Value
5 if k == callDeferStack {
6 ...
7 } else {
8 // 在堆上创建 defer
9 argStart := Ctxt.FixedFrameSize
10 // Defer 参数
11 if k != callNormal {
12 // 记录 deferproc 的参数
13 argsize := s.constInt32(types.Types[TUINT32] int32(stksize))
14 addr := s.constOffPtrSP(s.f.Config.Types.UInt32Ptr argStart)
15 s.store(types.Types[TUINT32] addr argsize) // 保存参数大小 siz
16 addr = s.constOffPtrSP(s.f.Config.Types.UintptrPtr argStart int64(Widthptr))
17 s.store(types.Types[TUINTPTR] addr closure) // 保存函数地址 fn
18 stksize = 2 * int64(Widthptr)
19 argStart = 2 * int64(Widthptr)
20 }
21 ...
22
23 // 创建 deferproc 调用
24 switch {
25 case k == callDefer:
26 call = s.newValue1A(ssa.OpStaticCall types.TypeMem deferproc s.mem)
27 ...
28 }
29 ...
30 }
31 ...
32
33 // 结束 defer 块
34 if k == callDefer || k == callDeferStack {
35 s.exit
36 ...
37 }
38 ...
39}
40func (s *state) exit *ssa.Block {
41 if s.hasdefer {
42 if s.hasOpenDefers {
43 ...
44 } else {
45 // 调用 deferreturn
46 s.rtcall(Deferreturn true nil)
47 }
48 }
49 ...
50}


例如，对于一个纯粹的 defer 调用而言：

1package main
2
3func foo {
4 return
5}
6
7func main {
8 defer foo
9 return
10}


如果我们将其强制编译为在堆上分配的形式，可以观察到如下的汇编代码。其中 defer foo被转化为了 deferproc 调用，并在函数返回前，调用了 deferreturn：

1TEXT main.foo(SB) /Users/changkun/Desktop/defer/ssa/main.go
2 return
3 0x104ea20 c3 RET
4
5TEXT main.main(SB) /Users/changkun/Desktop/defer/ssa/main.go
6func main {
7 ...
8 // 将 defer foo { ... } 转化为一个 deferproc 调用
9 // 在调用 deferproc 前完成参数的准备工作，这个例子中没有参数
10 0x104ea4d c7042400000000 MOVL $0x0 0(SP)
11 0x104ea54 488d0585290200 LEAQ go.func.* 60(SB) AX
12 0x104ea5b 4889442408 MOVQ AX 0x8(SP)
13 0x104ea60 e8bb31fdff CALL runtime.deferproc(SB)
14 ...
15 // 函数返回指令 RET 前插入的 deferreturn 语句
16 0x104ea7b 90 NOPL
17 0x104ea7c e82f3afdff CALL runtime.deferreturn(SB)
18 0x104ea81 488b6c2410 MOVQ 0x10(SP) BP
19 0x104ea86 4883c418 ADDQ $0x18 SP
20 0x104ea8a c3 RET
21 // 函数的尾声
22 0x104ea8b e8d084ffff CALL runtime.morestack_noctxt(SB)
23 0x104ea90 eb9e JMP main.main(SB)

运行阶段

一个函数中的延迟语句会被保存为一个 _defer 记录的链表，附着在一个 Goroutine 上。_defer 记录的具体结构也非常简单，主要包含了参与调用的参数大小、当前 defer 语句所在函数的 PC 和 SP 寄存器、被 defer 的函数的入口地址以及串联多个 defer 的 link 链表，该链表指向下一个需要执行的 defer，如图 9.2.1 所示。

1// src/runtime/panic.go
2type _defer struct {
3 siz int32
4 heap bool
5 sp uintptr
6 pc uintptr
7 fn *funcval
8 link *_defer
9 ...
10}
11// src/runtime/runtime2.go
12type g struct {
13 ...
14 _defer *_defer
15 ...
16}


附着在 Goroutine 上的 _defer 记录的链表

现在我们知道，一个在堆上分配的延迟语句被编译为了 deferproc，用于记录被延迟的函数调用；在函数的尾声，会插入 deferreturn 调用，用于执行被延迟的调用。

下面我们就来详细看看这两个调用具体发生了什么事情。

我们先看创建 defer 的第一种形式 deferproc。这个调用很简单，仅仅只是将需要被 defer 调用的函数做了一次记录：

1//go:nosplit
2func deferproc(siz int32 fn *funcval) {
3 ...
4 sp := getcallersp
5 argp := uintptr(unsafe.Pointer(&fn)) unsafe.Sizeof(fn)
6 callerpc := getcallerpc
7
8 d := newdefer(siz)
9 d.fn = fn
10 d.pc = callerpc
11 d.sp = sp
12
13 // 将参数保存到 _defer 记录中
14 switch siz {
15 case 0: // 什么也不做
16 case sys.PtrSize:
17 *(*uintptr)(deferArgs(d)) = *(*uintptr)(unsafe.Pointer(argp))
18 default:
19 memmove(deferArgs(d) unsafe.Pointer(argp) uintptr(siz))
20 }
21
22 return0
23}


这段代码中，本质上只是在做一些简单参数处理，比如 fn 保存了 defer 所调用函数的调用地址，siz 确定了其参数的大小。并且通过 newdefer 来创建一个新的 _defer 实例，然后由 fn、callerpc 和 sp 来保存调用该 defer 的 Goroutine 上下文。

注意，在这里我们看到了一个对参数进行拷贝的操作。这个操作也是我们在实践过程中经历过的，defer 调用被记录时，并不会对参数进行求值，而是会对参数完成一次拷贝。这么做原因是由于语义上的考虑。直觉上讲，defer 的参数应当在它所写的位置对传入的参数进行求值，而不是将求值步骤推迟，因为延后的参数可能发生变化，导致 defer 的语义发生意料之外的错误。

例如，f _ := os.Open("file.txt") 后立刻指定 defer f.Close，倘若随后的语句修改了 f 的值，那么将导致 f 无法被正常关闭。

出于性能考虑，newdefer 通过 P 或者调度器 sched 上的本地或全局 defer 池来复用已经在堆上分配的内存。defer 的资源池会根据被延迟的调用所需的参数来决定 defer 记录的大小等级，每 16 个字节分一个等级。此做法的动机与运行时内存分配器针对不同大小对象的分配思路雷同，这里不再做深入讨论。

1// src/runtime/runtime2.go
2type p struct {
3 ...
4 // 不同大小的本地 defer 池
5 deferpool [5]*_defer
6 deferpoolbuf [5][32]*_defer
7 ...
8}
9type schedt struct {
10 ...
11 // 不同大小的全局 defer 池
12 deferlock mutex
13 deferpool [5]*_defer
14 ...
15}


对于新建的 _defer 实例而言，会将其加入到 Goroutine 所保留的 defer 链表上，通过 link 字段串联:

1// src/runtime/panic.go
2
3//go:nosplit
4func newdefer(siz int32) *_defer {
5 var d *_defer
6 sc := deferclass(uintptr(siz))
7 gp := getg
8 // 检查 defer 参数的大小是否从 p 的 deferpool 直接分配
9 if sc < uintptr(len(p{}.deferpool)) {
10 pp := gp.m.p.ptr
11
12 // 如果 p 本地无法分配，则从全局池中获取一半 defer，来填充 P 的本地资源池
13 if len(pp.deferpool[sc]) == 0 && sched.deferpool[sc] != nil {
14 // 出于性能考虑，如果发生栈的增长，则会调用 morestack，
15 // 进一步降低 defer 的性能。因此切换到系统栈上执行，进而不会发生栈的增长。
16 systemstack(func {
17 lock(&sched.deferlock)
18 for len(pp.deferpool[sc]) < cap(pp.deferpool[sc])/2 && sched.deferpool[sc] != nil {
19 d := sched.deferpool[sc]
20 sched.deferpool[sc] = d.link
21 d.link = nil
22 pp.deferpool[sc] = append(pp.deferpool[sc] d)
23 }
24 unlock(&sched.deferlock)
25 })
26 }
27
28 // 从 P 本地进行分配
29 if n := len(pp.deferpool[sc]); n > 0 {
30 d = pp.deferpool[sc][n-1]
31 pp.deferpool[sc][n-1] = nil
32 pp.deferpool[sc] = pp.deferpool[sc][:n-1]
33 }
34 }
35 // 没有可用的缓存，直接从堆上分配新的 defer 和 args
36 if d == nil {
37 systemstack(func {
38 total := roundupsize(totaldefersize(uintptr(siz)))
39 d = (*_defer)(mallocgc(total deferType true))
40 })
41 }
42 // 将 _defer 实例添加到 Goroutine 的 _defer 链表上。
43 d.siz = siz
44 d.heap = true
45 d.link = gp._defer
46 gp._defer = d
47 return d
48}


deferreturn 被编译器插入到函数末尾，当跳转到它时，会将需要被 defer 的入口地址取出，然后跳转并执行：

1// src/runtime/panic.go
2
3//go:nosplit
4func deferreturn(arg0 uintptr) {
5 gp := getg
6 d := gp._defer
7 if d == nil {
8 return
9 }
10 // 确定 defer 的调用方是不是当前 deferreturn 的调用方
11 sp := getcallersp
12 if d.sp != sp {
13 return
14 }
15 ...
16
17 // 将参数复制出 _defer 记录外
18 switch d.siz {
19 case 0: // 什么也不做
20 case sys.PtrSize:
21 *(*uintptr)(unsafe.Pointer(&arg0)) = *(*uintptr)(deferArgs(d))
22 default:
23 memmove(unsafe.Pointer(&arg0) deferArgs(d) uintptr(d.siz))
24 }
25 // 获得被延迟的调用 fn 的入口地址，并随后立即将 _defer 释放掉
26 fn := d.fn
27 d.fn = nil
28 gp._defer = d.link
29 freedefer(d)
30
31 // 调用，并跳转到下一个 defer
32 jmpdefer(fn uintptr(unsafe.Pointer(&arg0)))
33}


在这个函数中，会在需要时对 defer 的参数再次进行拷贝，多个 defer 函数以 jmpdefer 尾调用形式被实现。在跳转到 fn 之前，_defer 实例被释放归还，jmpdefer 真正需要的仅仅只是函数的入口地址和参数，以及它的调用方 deferreturn 的 SP：

1// src/runtime/asm_amd64.s
2
3// func jmpdefer(fv *funcval argp uintptr)
4TEXT runtime·jmpdefer(SB) NOSPLIT $0-16
5 MOVQ fv 0(FP) DX // DX = fn
6 MOVQ argp 8(FP) BX // 调用方 SP
7 LEAQ -8(BX) SP // CALL 后的调用方 SP
8 MOVQ -8(SP) BP // 恢复 BP，好像 deferreturn 返回
9 SUBQ $5 (SP) // 再次返回到 CALL
10 MOVQ 0(DX) BX // BX = DX
11 JMP BX // 最后才运行被 defer 的函数


这个 jmpdefer 巧妙的地方在于，它通过调用方 SP 来推算了 deferreturn的入口地址，从而在完成某个 defer 调用后，由于被 defer 的函数返回时会出栈，会再次回到 deferreturn 的初始位置，进而继续反复调用，从而模拟 deferreturn 不断地对自己进行尾递归的假象。

释放操作非常普通，只是简单地将其归还到 P 的 deferpool 中， 并在本地池已满时将其归还到全局资源池:

1// src/runtime/panic.go
2
3//go:nosplit
4func freedefer(d *_defer) {
5 ...
6 sc := deferclass(uintptr(d.siz))
7 if sc >= uintptr(len(p{}.deferpool)) {
8 return
9 }
10 pp := getg.m.p.ptr
11 // 如果 P 本地池已满，则将一半资源放入全局池，同样也是出于性能考虑
12 // 操作会切换到系统栈上执行。
13 if len(pp.deferpool[sc]) == cap(pp.deferpool[sc]) {
14 systemstack(func {
15 var first last *_defer
16 for len(pp.deferpool[sc]) > cap(pp.deferpool[sc])/2 {
17 n := len(pp.deferpool[sc])
18 d := pp.deferpool[sc][n-1]
19 pp.deferpool[sc][n-1] = nil
20 pp.deferpool[sc] = pp.deferpool[sc][:n-1]
21 if first == nil {
22 first = d
23 } else {
24 last.link = d
25 }
26 last = d
27 }
28 lock(&sched.deferlock)
29 last.link = sched.deferpool[sc]
30 sched.deferpool[sc] = first
31 unlock(&sched.deferlock)
32 })
33 }
34
35 // 恢复 _defer 的零值，即 *d = _defer{}
36 d.siz = 0
37 ...
38 d.sp = 0
39 d.pc = 0
40 d.framepc = 0
41 ...
42 d.link = nil
43
44 // 放入 P 本地资源池
45 pp.deferpool[sc] = append(pp.deferpool[sc] d)
46}


在栈上创建 defer

defer 还可以直接在栈上进行分配，也就是第二种记录 defer 的形式 deferprocStack。在栈上分配 defer 的好处在于函数返回后 _defer 便已得到释放，不再需要考虑内存分配时产生的性能开销，只需要适当地维护 _defer 的链表即可。

在 SSA 阶段与在堆上分配的区别在于，在栈上创建 defer， 需要直接在函数调用帧上使用编译器来初始化 _defer 记录，并作为参数传递给 deferprocStack：

1// src/cmd/compile/internal/gc/ssa.go
2func (s *state) call(n *Node k callKind) *ssa.Value {
3 ...
4 var call *ssa.Value
5 if k == callDeferStack {
6 // 直接在栈上创建 defer 记录
7 t := deferstruct(stksize) // 从编译器角度构造 _defer 结构
8 d := tempAt(n.Pos s.curfn t)
9
10 s.vars[&memVar] = s.newValue1A(ssa.OpVarDef types.TypeMem d s.mem)
11 addr := s.addr(d false)
12
13 // 在栈上预留记录 _defer 的各个字段的空间
14 s.store(types.Types[TUINT32]
15 s.newValue1I(ssa.OpOffPtr types.Types[TUINT32].PtrTo t.FieldOff(0) addr)
16 s.constInt32(types.Types[TUINT32] int32(stksize)))
17 s.store(closure.Type
18 s.newValue1I(ssa.OpOffPtr closure.Type.PtrTo t.FieldOff(6) addr)
19 closure)
20
21 // 记录参与 defer 调用的函数参数
22 ft := fn.Type
23 off := t.FieldOff(12)
24 args := n.Rlist.Slice
25
26 // 调用 deferprocStack，以 _defer 记录的指针作为参数传递
27 arg0 := s.constOffPtrSP(types.Types[TUINTPTR] Ctxt.FixedFrameSize)
28 s.store(types.Types[TUINTPTR] arg0 addr)
29 call = s.newValue1A(ssa.OpStaticCall types.TypeMem deferprocStack s.mem)
30 ...
31 } else { ... }
32
33 // 函数尾声与堆上分配的栈一样，调用 deferreturn
34 if k == callDefer || k == callDeferStack {
35 ...
36 s.exit
37 }
38 ...
39 }

可见，在编译阶段，一个 _defer 记录的空间已经在栈上得到保留，deferprocStack 的作用就仅仅承担了运行时对该记录的初始化这一功能：

1// src/runtime/panic.go
2
3//go:nosplit
4func deferprocStack(d *_defer) {
5 gp := getg
6 // 注意，siz 和 fn 已经在编译阶段完成设置，这里只初始化了其他字段
7 d.started = false
8 d.heap = false // 可见此时 defer 被标记为不在堆上分配
9 d.openDefer = false
10 d.sp = getcallersp
11 d.pc = getcallerpc
12 ...
13 // 尽管在栈上进行分配，仍然需要将多个 _defer 记录通过链表进行串联，
14 // 以便在 deferreturn 中找到被延迟的函数的入口地址：
15 // d.link = gp._defer
16 // gp._defer = d
17 *(*uintptr)(unsafe.Pointer(&d.link)) = uintptr(unsafe.Pointer(gp._defer))
18 *(*uintptr)(unsafe.Pointer(&gp._defer)) = uintptr(unsafe.Pointer(d))
19 return0
20 }

至于函数尾声的行为，与在堆上进行分配的操作同样是调用 deferreturn，我们就不再重复说明了。当然，里面涉及的 freedefer 调用由于不需要释放任何内存，也就早早返回了：

1// src/runtime/panic.go
2func freedefer(d *_defer) {
3 if !d.heap { return }
4 ...
5}


开放编码式 defer

正如本文最初所描述的那样，defer 给我们的第一感觉其实是一个编译期特性。前面我们讨论了为什么 defer 会需要运行时的支持，以及需要运行时的 defer 是如何工作的。现在我们来探究一下什么情况下能够让 defer 进化为一个仅编译期特性，即在函数末尾直接对延迟函数进行调用，做到几乎不需要额外的开销。这类几乎不需要额外运行时性能开销的 defer，正是开放编码式 defer。这类 defer 与直接调用产生的性能差异有多大呢？我们不妨编写两个性能测试：

1func call { func {} }
2func callDefer { defer func {} }
3func BenchmarkDefer(b *testing.B) {
4 for i := 0; i < b.N; i {
5 call // 第二次运行时替换为 callDefer
6 }
7}

在 Go 1.14 版本下，读者可以获得类似下方的性能估计，其中使用 callDefer后，性能损耗大约为 1 ns。这种纳秒级的性能损耗不到一个 CPU 时钟周期，我们已经可以认为开放编码式 defer 几乎没有了性能开销：

1ame old time/op new time/op delta
2Defer-12 1.24ns ± 1% 2.23ns ± 1% 80.06% (p=0.000 n=10 9)


我们再来观察一下开放编码式 defer 最终被编译的形式：

1$ go build -gcflags "-l" -ldflags=-compressdwarf=false -o main.out main.go
2$ go tool objdump -S main.out > main.s


对于如下形式的函数调用：

1var mu sync.Mutex
2func callDefer {
3 mu.Lock
4 defer mu.Unlock
5}


整个调用最终编译结果既没有 deferproc 或者 deferprocStack，也没有了 deferreturn。延迟语句被直接插入到了函数的末尾：

1TEXT main.callDefer(SB) /Users/changkun/Desktop/defer/main.go
2func callDefer {
3 ...
4 mu.Lock
5 0x105794a 488d05071f0a00 LEAQ main.mu(SB) AX
6 0x1057951 48890424 MOVQ AX 0(SP)
7 0x1057955 e8f6f8ffff CALL sync.(*Mutex).Lock(SB)
8 defer mu.Unlock
9 0x105795a 488d057f110200 LEAQ go.func.* 1064(SB) AX
10 0x1057961 4889442418 MOVQ AX 0x18(SP)
11 0x1057966 488d05eb1e0a00 LEAQ main.mu(SB) AX
12 0x105796d 4889442410 MOVQ AX 0x10(SP)
13}
14 0x1057972 c644240f00 MOVB $0x0 0xf(SP)
15 0x1057977 488b442410 MOVQ 0x10(SP) AX
16 0x105797c 48890424 MOVQ AX 0(SP)
17 0x1057980 e8ebfbffff CALL sync.(*Mutex).Unlock(SB)
18 0x1057985 488b6c2420 MOVQ 0x20(SP) BP
19 0x105798a 4883c428 ADDQ $0x28 SP
20 0x105798e c3 RET
21 ...


那么开放编码式 defer 是怎么实现的？所有的 defer 都是开放编码式的吗？什么情况下，开放编码式 defer 会退化为一个依赖运行时的特性？

产生条件

我们先来看开放编码式 defer 的产生条件。在 SSA 的构建阶段 buildssa，我们有：

1// src/cmd/compile/internal/gc/ssa.go
2const maxOpenDefers = 8
3func walkstmt(n *Node) *Node {
4 ...
5 switch n.Op {
6 case ODEFER:
7 Curfn.Func.SetHasDefer(true)
8 Curfn.Func.numDefers
9 // 超过 8 个 defer 时，禁用对 defer 进行开放编码
10 if Curfn.Func.numDefers > maxOpenDefers {
11 Curfn.Func.SetOpenCodedDeferDisallowed(true)
12 }
13 // 存在循环语句中的 defer，禁用对 defer 进行开放编码。
14 // 是否有 defer 发生在循环语句内，会在 SSA 之前的逃逸分析中进行判断，
15 // 逃逸分析会检查是否存在循环（loopDepth）：
16 // if where.Op == ODEFER && e.loopDepth == 1 {
17 // where.Esc = EscNever
18 // ...
19 // }
20 if n.Esc != EscNever {
21 Curfn.Func.SetOpenCodedDeferDisallowed(true)
22 }
23 case ...
24 }
25 ...
26}
27
28func buildssa(fn *Node worker int) *ssa.Func {
29 ...
30 var s state
31 ...
32 s.hasdefer = fn.Func.HasDefer
33 ...
34 // 可以对 defer 进行开放编码的条件
35 s.hasOpenDefers = Debug['N'] == 0 && s.hasdefer && !s.curfn.Func.OpenCodedDeferDisallowed
36 if s.hasOpenDefers &&
37 s.curfn.Func.numReturns*s.curfn.Func.numDefers > 15 {
38 s.hasOpenDefers = false
39 }
40 ...
41}


这样，我们得到了允许进行 defer 的开放编码的主要条件（此处略去了一些常见生产环境无关的条件，例如启用竞争检查时也不能对 defer 进行开放编码）：

1. 没有禁用编译器优化，即没有设置 -gcflags "-N"

2. 存在 defer 调用

3. 函数内 defer 的数量不超过 8 个、且返回语句与延迟语句个数的乘积不超过 15

4. 没有 defer 发生在循环语句中

延迟比特

当然，正常编写的 defer 可以直接被编译器分析得出，但是如本文开头提到的，如果一个 defer 发生在一个条件语句中，而这个条件必须等到运行时才能确定：

1if rand.Intn(100) < 42 {
2 defer fmt.Println("meaning-of-life")
3}


那么如何才能使用最小的成本，让插入到函数末尾的延迟语句，在条件成立时候被正确执行呢？这便需要一种机制，能够记录存在延迟语句的条件分支是否被执行，这种机制在 Go 中利用了延迟比特（defer bit）。这种做法非常巧妙，但原理却非常简单。

对于下面的代码而言：

1defer f1(a1)
2if cond {
3 defer f2(a2)
4}
5...


使用延迟比特的核心思想可以用下面的伪代码来概括。在创建延迟调用的阶段，首先通过延迟比特的特定位置记录哪些带条件的 defer 被触发。这个延迟比特是一个长度为 8 位的二进制码（也是硬件架构里最小、最通用的情况），以每一位是否被设置为 1，来判断延迟语句是否在运行时被设置，如果设置，则发生调用。否则则不调用：

1deferBits = 0 // 初始值 00000000
2deferBits |= 1 << 0 // 遇到第一个 defer，设置为 00000001
3_f1 = f1
4_a1 = a1
5if cond {
6 // 如果第二个 defer 被设置，则设置为 00000011，否则依然为 00000001
7 deferBits |= 1 << 1
8 _f2 = f2
9 _a2 = a2
10}


在退出位置，再重新根据被标记的延迟比特，反向推导哪些位置的 defer 需要被触发，从而执行延迟调用：

1exit:
2// 按顺序倒序检查延迟比特。如果第二个 defer 被设置，则
3// 00000011 & 00000010 == 00000010，即延迟比特不为零，应该调用 f2。
4// 如果第二个 defer 没有被设置，则
5// 00000001 & 00000010 == 00000000，即延迟比特为零，不应该调用 f2。
6if deferBits & 1 << 1 != 0 { // 00000011 & 00000010 != 0
7 deferBits &^= 1<<1 // 00000001
8 _f2(_a2)
9}
10// 同理，由于 00000001 & 00000001 == 00000001，因此延迟比特不为零，应该调用 f1
11if deferBits && 1 << 0 != 0 {
12 deferBits &^= 1<<0
13 _f1(_a1)
14}


在实际的实现中，可以看到，当可以设置开放编码式 defer 时，buildssa 会首先创建一个长度位 8 位的临时变量：

1// src/cmd/compile/internal/gc/ssa.go
2func buildssa(fn *Node worker int) *ssa.Func {
3 ...
4 if s.hasOpenDefers {
5 // 创建 deferBits 临时变量
6 deferBitsTemp := tempAt(src.NoXPos s.curfn types.Types[TUINT8])
7 s.deferBitsTemp = deferBitsTemp
8 // deferBits 被设计为 8 位二进制，因此可以被开放编码的 defer 数量不能超过 8 个
9 // 此处还将起始 deferBits 设置为零
10 startDeferBits := s.entryNewValue0(ssa.OpConst8 types.Types[TUINT8])
11 s.vars[&deferBitsVar] = startDeferBits
12 s.deferBitsAddr = s.addr(deferBitsTemp false)
13 s.store(types.Types[TUINT8] s.deferBitsAddr startDeferBits)
14 ...
15 }
16 ...
17 s.stmtList(fn.Nbody) // 调用 s.stmt
18 ...
19}

随后针对出现 defer 的语句，进行编码：

1// src/cmd/compile/internal/gc/ssa.go
2func (s *state) stmt(n *Node) {
3 ...
4 switch n.Op {
5 case ODEFER:
6 // 开放编码式 defer
7 if s.hasOpenDefers {
8 s.openDeferRecord(n.Left)
9 } else { ... }
10 case ...
11 }
12 ...
13}
14
15// 存储一个 defer 调用的相关信息，例如所在的语法树结点、被延迟的调用、参数等等
16type openDeferInfo struct {
17 n *Node
18 closure *ssa.Value
19 closureNode *Node
20 ...
21 argVals *ssa.Value
22 argNodes *Node
23}
24func (s *state) openDeferRecord(n *Node) {
25 ...
26 var args *ssa.Value
27 var argNodes *Node
28
29 // 记录与 defer 相关的入口地址与参数信息
30 opendefer := &openDeferInfo{n: n}
31 fn := n.Left
32 // 记录函数入口地址
33 if n.Op == OCALLFUNC {
34 closureVal := s.expr(fn)
35 closure := s.openDeferSave(nil fn.Type closureVal)
36 opendefer.closureNode = closure.Aux.(*Node)
37 if !(fn.Op == ONAME && fn.Class == PFUNC) {
38 opendefer.closure = closure
39 }
40 } else {
41 ...
42 }
43 // 记录需要立即求值的的参数
44 for _ argn := range n.Rlist.Slice {
45 var v *ssa.Value
46 if canSSAType(argn.Type) {
47 v = s.openDeferSave(nil argn.Type s.expr(argn))
48 } else {
49 v = s.openDeferSave(argn argn.Type nil)
50 }
51 args = append(args v)
52 argNodes = append(argNodes v.Aux.(*Node))
53 }
54 opendefer.argVals = args
55 opendefer.argNodes = argNodes
56
57 // 每多出现一个 defer，len(defers) 会增加，进而
58 // 延迟比特 deferBits |= 1<<len(defers) 被设置在不同的位上
59 index := len(s.openDefers)
60 s.openDefers = append(s.openDefers opendefer)
61 bitvalue := s.constInt8(types.Types[TUINT8] 1<<uint(index))
62 newDeferBits := s.newValue2(ssa.OpOr8 types.Types[TUINT8] s.variable(&deferBitsVar types.Types[TUINT8]) bitvalue)
63 s.vars[&deferBitsVar] = newDeferBits
64 s.store(types.Types[TUINT8] s.deferBitsAddr newDeferBits)
65 }

在函数返回退出前，state 的 exit 函数会依次倒序创建对延迟比特的检查代码，从而顺序调用被延迟的函数调用：

1// src/cmd/compile/internal/gc/ssa.go
2func (s *state) exit *ssa.Block {
3 if s.hasdefer {
4 if s.hasOpenDefers {
5 ...
6 s.openDeferExit
7 } else {
8 ...
9 }
10 }
11 ...
12}
13
14func (s *state) openDeferExit {
15 deferExit := s.f.NewBlock(ssa.BlockPlain)
16 s.endBlock.AddEdgeTo(deferExit)
17 s.startBlock(deferExit)
18 s.lastDeferExit = deferExit
19 s.lastDeferCount = len(s.openDefers)
20 zeroval := s.constInt8(types.Types[TUINT8] 0)
21 // 倒序检查 defer
22 for i := len(s.openDefers) - 1; i >= 0; i-- {
23 r := s.openDefers[i]
24 bCond := s.f.NewBlock(ssa.BlockPlain)
25 bEnd := s.f.NewBlock(ssa.BlockPlain)
26
27 // 检查 deferBits
28 deferBits := s.variable(&deferBitsVar types.Types[TUINT8])
29 // 创建 if deferBits & 1 << len(defer) != 0 { ... }
30 bitval := s.constInt8(types.Types[TUINT8] 1<<uint(i))
31 andval := s.newValue2(ssa.OpAnd8 types.Types[TUINT8] deferBits bitval)
32 eqVal := s.newValue2(ssa.OpEq8 types.Types[TBOOL] andval zeroval)
33 b := s.endBlock
34 b.Kind = ssa.BlockIf
35 b.SetControl(eqVal)
36 b.AddEdgeTo(bEnd)
37 b.AddEdgeTo(bCond)
38 bCond.AddEdgeTo(bEnd)
39 s.startBlock(bCond)
40
41 // 如果创建的条件分支被触发，则清空当前的延迟比特: deferBits &^= 1 << len(defers)
42 nbitval := s.newValue1(ssa.OpCom8 types.Types[TUINT8] bitval)
43 maskedval := s.newValue2(ssa.OpAnd8 types.Types[TUINT8] deferBits nbitval)
44 s.store(types.Types[TUINT8] s.deferBitsAddr maskedval)
45 s.vars[&deferBitsVar] = maskedval
46
47 // 处理被延迟的函数调用，取出保存的入口地址、参数信息
48 argStart := Ctxt.FixedFrameSize
49 fn := r.n.Left
50 stksize := fn.Type.ArgWidth
51 ...
52 for j argAddrVal := range r.argVals {
53 f := getParam(r.n j)
54 pt := types.NewPtr(f.Type)
55 addr := s.constOffPtrSP(pt argStart f.Offset)
56 if !canSSAType(f.Type) {
57 s.move(f.Type addr argAddrVal)
58 } else {
59 argVal := s.load(f.Type argAddrVal)
60 s.storeType(f.Type addr argVal 0 false)
61 }
62 }
63 // 调用
64 var call *ssa.Value
65 ...
66 call = s.newValue1A(ssa.OpStaticCall types.TypeMem fn.Sym.Linksym s.mem)
67 call.AuxInt = stksize
68 s.vars[&memVar] = call
69 ...
70 s.endBlock
71 s.startBlock(bEnd)
72 }
73}


从整个过程中我们可以看到，开放编码式 defer 并不是绝对的零成本，尽管编译器能够做到将延迟调用直接插入返回语句之前，但出于语义的考虑，需要在栈上对参与延迟调用的参数进行一次求值；同时出于条件语句中可能存在的 defer，还额外需要通过延迟比特来记录一个延迟语句是否在运行时被设置。因此，开放编码式 defer 的成本体现在非常少量的指令和位运算来配合在运行时判断是否存在需要被延迟调用的 defer。

defer 的优化之路

我们最后来回顾一下延迟语句的整个演进过程。

defer 的早期实现其实是非常的粗糙的。每当出现一个 defer 调用，都会在堆上分配 defer 记录，并对参与调用的参数实施一次拷贝操作，然后将其加入到 defer 链表上；当函数返回需要触发 defer 调用时，依次将 defer 从链表中取出，完成调用。当然最初的实现并不需要完美，未来总是可以迭代其性能问题。

在 Go 1.1 的开发阶段，defer 获得了它的第一次优化 [Cox 2011]。Russ Cox 意识到 defer 性能问题的根源是当产生多个 defer 调用时，造成的过多的内存分配与拷贝操作，进而提出将 defer 的分配和释放过程在每个 Goroutine 内进行批量处理。当时 Dmitry Vyukov 则提议在栈上分配会更加有效，但 Russ Cox 错误的认为在执行栈上分配 defer 记录与在其他地方进行分配并没有带来太多收益，最终实现了 per-G 批量式分配的 defer 机制。

由于后续调度器的改进，工作窃取调度的引入，运行时开始支持 per-P 的局部资源池，defer 作为发生在 Goroutine 内的调用，所需的内存自然也是一类可以被视作局部持有的资源。因此分配和释放 defer 的资源在 Go 1.3 时得到优化 [Vyukov 2014]，Dmitry Vyukov 将 per-G 分配的 defer 改为了从 per-P 资源池分配的机制。

由于分配延迟记录 _defer 的调用 newdefer 可能存在本地资源池、全局资源池均不存在可复用的内存，进而导致栈分裂，更糟糕的情况下甚至可能发生抢占，导致 M/P 解绑与绑定等额外的调度开销。因此，Austin Clements 对 defer 做的一个优化 [Clements 2016] 是在每个 deferproc 和 deferreturn 中都切换至系统栈，从而阻止了抢占和栈增长的发生，也就优化消除了抢占带来的 M/P 绑定所带来的开销。除此之外，对于每次产生记录时，无论参数大小如何都涉及 memmove 系统调用，从而产生一次 memmove 的调用成本，Austin 的优化中还特地针对没有参数和指针大小参数的这两种情况进行了判断，从而跳过了这些特殊情况下情况下 memmove 带来的开销。

后来，Keith Randall 终于实现了 [Randall 2013] 很早之前 Dmitry Vyukov 就已经提出的在栈上分配 defer 的优化 [Cox 2011]，简单情况下不再需要使用运行时对延迟记录的内存管理。为 Go 1.13 进一步提升了 defer 的性能。

在 Go 1.14 中，Dan Scales 作为 Go 团队的新成员，defer 的优化成为了他的第一个项目。他提出开放式编码 defer [Scales 2019]，通过编译器辅助信息和延迟比特在函数末尾处直接获取调用函数及参数，完成了近乎零成本的 defer 调用，成为了 Go 1.14 中几个出色的运行时性能优化之一。

至此，defer 的优化之路正式告一段落。

小结

我们最后来总结一下 defer 的基本工作原理以及三种 defer 的性能取舍，如下图：

不同类型 defer 的编译与运行时成本之间的取舍

1. 对于开放编码式 defer 而言：

• 编译器会直接将所需的参数进行存储，并在返回语句的末尾插入被延迟的调用；

• 当整个调用中逻辑上会执行的 defer 不超过 15 个（例如 7 个 defer 作用在 2 个返回语句）、总 defer 数量不超过 8 个、且没有出现在循环语句中时，会激活使用此类 defer；

• 此类 defer 的唯一的运行时成本就是存储参与延迟调用的相关信息，运行时性能最好。

对于栈上分配的 defer 而言：

• 编译器会直接在栈上记录一个 _defer 记录，该记录不涉及内存分配，并将其作为参数，传入被翻译为 deferprocStack 的延迟语句，在延迟调用的位置将 _defer 压入 Goroutine 对应的延迟调用链表中；

• 在函数末尾处，通过编译器的配合，在调用被 defer 的函数前，调用 deferreturn，将被延迟的调用出栈并执行；

• 此类 defer 的唯一运行时成本是从 _defer 记录中将参数复制出，以及从延迟调用记录链表出栈的成本，运行时性能其次。

对于堆上分配的 defer 而言：

• 编译器首先会将延迟语句翻译为一个 deferproc 调用，进而从运行时分配一个用于记录被延迟调用的 _defer 记录，并将被延迟的调用的入口地址及其参数复制保存，入栈到 Goroutine 对应的延迟调用链表中；

• 在函数末尾处，通过编译器的配合，在调用被 defer 的函数前，调用 deferreturn，从而将 _defer 实例归还到资源池，而后通过模拟尾递归的方式来对需要 defer 的函数进行调用。

• 此类 defer 的主要性能问题存在于每个 defer 语句产生记录时的内存分配，记录参数和完成调用时的参数移动时的系统调用，运行时性能最差。

进一步阅读的参考文献

• [Griesemer 2009] Robert Griesemer. defer statement. Jan 27 2009. https://github.com/golang/go/commit/4a903e0b32be5a590880ceb7379e68790602c29d

• [Thompson 2009] Ken Thompson. defer. Jan 27 2009. https://github.com/golang/go/commit/1e1cc4eb570aa6fec645ff4faf13431847b99db8

• [Cox 2011] Russ Cox. runtime: aggregate defer. Oct 2011. https://github.com/golang/go/issues/2364

• [Clements 2016] Austin Clements. runtime: optimize defer code. Sep 2016. https://github.com/golang/go/commit/4c308188cc05d6c26f2a2eb30631f9a368aaa737

• [Ma 2016] Minux Ma. runtime: defer is slow. Mar 2016. https://github.com/golang/go/issues/14939

• [Randall 2013] Keith Randall. cmd/compile: allocate some defers in stack frames. Dec 2013. https://github.com/golang/go/issues/6980

• [Vyukov 2014] Dmitry Vyukov. runtime: per-P defer pool. Jan 2014. https://github.com/golang/go/commit/1ba04c171a3c3a1ea0e5157e8340b606ec9d8949

• [Scales 2019] Dan Scales Keith Randall and Austin Clements. Proposal: Low-cost defers through inline code and extra funcdata to manage the panic case. Sep 2019. https://go.googlesource.com/proposal/ /refs/heads/master/design/34481-opencoded-defers.md