golang range 内全局变量

发布时间:2024-07-02 21:08:28

开发过程中，我们经常需要使用range来遍历数组、切片、映射等数据结构。然而，在使用range时，我们需要注意其中一个特殊的场景：当range内部使用全局变量时，可能会出现一些意想不到的问题。在本文中，我们将深入探讨这个问题，并提供一些解决方案。

问题的背景

首先，让我们来看一个简单的例子，通过一个for循环来遍历一个切片：


package main

import "fmt"

var results []int

func main() {
    nums := []int{1, 2, 3, 4}
    for _, num := range nums {
        results = append(results, num*num)
    }
    fmt.Println("Results:", results)
}

上面的代码会输出 [1 4 9 16]。这是符合我们预期的结果，因为在每次迭代时，结果会被添加到全局切片results中。

问题的产生

然而，如果我们将上述代码稍作修改，将全局切片换成局部变量，就会遇到问题：


package main

import "fmt"

func main() {
    nums := []int{1, 2, 3, 4}
    var results []int
    for _, num := range nums {
        results = append(results, num*num)
    }
    fmt.Println("Results:", results)
}

上述代码会输出一个空切片 []，这可能会让你感到惊讶。为什么会出现这种情况？原因在于range创建了一个隐式的循环变量副本。

尽管我们在循环中修改了results切片，但实际上修改的只是range创建的循环副本中的切片引用。由于我们使用的是局部变量，因此在每次迭代结束时，循环副本都会被销毁。因此，最终的结果是我们传入range的切片并没有被修改。

解决方案

为了解决这个问题，我们有几种方法可以选择：

方法一：使用全局变量

最简单的解决方案是改回使用全局变量。当我们在range内部使用全局变量时，循环副本将引用该全局变量，而不是局部变量。这样，在每次迭代结束时，全局变量的修改将保留下来。例如：


package main

import "fmt"

var results []int

func main() {
    nums := []int{1, 2, 3, 4}
    for _, num := range nums {
        results = append(results, num*num)
    }
    fmt.Println("Results:", results)
}

这将输出 [1 4 9 16]，和我们预期的结果一致。

方法二：使用局部变量的副本

如果你不想使用全局变量，那么另一种解决方案是在循环中创建一个局部变量的副本。这样，在每次迭代时，我们都将修改副本，并在循环结束后将其赋值给我们想要的地方。


package main

import "fmt"

func main() {
    nums := []int{1, 2, 3, 4}
    results := make([]int, len(nums))
    
    for i, num := range nums {
        localResult := num * num
        results[i] = localResult
    }
    fmt.Println("Results:", results)
}

这将输出 [1 4 9 16]，因为我们在每次迭代时修改的是局部变量的副本。

方法三：使用闭包

最后一种解决方案是使用闭包。通过将range内部的逻辑封装在一个匿名函数中，我们可以避免使用循环副本，并且可以直接访问外部变量。下面是一个示例代码：


package main

import "fmt"

func main() {
    nums := []int{1, 2, 3, 4}
    var results []int
    
    for _, num := range nums {
        func() {
            results = append(results, num*num)
        }()
    }
    fmt.Println("Results:", results)
}

这将输出 [1 4 9 16]，因为我们通过使用闭包访问了外部变量。

总的来说，当在range内部使用全局变量时，需要注意循环副本的问题。通过使用全局变量、使用局部变量的副本或使用闭包，我们可以解决这个问题，并确保符合预期的结果。