Go 函数 Map 型参数扩容时引用改变之辩解

后端

2024-01-29 18:06:45

Go 语言中 Map 作为函数参数的注意事项

在 Go 语言中，Map 是一种复杂类型，由键值对组成。当我们把 Map 作为函数参数传递时，需要考虑一些特殊情况，以确保函数能够正确地处理 Map。

Map 的底层实现

Map 是通过哈希表实现的。哈希表是一种数据结构，它根据键的哈希值来快速查找键值对。当我们向 Map 中插入一个键值对时，哈希表会计算出该键值对的哈希值，然后将该键值对存储在哈希表中计算出的位置。

函数的参数传递机制

当我们把 Map 作为函数参数传递时，函数会收到 Map 的一个副本。这意味着函数只能访问 Map 的副本，而不能访问 Map 的底层数据结构。因此，如果函数对 Map 进行修改，这些修改不会影响到 Map 的底层数据结构。

例外情况：Map 扩容

但是，有一种例外情况，那就是当 Map 的容量不够时，Map 会自动扩容。当 Map 扩容时，Map 的底层数据结构会发生变化。如果函数在 Map 扩容后继续访问 Map，那么函数可能会访问到错误的数据。

使用指针传递 Map

为了避免这种情况，我们可以使用指针来传递 Map。当我们使用指针来传递 Map 时，函数会收到 Map 的指针，而不是 Map 的副本。这意味着函数可以访问 Map 的底层数据结构。如果函数对 Map 进行修改，这些修改会影响到 Map 的底层数据结构。

何时使用指针传递 Map

如果我们需要在函数中修改 Map，那么我们应该使用指针来传递 Map。如果我们只需要访问 Map，那么我们可以使用 Map 的副本。

代码示例

func main() {
    m := make(map[int]int)
    m[1] = 10
    m[2] = 20

    // 使用 Map 的副本
    fmt.Println(copyMap(m)) // 输出：map[1:10 2:20]

    // 使用 Map 的指针
    changeMap(&m)
    fmt.Println(m) // 输出：map[1:10 2:20 3:30]
}

func copyMap(m map[int]int) map[int]int {
    // 返回 Map 的副本
    return m
}

func changeMap(m *map[int]int) {
    // 使用 Map 的指针
    (*m)[3] = 30
}