乐趣区

关于java:Go语言中for-range的坑

前言

Go 中的 for range 组合能够和不便的实现对一个数组或切片进行遍历,然而在某些状况下应用 for range 时很可能就会被"坑",上面用一段代码来模仿下:

func main() {arr1 := []int{1, 2, 3}
    arr2 := make([]*int, len(arr1))

    for i, v := range arr1 {arr2[i] = &v
    }

    for _, v := range arr2 {fmt.Println(*v)
    }
}

代码解析:

  1. 创立一个 int slice,变量名为arr1 并初始化 1,2,3 作为切片的值。
  2. 创立一个*int slice,变量名为arr2
  3. 通过 for range 遍历 arr1,而后获取每一个元素的指针,赋值到对应arr2 中。
  4. 逐行打印 arr2 中每个元素的值。

从代码上看,打印进去的后果应该是

1
2
3

然而真正的后果是

3
3
3

起因

因为 for range 在遍历 值类型 时,其中的 v 变量是一个 的拷贝,当应用 & 获取指针时,实际上是获取到 v 这个长期变量的指针,而 v 变量在 for range 中只会创立一次,之后循环中会被始终重复使用,所以在 arr2 赋值的时候其实都是 v 变量的指针,而 &v 最终会指向 arr1 最初一个元素的值拷贝。

来看看上面这个代码,用 for i 来模仿for range,这样更易于了解:

func main() {arr1 := []int{1, 2, 3}
    arr2 := make([]*int, len(arr1))

    var v int
    for i:=0;i<len(arr1);i++ {v = arr1[i]
        arr2[i] = &v
    }

    for _, v := range arr2 {fmt.Println(*v)
    }
}

解决方案

  1. 传递原始指针
func main() {arr1 := []int{1, 2, 3}
    arr2 := make([]*int, len(arr1))

    for i := range arr1 {arr2[i] = &arr1[i]
    }

    for _, v := range arr2 {fmt.Println(*v)
    }
}
  1. 应用长期变量
func main() {arr1 := []int{1, 2, 3}
    arr2 := make([]*int, len(arr1))

    for i, v := range arr1 {
        t := v
        arr2[i] = &t
    }

    for _, v := range arr2 {fmt.Println(*v)
    }
}
  1. 应用闭包
func main() {arr1 := []int{1, 2, 3}
    arr2 := make([]*int, len(arr1))

    for i, v := range arr1 {func(v int){arr2[i] = &v
        }(v)
    }

    for _, v := range arr2 {fmt.Println(*v)
    }
}

官网提醒

因为这一问题过于广泛,Golang 甚至将其写入了文档的『常见谬误』局部:文档

我是 MonkeyWie,欢送扫码???????? 关注!不定期在公众号中分享JAVAGolang 前端 dockerk8s 等干货常识。

退出移动版