Go语言中利用接口实现map[string]T键的通用提取与排序

package main

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

// SortedKeys 通过反射从map[string]T中提取并排序键
func SortedKeys(mapWithStringKey interface{}) []string {
    keys := []string{}
    typ := reflect.TypeOf(mapWithStringKey)

    // 检查是否为map类型且键类型为string
    if typ.Kind() == reflect.Map && typ.Key().Kind() == reflect.String {
        // 根据值类型进行断言和遍历
        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 reflect.Float64:
        //  for key := range mapWithStringKey.(map[string]float64) {
        //      keys = append(keys, key)
        //  }
        default:
            log.Fatalf("Error: SortedKeys() does not handle map[string]%s\n", typ.Elem().Kind())
        }
        sort.Strings(keys)
    } else {
        log.Fatalln("Error: parameter to SortedKeys() not map[string]...")
    }
    return keys
}

func main() {
    // 示例使用
    myMapInt := map[string]int{"c": 3, "a": 1, "b": 2}
    sortedIntKeys := SortedKeys(myMapInt)
    log.Printf("Sorted int keys: %v\n", sortedIntKeys)

    myMapString := map[string]string{"grape": "purple", "apple": "red", "banana": "yellow"}
    sortedStringKeys := SortedKeys(myMapString)
    log.Printf("Sorted string keys: %v\n", sortedStringKeys)

    // 尝试传入不支持的值类型，会导致运行时错误
    // myMapFloat := map[string]float64{"pi": 3.14}
    // SortedKeys(myMapFloat) // 会导致程序终止，因为float64未在switch中处理
}

package main

import (
    "fmt"
    "sort"
)

// SortableKeysValue 定义了一个接口，要求实现类型能够返回其字符串键切片
type SortableKeysValue interface {
    Keys() []string
}

// 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()方法
func (m MyMap) Keys() []string {
    keys := make([]string, 0, len(m))
    for k := range m { // 遍历map，提取键
        keys = append(keys, k)
    }
    return keys
}

// MyIntMap 是另一个具体的map[string]int类型
type MyIntMap map[string]int

// Keys 为MyIntMap类型实现了SortableKeysValue接口的Keys()方法
func (m MyIntMap) Keys() []string {
    keys := make([]string, 0, len(m))
    for k := range m { // 遍历map，提取键
        keys = append(keys, k)
    }
    return keys
}

func main() {
    // 使用MyMap类型
    myStringMap := MyMap{"grape": "purple", "apple": "red", "banana": "yellow"}
    sortedStringKeys := SortedKeys(myStringMap)
    fmt.Printf("Sorted string keys from MyMap: %v\n", sortedStringKeys)

    // 使用MyIntMap类型
    myIntegerMap := MyIntMap{"c": 3, "a": 1, "b": 2}
    sortedIntKeys := SortedKeys(myIntegerMap)
    fmt.Printf("Sorted string keys from MyIntMap: %v\n", sortedIntKeys)

    // 注意：不能直接传入原始的map[string]string或map[string]int
    // 因为它们没有直接实现SortableKeysValue接口，这会导致编译错误
    // 例如：SortedKeys(map[string]string{"x":"y"}) // 编译错误：map[string]string does not implement SortableKeysValue
}