Go语言Map结构体：值还是指针选哪个？

大家好，今天本人给大家带来文章《Go语言Map结构体存储：值与指针选择全解析》，文中内容主要涉及到，如果你对Golang方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

Go语言中值类型与指针类型的基础

在Go语言中，数据类型可以分为值类型和指针类型。理解它们在Map中的行为差异是至关重要的。

• 值类型 (Value Type)：当一个值类型的数据被赋值给另一个变量时，会创建一个该数据的新副本。对新副本的修改不会影响原始数据。常见的Go值类型包括基本数据类型（如int, float, bool, string）、数组和结构体（当它们不包含指针字段时，或者即使包含，也是复制指针本身而非其指向的数据）。
• 指针类型 (Pointer Type)：指针存储的是另一个变量的内存地址。当一个指针被赋值给另一个变量时，它们都指向内存中的同一个底层数据。通过任一指针修改底层数据，会影响所有指向该数据的指针。

Map中存储结构体的行为差异

当我们将结构体作为Map的值类型时，选择map[int]vertex或map[int]*vertex会带来截然不同的行为。

1. map[int]vertex：存储结构体副本

当Map的值类型是结构体本身时（例如map[int]vertex），Map中存储的是结构体的副本。这意味着：

• 赋值行为：当您将一个结构体赋值给Map中的某个键时，Map会复制该结构体的所有字段，并将这个副本存储起来。原始结构体与Map中存储的副本是完全独立的。
• 修改行为：您不能直接修改Map中已存储的结构体成员。例如，a[0].x = 3这样的操作会导致编译错误。这是因为Go语言的Map元素是不可寻址的（unaddressable）。Map的底层实现可能会在内存中移动元素，如果允许直接修改，可能会导致悬空指针或数据不一致。 如果需要修改map[int]vertex中的结构体，您必须先取出该结构体的副本，修改副本，然后再将修改后的副本重新赋值回Map。

2. map[int]*vertex：存储结构体指针

当Map的值类型是结构体指针时（例如map[int]*vertex），Map中存储的是结构体的内存地址。这意味着：

• 赋值行为：当您将一个结构体指针赋值给Map中的某个键时，Map会存储这个指针本身。Map中的元素和原始结构体变量都指向内存中的同一个底层结构体。
• 修改行为：您可以直接通过Map中存储的指针来修改底层结构体的成员。例如，b[0].x = 3是完全合法的，它会修改b[0]所指向的vertex结构体的x字段。

代码示例与解析

为了更清晰地展示这两种方式的区别，我们来看一个具体的Go语言程序：

package main

import "fmt"

type vertex struct {
    x, y int
}

func main() {
    // 声明两个Map：a 存储 vertex 结构体值，b 存储 *vertex 结构体指针
    a := make(map[int]vertex)
    b := make(map[int]*vertex)

    // 创建一个 vertex 结构体实例，并获取其指针
    v := &vertex{0, 0} 

    // 将 v 指向的结构体的值（副本）存入 a[0]
    a[0] = *v 
    // 将 v 这个指针本身存入 b[0]
    b[0] = v 

    // 第一次修改：通过原始指针 v 修改其指向的结构体
    // 此时，v 和 b[0] 都指向同一个内存地址，a[0] 是 v 的一个独立副本
    v.x, v.y = 4, 4
    fmt.Printf("修改原始v后: a[0]: {%d, %d}, b[0]: {%d, %d}\n", a[0].x, a[0].y, b[0].x, b[0].y)

    // 尝试直接修改 map[int]vertex 中的元素 - 编译错误
    // a[0].x = 3 // 编译错误: cannot assign to (a[0]).x
    // a[0].y = 3 // 编译错误: cannot assign to (a[0]).y

    // 正确修改 map[int]vertex 中元素的方式：取出、修改、重新赋值
    tempA := a[0] // 取出 a[0] 的副本
    tempA.x = 3
    tempA.y = 3
    a[0] = tempA  // 将修改后的副本重新赋值回 a[0]
    fmt.Printf("通过取出-修改-赋值修改a[0]后: a[0]: {%d, %d}, b[0]: {%d, %d}\n", a[0].x, a[0].y, b[0].x, b[0].y)

    // 修改 map[int]*vertex 中的元素 - 可以直接修改
    // b[0] 是一个指针，直接修改其指向的结构体成员
    b[0].x = 3 
    b[0].y = 3 
    fmt.Printf("直接修改b[0]指向的结构体后: a[0]: {%d, %d}, b[0]: {%d, %d}\n", a[0].x, a[0].y, b[0].x, b[0].y)

    // 观察从Map中取出元素后的修改行为
    u1 := a[0] // u1 是 a[0] 的一个值副本
    u1.x = 2 
    u1.y = 2 
    // u1 的修改不影响 a[0]，因为 u1 是 a[0] 的一个独立副本
    fmt.Printf("u1 (a[0]的副本) 修改后: a[0]: {%d, %d}, b[0]: {%d, %d}\n", a[0].x, a[0].y, b[0].x, b[0].y)

    u2 := b[0] // u2 是 b[0] 的一个指针副本，它们都指向同一个底层结构体
    u2.x = 2 
    u2.y = 2 
    // u2 的修改通过指针影响了 b[0] 指向的底层结构体
    fmt.Printf("u2 (b[0]的指针副本) 修改后: a[0]: {%d, %d}, b[0]: {%d, %d}\n", a[0].x, a[0].y, b[0].x, b[0].y)
}

输出结果：

修改原始v后: a[0]: {0, 0}, b[0]: {4, 4}
通过取出-修改-赋值修改a[0]后: a[0]: {3, 3}, b[0]: {4, 4}
直接修改b[0]指向的结构体后: a[0]: {3, 3}, b[0]: {3, 3}
u1 (a[0]的副本) 修改后: a[0]: {3, 3}, b[0]: {3, 3}
u2 (b[0]的指针副本) 修改后: a[0]: {3, 3}, b[0]: {2, 2}

解析：

1. v.x, v.y = 4, 4 后：

   • a[0] 仍然是 {0, 0}，因为 a[0] = *v 存储的是 v 在 {0, 0} 时的一个副本，与 v 后来的变化无关。
   • b[0] 变为 {4, 4}，因为 b[0] = v 存储的是 v 的指针，v 所指向的内存地址中的值被修改为 {4, 4}，所以 b[0] 也反映了这一变化。

2. a[0].x = 3 编译错误的原因： 如前所述，Go语言的Map元素是不可寻址的。这意味着您不能获取Map中元素的内存地址，因此也无法直接对其进行成员修改。如果您尝试这样做，编译器会报错。正确的做法是先将元素取出（这会得到一个副本），修改副本，然后将修改后的副本重新赋值回Map。

3. b[0].x = 3 可以的原因：b[0] 存储的是一个指针。通过指针访问结构体成员是合法的，因为您实际上是在操作指针所指向的内存区域，而不是Map本身存储的那个指针值。

4. u1 := a[0] 与 u2 := b[0] 的区别：

   • u1 := a[0] 得到了 a[0] 存储的 vertex 结构体的一个完整副本。对 u1 的修改只影响 u1 这个副本，不影响 a[0] 中存储的原始副本。
   • u2 := b[0] 得到了 b[0] 存储的 *vertex 指针的一个副本。这意味着 u2 和 b[0] 现在都指向内存中的同一个 vertex 结构体。因此，通过 u2 修改结构体成员，也会影响到 b[0] 所引用的结构体。

选择建议与注意事项

理解了值类型和指针类型在Map中的行为后，我们可以根据实际需求做出选择：

何时使用 map[int]struct (值类型)

• 结构体较小：如果结构体包含的字段不多，复制开销很小，那么使用值类型可以简化内存管理。
• 数据隔离：当您希望Map中的每个元素都是独立的副本，对它们的修改不会影响到Map外部的原始结构体，或者不希望通过Map元素来修改共享数据时，值类型是更好的选择。
• 无需频繁原地修改：如果Map中的结构体一旦存入就不再需要修改其内部成员，或者修改频率很低，且可以接受“取出-修改-重新赋值”的模式，那么值类型是合适的。

何时使用 map[int]*struct (指针类型)

• 结构体较大：如果结构体包含大量字段或占用内存较大，使用指针可以避免不必要的深拷贝，从而提高性能和减少内存消耗。
• 需要频繁原地修改：当您需要频繁地修改Map中已存在的结构体实例的成员时，使用指针类型可以允许直接修改，而无需取出和重新赋值的步骤，代码更简洁高效。
• 共享数据：当多个部分需要引用和操作同一个结构体实例时，使用指针可以实现数据共享，保持数据的一致性。
• 实现接口：某些情况下，如果结构体需要实现某个接口，并且接口方法需要修改结构体自身，那么Map中存储指针会更方便。

注意事项

• *空指针问题 (`map[int]struct)**：当使用指针类型时，需要注意空指针（nil）的可能性。如果您从Map中取出一个指针，但该键不存在，或者存储的值是nil`，尝试访问其成员会导致运行时错误（panic）。

  var p *vertex
  p = b[100] // 如果 b[100] 不存在，p 将是 nil
  if p != nil {
      fmt.Println(p.x)
  } else {
      fmt.Println("键不存在或值为nil")
  }

• 并发安全：Go语言的Map本身不是并发安全的。无论您存储的是值类型还是指针类型，在多个goroutine同时读写Map时，都需要通过互斥锁（sync.Mutex）或其他并发原语来保护Map的访问，以避免数据竞争。即使您存储的是指针，对指针指向的数据的修改也可能需要同步。

总结

选择在Go语言Map中存储结构体的值类型还是指针类型，取决于您的具体应用场景、性能需求以及对数据修改行为的期望。

• map[int]struct 提供了数据隔离，每次操作都是对副本进行，但不能直接修改Map中的元素。适用于结构体较小、数据不需共享或不频繁修改的场景。
• *`map[int]struct`** 允许直接修改Map中元素所指向的底层结构体，避免了大量数据拷贝，实现了数据共享。适用于结构体较大、需要频繁原地修改或多处共享同一实例的场景。

理解这两种方式的底层机制和行为差异，是编写高效、健壮Go程序的关键。

理论要掌握，实操不能落！以上关于《Go语言Map结构体：值还是指针选哪个？》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！