讓我們想象一下我有地圖：map[string]string。我想獲取此地圖的排序鍵列表。所以我可以做這樣的事情：func SortedMapKeys(m map[string]string) (keyList []string) { for key := range m { keyList = append(keyList, key) } sort.Strings(keyList) return}然后我會有另一個類型的地圖map[string]bool。我也想得到它的鑰匙。但問題是函數 SortedMapKeys 接受一個map[string]string參數。所以我需要編寫完全相同的函數，只有一個區別 - 它會接受map[string]bool.出于顯而易見的原因，這不是一種選擇。如果有一天我想改變獲取和排序密鑰的邏輯，我將需要跟蹤和更新所有這些功能。此外，我將不得不為所有這些實際上做同樣事情的函數編寫相同的單元測試，因為它們的主體是 100% 相等的（代碼重復）。有沒有辦法創建一個可以接受map[string] 任何東西的通用函數？
2 回答
素胚勾勒不出你
TA貢獻1827條經驗 獲得超9個贊
由于map[string]bool和map[string]string和map[string]Whatever都是不同的類型，創建單個函數來對所有可能map[string]*類型的鍵進行排序的唯一方法是通過反射。
func SortedMapKeys(m interface{}) (keyList []string) {
keys := reflect.ValueOf(m).MapKeys()
for _, key := range keys {
keyList = append(keyList, key.Interface().(string))
}
sort.Strings(keyList)
return
}
對于中間解決方案，由于您可能只關心幾種類型的組合，您可以使用類型開關來提取鍵
func SortedMapKeys(m interface{}) (keyList []string) {
switch m := m.(type) {
case map[string]string:
for k := range m {
keyList = append(keyList, k)
}
case map[string]bool:
for k := range m {
keyList = append(keyList, k)
}
default:
panic("unknown map type")
}
sort.Strings(keyList)
return
}
湖上湖
TA貢獻2003條經驗 獲得超2個贊
這是我的 0.02 美元。由于密鑰提取邏輯不太可能改變，并且您希望將所有內容都放在一個地方，您可以創建變體并從中選擇非零映射：
type MapVariant struct {
Bool map[string]bool
String map[string]string
}
func SortedMapKeys(variant MapVariant) (keyList []string) {
if variant.String != nil {
for k := range variant.String {
keyList = append(keyList, k)
}
goto SORT
}
if variant.Bool != nil {
for k := range variant.Bool {
keyList = append(keyList, k)
}
goto SORT
}
SORT:
sort.Strings(keyList)
return
}
當然，您可以通過添加更多條件來避免 goto 語句，但我個人認為它更清晰。
然后你可以使用如下函數：
SortedMapKeys(MapVariant{
Bool: map[string]bool{"a": true, "b": false}
})
SortedMapKeys(MapVariant{
String: map[string]string{"c": "v1", "b": "v2"}
})
- 2 回答
- 0 關注
- 189 瀏覽
添加回答
舉報
0/150
提交
取消