迹忆客 专注技术分享

引言

Go的函数返回可以说是别出心裁，既可以用匿名返回值，也可以用命名返回值，虽然这两种方式都可以完成相同的功能，但是这里边还是有一些坑，尤其是在搭配defer的时候，返回的值可能不是你想的那样，很容易产生不太好排查的问题

举个例子

func f1() int {
    k := 0
    for i := 0; i < 3; i++ {
        defer func() {k++}()
    }
    return k
}

func f2() (k int) {
    for i := 0; i < 3; i++ {
        defer func() {k++}()
    }
    return
}

func f3() (k int) {
    s := 5
    for i := 0; i < 3; i++ {
        defer func() {k++}()
    }
    return s
}

func f4() (k int) {
    for i := 0; i < 3; i++ {
        defer func(k int) {k++}(k)
    }
    return
}

思考一下，这四个函数，分别返回的值都是多少。分别是0，3，8，0。看起来差不多的函数，返回来的值却大不一样。

解释

关键点

• return语句 这个return v 语句，这个语句编译后分成多步进行，
  • 赋值
    • 如果返回值是匿名值的话，做返回值=v，也就是找一个临时变量保存住这个返回值
    • 如果不是匿名返回值
      • return后边有值v：那还是做返回值=v，返回值就是函数定义的那个命名返回值
      • return后边没有值: 默认就是返回这个命名返回值
  • 如果有defer，后进先出的方式执行defer
  • 返回返回值
• defer的值拷贝和值引用
  • 如果没有定义传参，那就是用defer外的同名变量，而且是采用值引用的方式
  • 如果定义了传参，那就是值拷贝，不影响defer外边的变量

函数解释

• f1 由于是匿名返回值，需要先试用临时变量(tmp)保存返回值，tmp=k，不管后边对k做怎样的操作，那都是k的变化，影响不到tmp，然后最后返回tmp，也就是0

• f2 返回值为命名返回值，而且return后边没有显式返回数据，默认就是返回k，不需要提前赋值，后续的defer中没有采用值传递的方式传参，是地址引用，因此，始终在操作这个k，最后返回k时，肯定是最新的k，也就是3

• f3 返回值也是命名返回值，return却显式的返回了s，所以需要有赋值动作，也就是k=s（为什么不是tmp，因为不需要临时值，返回值已经命名为k了），后边的defer还是地址引用传参，所有的操作对k均产生了影响，返回是当然是k的最新值，也就是8

• f4 虽然也是命名返回值，return后边也没有显式的返回其他值，默认就是返回k，但是注意看defer却是用的值传递，应为定义了参数，也就是defer在入栈的时候，已经将k复制成了另外的变量（比如tmp1，tmp2，tmp3）分别跟着入了栈，defer执行时，显然不会影响真正的返回值k，毕竟地址都不是一个了。所以最后的返回值是0。

想想如果在k++后边加个fmt.Println(“—k–”,k)，会是多少？

func f4() (k int) {
    for i := 0; i < 3; i++ {
        defer func(k int) {
      k++
      fmt.Println("---k--", &k, k)
    }(k)
    }
    return
}

结果如下所示

---k-- 0xc000124048 1
---k-- 0xc000124050 1
---k-- 0xc000124058 1

答案：输出三次，都是1，而且地址都不一样。原因很简单，上边说的defer会有三次入栈，且每次入栈的时候分别新创建了三个内部变量（所以地址不一样），并都赋值成了0，也就是看着是k，实际早已经不是一个k了，所以每次给0++ ，那只能是输出0了。