Go：一个使用不同类型值的映射的函数

Golang不知道大家是否熟悉？今天我将给大家介绍《Go：一个使用不同类型值的映射的函数》，这篇文章主要会讲到等等知识点，如果你在看完本篇文章后，有更好的建议或者发现哪里有问题，希望大家都能积极评论指出，谢谢！希望我们能一起加油进步！

在我的代码中，我需要一个函数来从映射返回有序的键片。

m1 := make(map[string]string)
m2 := make(map[string]int)

现在我需要调用一个传递两种类型的地图的函数：

keys1 := sortedkeys(m1)
keys2 := sortedkeys(m1)

问题：我必须编写两个函数，因为该函数应该使用两种不同类型的地图。同时，两种情况下的函数体都是相同的。

问题：如何对两个地图使用单一实现？或者有其他方法可以优雅地解决问题吗？

我的第一个想法是使用 map[string]interface{} 作为参数类型，但您不能将 map[string]string 或 map[string]int 分配给它。

我的代码：

func sortedKeys(m map[string]string) []string {
    var keys []string
    for key := range m {
        keys = append(keys, key)
    }
    sort.Strings(keys)
    return keys
}

我必须重复相同的代码，但对于 map[string]int。

解决方案

您可以使用接口{}并使用反射来实现此目的。

您可以为其编写两个函数，但它不可扩展，例如，您现在支持 string 和 int，但您希望将来支持 int64、float64、bool 或 struct。使用 map[string]interface{} 并使用反射来实现通用函数是正确的方法。

建议代码：

package main

import (
    "fmt"
    "reflect"
)

func main() {
    m1 := make(map[string]string)
    m2 := make(map[string]int)

    m1["a"] = "b"
    m1["b"] = "c"

    m2["a"] = 1
    m2["b"] = 2

    fmt.Println(sortedKeys(m1))
    fmt.Println(sortedKeys(m2))
}

// Returns slice of values in the type which is sent to it
func sortedKeys(m interface{}) interface{} {

    if m == nil {
       return nil
    }

    if reflect.TypeOf(m).Kind() != reflect.Map {
        return nil
    }

    mapIter := reflect.ValueOf(m).MapRange()
    mapVal := reflect.ValueOf(m).Interface()

    typ := reflect.TypeOf(mapVal).Elem()
    outputSlice := reflect.MakeSlice(reflect.SliceOf(typ), 0, 0)


    for mapIter.Next() {
       outputSlice = reflect.Append(outputSlice, mapIter.Value())
    }


    return outputSlice.Interface()


}

输出：

[bc]

[1 2]

https://play.golang.org/p/2fkpydH9idG

今天关于《Go：一个使用不同类型值的映射的函数》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！