别再顺序读取 Go 语言 map 了，快来看看更优解！

2023-08-19 10:22:16

轻松掌握 Go 语言中 map 的顺序读取

在 Go 语言中，map 是一种存储键值对的强大数据结构。顺序读取 map 对于许多应用程序来说至关重要，例如数据处理、配置管理和缓存。本文将深入探讨各种技术，帮助您从 map 中高效地提取数据。

range 循环：简单直接

遍历 map 的最简单方法是使用 range 循环。它将按键的顺序依次迭代每个键值对：

package main

import "fmt"

func main() {
    m := map[string]int{"a": 1, "b": 2, "c": 3}

    for key, value := range m {
        fmt.Println(key, value)
    }
}

map.Keys() 和 map.Values()：专注于键或值

如果您只关心 map 的键或值，可以使用 map.Keys() 和 map.Values() 方法。它们返回包含所有键或值的切片：

package main

import "fmt"

func main() {
    m := map[string]int{"a": 1, "b": 2, "c": 3}

    keys := map.keys(m)
    for _, key := range keys {
        fmt.Println(key)
    }

    values := map.values(m)
    for _, value := range values {
        fmt.Println(value)
    }
}

for-each 循环：Go 1.18 及更高版本

Go 1.18 引入了 for-each 循环，为顺序读取 map 提供了更简洁的语法：

package main

import "fmt"

func main() {
    m := map[string]int{"a": 1, "b": 2, "c": 3}

    m.ForEach(func(key string, value int) {
        fmt.Println(key, value)
    })
}

优化顺序读取

对于需要频繁顺序读取 map 的场景，以下优化技巧可以显着提高性能：

使用 sync.Map： sync.Map 是一个并发安全的 map，适合在多 goroutine 中并发读取。
使用有序 map： 有序 map 维护键的顺序，因此顺序读取更加高效。

示例代码：

package main

import (
    "fmt"
    "sort"
    "sync"
)

// 有序 map
type OrderedMap struct {
    keys   []string
    values []int
}

func (m *OrderedMap) Set(key string, value int) {
    i := sort.SearchStrings(m.keys, key)
    m.keys = append(m.keys, "")
    copy(m.keys[i+1:], m.keys[i:])
    m.keys[i] = key

    m.values = append(m.values, 0)
    copy(m.values[i+1:], m.values[i:])
    m.values[i] = value
}

func (m *OrderedMap) Get(key string) (int, bool) {
    i := sort.SearchStrings(m.keys, key)
    if i < len(m.keys) && m.keys[i] == key {
        return m.values[i], true
    }
    return 0, false
}

func main() {
    // 并发安全 map
    var m sync.Map
    m.Store("a", 1)
    m.Store("b", 2)
    m.Store("c", 3)

    m.Range(func(key, value interface{}) bool {
        fmt.Println(key, value)
        return true
    })

    // 有序 map
    var omap OrderedMap
    omap.Set("a", 1)
    omap.Set("b", 2)
    omap.Set("c", 3)

    omap.Range(func(key string, value int) bool {
        fmt.Println(key, value)
        return true
    })
}