Go语言中如何使用接口实现泛型排序字符串键的Map

import (
    "log"
    "reflect"
    "sort"
)

// SortedKeysReflect 函数使用反射机制从键为字符串的Map中提取并排序键。
// 它需要针对Map的值类型进行显式的类型断言。
func SortedKeysReflect(mapWithStringKey interface{}) []string {
    keys := []string{}
    typ := reflect.TypeOf(mapWithStringKey)

    if typ.Kind() == reflect.Map && typ.Key().Kind() == reflect.String {
        // 根据Map的值类型进行类型断言，并提取键
        switch typ.Elem().Kind() {
        case reflect.Int:
            for key := range mapWithStringKey.(map[string]int) {
                keys = append(keys, key)
            }
        case reflect.String:
            for key := range mapWithStringKey.(map[string]string) {
                keys = append(keys, key)
            }
        // ... 根据需要添加更多case以支持其他值类型
        default:
            log.Fatalf("错误：SortedKeysReflect() 不支持类型 %s\n", typ)
        }
        sort.Strings(keys) // 对收集到的键进行排序
    } else {
        log.Fatalln("错误：SortedKeysReflect() 的参数不是 map[string]...")
    }
    return keys
}

// SortableKeysValue 定义了一个接口，任何实现此接口的类型都必须能够提供其字符串键的切片。
type SortableKeysValue interface {
    Keys() []string
}

import "sort"

// SortedKeys 接收一个 SortableKeysValue 接口的实现，提取其键并返回一个排序后的字符串切片。
func SortedKeys(s SortableKeysValue) []string {
    keys := s.Keys()
    sort.Strings(keys) // 对键进行排序
    return keys
}

// MyMap 是一个示例Map类型，键为string，值为string。
type MyMap map[string]string

// Keys 为 MyMap 类型实现 SortableKeysValue 接口的 Keys() 方法。
// 它遍历Map，收集所有键并返回一个字符串切片。
func (m MyMap) Keys() []string {
    keys := make([]string, 0, len(m)) // 预分配容量，优化性能
    for k := range m {
        keys = append(keys, k)
    }
    return keys
}

// IntMap 是另一个示例Map类型，键为string，值为int。
type IntMap map[string]int

// Keys 为 IntMap 类型实现 SortableKeysValue 接口的 Keys() 方法。
func (m IntMap) Keys() []string {
    keys := make([]string, 0, len(m))
    for k := range m {
        keys = append(keys, k)
    }
    return keys
}

package main

import (
    "fmt"
    "sort"
)

// SortableKeysValue 接口定义
type SortableKeysValue interface {
    Keys() []string
}

// SortedKeys 泛型函数
func SortedKeys(s SortableKeysValue) []string {
    keys := s.Keys()
    sort.Strings(keys)
    return keys
}

// MyMap 类型及其接口实现
type MyMap map[string]string

func (m MyMap) Keys() []string {
    keys := make([]string, 0, len(m))
    for k := range m {
        keys = append(keys, k)
    }
    return keys
}

// IntMap 类型及其接口实现
type IntMap map[string]int

func (m IntMap) Keys() []string {
    keys := make([]string, 0, len(m))
    for k := range m {
        keys = append(keys, k)
    }
    return keys
}

func main() {
    // 使用 MyMap
    myStringMap := MyMap{
        "apple":  "red",
        "banana": "yellow",
        "cherry": "red",
    }
    sortedStringKeys := SortedKeys(myStringMap)
    fmt.Println("Sorted string keys (MyMap):", sortedStringKeys) // 输出: [apple banana cherry]

    // 使用 IntMap
    myIntMap := IntMap{
        "z": 3,
        "a": 1,
        "b": 2,
    }
    sortedIntKeys := SortedKeys(myIntMap)
    fmt.Println("Sorted string keys (IntMap):", sortedIntKeys) // 输出: [a b z]
}