Go语言中map[string]string的有序遍历策略

package main

import (
    "fmt"
    "sort" // 引入sort包，以备不时之需，例如按字母顺序排序
)

// OrderedMap 结构体，用于存储键值对并维护键的插入顺序
type OrderedMap struct {
    data map[string]string // 实际存储数据的map
    keys []string          // 存储键的切片，按插入顺序排列
}

// NewOrderedMap 创建并返回一个新的OrderedMap实例
func NewOrderedMap() *OrderedMap {
    return &OrderedMap{
        data: make(map[string]string),
        keys: make([]string, 0),
    }
}

// Set 方法向OrderedMap中添加或更新键值对
func (om *OrderedMap) Set(key, value string) {
    // 如果键是新键，则将其添加到keys切片中
    if _, ok := om.data[key]; !ok {
        om.keys = append(om.keys, key)
    }
    // 更新或添加map中的值
    om.data[key] = value
}

// Get 方法从OrderedMap中获取指定键的值
func (om *OrderedMap) Get(key string) (string, bool) {
    val, ok := om.data[key]
    return val, ok
}

// Delete 方法从OrderedMap中删除指定键及其对应的值
func (om *OrderedMap) Delete(key string) {
    if _, ok := om.data[key]; ok {
        delete(om.data, key)
        // 从keys切片中移除该键
        for i, k := range om.keys {
            if k == key {
                om.keys = append(om.keys[:i], om.keys[i+1:]...)
                break
            }
        }
    }
}

// IterateOrdered 方法按插入顺序遍历OrderedMap
func (om *OrderedMap) IterateOrdered() {
    fmt.Println("--- 按插入顺序遍历 ---")
    for _, key := range om.keys {
        // 再次检查键是否存在，以防在Set后Delete了但未同步到keys切片（虽然Delete方法已处理）
        if val, ok := om.data[key]; ok {
            fmt.Printf("Key: %s, Value: %s\n", key, val)
        }
    }
}

// IterateAlphabetical 方法按键的字母顺序遍历OrderedMap (演示不同排序)
func (om *OrderedMap) IterateAlphabetical() {
    fmt.Println("\n--- 按字母顺序遍历 ---")
    // 复制键切片并进行排序
    sortedKeys := make([]string, 0, len(om.keys))
    for k := range om.data { // 从data中获取当前所有键，确保是最新的
        sortedKeys = append(sortedKeys, k)
    }
    sort.Strings(sortedKeys) // 对键进行字母排序

    for _, key := range sortedKeys {
        if val, ok := om.data[key]; ok {
            fmt.Printf("Key: %s, Value: %s\n", key, val)
        }
    }
}

func main() {
    // 创建一个有序map实例
    myMap := NewOrderedMap()

    // 按照特定顺序添加元素
    myMap.Set("i", "我")
    myMap.Set("we", "我们")
    myMap.Set("he", "他")
    myMap.Set("she", "她")
    myMap.Set("it", "它")
    myMap.Set("they", "他们")

    // 演示按插入顺序遍历
    myMap.IterateOrdered()

    // 演示按字母顺序遍历
    myMap.IterateAlphabetical()

    // 演示删除元素后的有序遍历
    myMap.Delete("we")
    fmt.Println("\n--- 删除 'we' 后的按插入顺序遍历 ---")
    myMap.IterateOrdered()

    // 演示更新元素
    myMap.Set("it", "它 (已更新)")
    fmt.Println("\n--- 更新 'it' 后的按插入顺序遍历 ---")
    myMap.IterateOrdered()
}