深入Go Map的使用技巧

原文链接:https://www.zhoubotong.site/post/60.html
之前写过一篇文章,Go map定义的几种方式以及修改技巧,今天发现还可以深入探讨下开发中容易被忽视遗漏的问题,

以下以map为例,演示大家日常开发中可能存在的问题。

Map的Value的赋值

我们来看下下面的代码编译会出现什么结果?

package main

import "fmt"

type Person struct {
    Name string
    Sex  int
}

func main() {

    u := make(map[string]Person) // 定义一个map ,值为上面Person的结构体

    person := Person{"乔峰", 0}

    u["p"] = person
    u["p"].Sex = 18 // 按道理上面u["p"] 已经赋值成了person即Person{"乔峰", 0}的结构体,使用结构体.key=value应该没毛病

    fmt.Println(u["p"])
}

输出:

./main.go:17:2: cannot assign to struct field u["p"].Sex in map

分析

为啥不能使用u["p"].属性赋值?map[string]Person 的value是一个Person结构值,所以当使用u["p"] = person,其实是一个值拷贝过程。

而u["p"]则是一个值引用(注意:所有像 int、float、bool 和 string 这些基本类型都属于值类型,使用这些类型的变量是直接指向存在内存中的值。

Golang 中引用类型如指针,slice,map,channel,接口,函数等),再说简单点就是:

[值类型]

包括:int、float、bool、string、数组、结构体

值类型变量声明后,不管是否已经赋值,编译器就会为它分配内存,此时该值储存在栈上,比如j=i 时候修改某变量i/j的值不会影响另一个。

[引用类型]

包括:指针、slice切片、map、chan、interface

变量直接存放的就是一个内存地址值,这个地址值指向的空间存的才是值。所以修改器中一个,另外一个也会修改。

但是需要注意的是引用类型必须申请内存才可以使用,make()是给引用类型申请内存空间。

既然u["p"]是值引用,那么对u["p"].Sex = 18的修改当然是不允许滴了。

那要怎么解决赋值问题?这里我整理了两种思路来解决:
解决方案一

package main

import "fmt"

type Person struct {
    Name string
    Sex  int
}

func main() {

    u := make(map[string]Person) // 定义一个map ,值为上面Person的结构体
    person := Person{"乔峰", 0}
    u["p"] = person
    //u["p"].Sex = 18 // 按道理上面u["p"] 已经赋值成了person即Person{"乔峰", 0}的结构体
    //解决思路一
    tmp := u["p"] //使用临时变量将u["p"]即person 赋值给tmp,该操作为值拷贝
    tmp.Sex = 18  //tmp 具备struct
    u["p"] = tmp  //或者person =tmp 将tmp 赋值给u["p"]即person ,该操作为值拷贝复制回去覆盖
    fmt.Println(u["p"])
}

以上输出:

{乔峰 18}

上面tmp := u["p"]是先做一次值拷贝,做出一个tmp副本,然后修改该副本,然后再次发生了一次值拷贝复制回去:u["p"] = tmp,但是这种会在整个过程中发生2次结构体值拷贝,很明显这周搞法性能很差。

自己觉的都看不下去了,有什么更好的实现方案?接下来我们看下思路二:

package main

import "fmt"

type Person struct {
    Name string
    Sex  int
}

func main() {
    u := make(map[string]*Person) // 注意这里的变化:定义一个map ,值为上面Person的结构体指针
    person := Person{"乔峰", 0}
    u["p"] = &person
    u["p"].Sex = 18
    fmt.Println(u["p"])
}

上面我们将map的类型的value由Person值,改成Person指针。

这样我们实际上每次修改的都是指针所指向的Person空间,指针本身是常指针,不能修改,只读属性,但是指向的Person是可以随便修改的,

而且这里我们看到并不需要值拷贝。只是一个指针的赋值。代码明显精简了很多。

Map的遍历赋值

我们来看下下面的代码有什么问题?能说明原因吗?

package main

import "fmt"

type Person struct {
    Name string
    Sex  int
}

func main() {
    u := make(map[string]*Person) // 注意这里的变化:定义一个map ,值为上面Person的结构体指针

    //定义Person数组
    persons := []Person{
        {Name: "乔峰", Sex: 18},
        {Name: "鸠摩智", Sex: 23},
        {Name: "慕容复", Sex: 22},
    }

    //将数组依次添加到map中
    for _, person := range persons {
        u[person.Name] = &person
    }

    //打印map
    for k, v := range u {
        fmt.Println(k, "=>", v.Name)
    }
}

遍历结果出现非预期的结果。仔细看,结果是错误的:

乔峰 => 慕容复
鸠摩智 => 慕容复
慕容复 => 慕容复

我们发现map中的3个key都指向数组中最后一个结构体。

分析

for循环中,person是结构体的一个拷贝副本,所以u

[person.Name]=&person

实际上一直都是指向了同一个指针, 最终该指针的值为遍历的最后一个struct的值拷贝

怎么改?

package main

import "fmt"

type Person struct {
    Name string
    Sex  int
}

func main() {
    u := make(map[string]*Person)

    //定义Person数组
    persons := []Person{
        {Name: "乔峰", Sex: 18},
        {Name: "鸠摩智", Sex: 23},
        {Name: "慕容复", Sex: 22},
    }

    // 遍历结构体数组,依次赋值给map
    for i := 0; i < len(persons); i++ {
        u[persons[i].Name] = &persons[i]
    }
    //打印map
    for k, v := range u {
        fmt.Println(k, "=>", v.Name)
    }
 }

输出:

鸠摩智 => 鸠摩智
慕容复 => 慕容复
乔峰 => 乔峰

通过上面的示例,希望能让大家少走弯路,避免踩坑。

posted @ 2022-07-20 13:31  周伯通之草堂  阅读(190)  评论(0编辑  收藏  举报