2

我写了一个函数，它会从 map[string]Foo 返回一个排序的字符串切片。我很好奇创建一个通用例程的最佳方法是什么，该例程可以从任何类型的以字符串为键的映射返回一个排序的字符串切片。

有没有办法使用接口规范来做到这一点？例如，有没有办法做类似的事情：

type MapWithStringKey interface {
    <some code here>
}

要实现上面的接口，类型需要字符串作为键。然后我可以编写一个通用函数，它返回一个排序的键列表以实现类型。

这是我目前使用反射模块的最佳解决方案：

func SortedKeys(mapWithStringKey interface{}) []string {
    keys := []string{}
    typ := reflect.TypeOf(mapWithStringKey)
    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)
            }
            // ... add more cases as needed
        default:
            log.Fatalf("Error: SortedKeys() does not handle %s\n", typ)
        }
        sort.Strings(keys)
    } else {
        log.Fatalln("Error: parameter to SortedKeys() not map[string]...")
    }
    return keys
}

点击Go Playground版本

我被迫为每种支持的类型编写类型断言，即使在编译时，我们应该知道 mapWithStringKey 参数的确切类型。

4

2 回答 2

5

您不能制作部分类型。但是您可以定义一个服务于您的目的的接口：

type SortableKeysValue interface {
    // a function that returns the strings to be sorted
    Keys() []string
}

func SortedKeys(s SortableKeysValue) []string {
    keys := s.Keys()
    sort.Strings(keys)
    return keys
}

type MyMap map[string]string

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

在这里试试：http ://play.golang.org/p/vKfri-h4Cp

于 2012-11-08T16:07:10.883 回答
0

希望有帮助（go-1.1）：

package main

import (
    "fmt"
"reflect"
)

var m = map[string]int{"a": 3, "b": 4}

func MapKeys(m interface{}) (keys []string) {
    v := reflect.ValueOf(m)
    for _, k := range v.MapKeys() {
        keys = append(keys, k.Interface().(string))
    }
    return
}

func main() {
    fmt.Printf("%#v\n", MapKeys(m))
}
于 2013-07-30T12:50:17.397 回答