【Go学习笔记】return语句是原子操作吗?

350 阅读1分钟

Golang 中的 return 语句不是原子操作,而是分为两步执行:

  1. 返回值赋值
  2. RET

defer 语句的执行是在 RET 指令执行之前,因此含有 defer 语句时 return 的执行步骤如下:

  1. 返回值赋值
  2. defer
  3. RET

以两段代码为例说明:

函数返回值为匿名返回值

func test() int {
   var z int
   defer func() {
      fmt.Println("defer :", z)
      z += 100
   }()
   return z
}
func main() {
   fmt.Println("test :", test())
}

输出结果如下:

defer : 0
test : 0

在 defer 语句执行之前,将 z 的值赋给一个临时变量,并将这个临时变量作为函数返回值,因此 defer 语句对 z 的操作不影响函数返回值。

函数返回值为命名返回值

func test() (z int) {
   defer func() {
      fmt.Println("defer :", z)
      z += 100
   }()
   return 100
}
func main() {
   fmt.Println("test :", test())
}

输出结果如下:

defer : 100
test : 200

函数返回值一开始就定义为变量 z。在 defer 语句执行之前,为 z 赋值 100;执行 defer 后,z 的值更新为 200,并作为函数返回值进行返回,因此返回值可以体现 defer 操作产生的变化。