Go for range 新手常会犯的错误写法

拼搏现实的明天。 2023-10-09 23:54 27阅读 0赞

#### 阅读目录 ####

*  切片示例
 *  结构体示例

## 切片示例 ##

package main
    
    import (
    	"fmt"
    )
    
    func main() {
        
    	slice := []int{
        0, 1, 2, 3}
    	m := make(map[int]*int)
    	for key, val := range slice {
        
    		m[key] = &val
    	}
    	for k, v := range m {
        
    		fmt.Println(k, "->", *v)
    	}
    }

root@debiancc:~/www/test# go run test.go 
    0 -> 3
    1 -> 3
    2 -> 3
    3 -> 3
    root@debiancc:~/www/test#

解析：

这是新手常会犯的错误写法，for range 循环的时候会创建每个元素的副本，而不是元素的引用，所以 `m[key] = &val` 取的都是变量 `val` 的地址，所以最后 map 中的所有元素的值都是变量 val 的地址，因为最后 val 被赋值为 3 所有输出都是 3。

正确的写法：

package main
    
    import (
    	"fmt"
    )
    
    func main() {
        
    	slice := []int{
        0, 1, 2, 3}
    	m := make(map[int]*int)
    	for key, val := range slice {
        
    		value := val
    		m[key] = &value
    	}
    	for k, v := range m {
        
    		fmt.Println(k, "===>", *v)
    	}
    }

root@debiancc:~/www/test# go run test.go 
    2 ===> 2
    3 ===> 3
    0 ===> 0
    1 ===> 1
    root@debiancc:~/www/test#

## 结构体示例 ##

package main
    
    import (
    	"fmt"
    )
    
    type Test struct {
        
    	name string
    }
    
    func (this *Test) Point() {
        
    	fmt.Println(this.name)
    }
    func main() {
        
    	ts := []Test{
        
    		{
        "a"},
    		{
        "b"},
    		{
        "c"},
    	}
    	for _, t := range ts {
        
    		//fmt.Println(reflect.TypeOf(t))
    		defer t.Point()
    	}
    }

root@debiancc:~/www/test# go run test.go 
    c
    c
    c
    root@debiancc:~/www/test#

defer 语句中使用的 t 是循环变量的副本，而不是原始切片中的元素。

为了达到预期的输出，你可以通过将循环变量 t 的地址传递给 defer 语句来解决这个问题。  
下面是修正后的代码：

package main
    
    import (
    	"fmt"
    )
    
    type Test struct {
        
    	name string
    }
    
    func (t *Test) Point() {
        
    	fmt.Println(t.name)
    }
    
    func main() {
        
    	ts := []*Test{
        
    		{
        "a"},
    		{
        "b"},
    		{
        "c"},
    	}
    
    	for _, t := range ts {
        
    		defer t.Point()
    	}
    }

root@debiancc:~/www/test# go run test.go 
    c
    b
    a
    root@debiancc:~/www/test#