Skip to main content

高效使用Golang中的 Map：关键考虑因素

· 3 min read
wen

在 Go (Golang) 编程中，map 是一种强大且灵活的数据结构，用于存储键值对。

然而，为了确保高效和正确的使用，有几个重要的点需要注意。

本文将详细介绍这些关键考虑因素，并通过示例代码展示如何在 Go 中有效地使用 map。

1. 初始化​

在使用 map 之前，必须对其进行初始化。可以使用 make 函数或使用 map 字面量来初始化。

// 使用 make
m := make(map[string]int)

// 使用 map 字面量
m := map[string]int{"🍎": 1, "🍌": 2}

2. Nil Map​

nil map 与 空 map 不同。

nil map 不能写入，尝试写入会导致运行时恐慌(panic)。

因此，始终要初始化你的 map。

var m map[string]int // m 是 nil
m["🍎"] = 1          // 这会导致恐慌

3. 从 Map 中读取​

从 map 中读取时，如果键不存在，会返回值类型的零值。为了区分缺失的键和实际的零值, 可以使用多返回值的方式。

value, ok := m["🍊"]
if ok {
    fmt.Println("找到了:", value)
} else {
    fmt.Println("未找到")
}

4. 从 Map 中删除​

使用 delete 函数从 map 中移除键值对。在键不存在的情况下调用 delete 是安全的。

delete(m, "🍎")

5. 并发访问​

Map 不是并发安全的。如果多个 goroutine 同时访问一个 map，并且至少有一个修改了 map，你必须使用互斥锁或通道来同步访问。

var mu sync.Mutex

mu.Lock()
m["🍎"] = 1
mu.Unlock()

6. 迭代顺序​

map 的迭代顺序不能保证在程序的不同运行中保持一致。如果你需要稳定的迭代顺序，必须显式地对键进行排序。

keys := make([]string, 0, len(m))
for k := range m {
    keys = append(keys, k)
}
sort.Strings(keys)
for _, k := range keys {
    fmt.Println(k, m[k])
}

7. 作为 set 使用​

Golang 中没有 set 类型，但可以使用 map 来模拟 set 。只需将值类型设置为 struct{} 即可。

struct{} 是一个空结构，不占用任何内存空间。这样可以节省内存，因为 map 的值是空结构，而不是实际的值。

s := make(map[string]struct{})
s["🍎"] = struct{}{}

// 检查键是否存在
_, ok := s["🍎"]

如果觉得麻烦，可以直接使用第三方库 golang-set。

Loading Comments...