我有一個下面的方法，我創建了三個不同的結構對象來檢查我的name地圖中的鍵。如果我能找到該值，我會返回，否則我會返回空字符串和狀態。func (r *clientRepo) GetName(id int64, name map[models.LocaleInfo]string, clientId int32, code models.Locale) (string, bool) { localeInfo := models.LocaleInfo{ Locale: code, GroupID: int(clientId), } localeInfoUS := models.LocaleInfo{ Locale: "US", GroupID: int(clientId), } localeInfoGB := models.LocaleInfo{ Locale: "GB", GroupID: int(clientId), } value := "" ok := false if value, ok = name[localeInfo]; !ok { if value, ok = name[localeInfoUS]; !ok { if value, ok := name[localeInfoGB]; !ok { errs.Newf("Unable to find locale for ProductId: %d", id) return value, ok } } } return value, ok}有什么方法可以改進上面的代碼，我不必為每個美國和英國區域設置信息保留單獨的行。我的 if 塊看起來也很嵌套，我認為這也可以改進很多。
1 回答
翻閱古今
TA貢獻1780條經驗 獲得超5個贊
特殊的“comma-ok”形式只能在作業中使用，所以你不能讓它更緊湊。
你可以做的是編寫一個實用函數，它使用 for 循環來檢查任意數量的鍵。
使用 Go 1.18 泛型，您可以使其適用于任何地圖類型：
func findKeys[K comparable, V any](m map[K]V, keys ...K) (v V, ok bool) {
for _, key := range keys {
if v, ok = m[key]; ok {
return
}
}
return
}
測試它：
m := map[int]string{
1: "one",
2: "two",
3: "three",
4: "four",
}
fmt.Println(findKeys(m, 5, 6, 4))
fmt.Println(findKeys(m, 1, 2, 6))
fmt.Println(findKeys(m, 6, 8))
哪些輸出（在Go Playground上嘗試）：
four true
one true
false
注意：如果您使用的是 Go 1.18 之前的 Go，只需使用非通用版本findKeys()：
func findKeys(m map[models.LocaleInfo]string, keys ...models.LocaleInfo) (v string, ok bool) {
for _, key := range keys {
if v, ok = m[key]; ok {
return
}
}
return
}
你可以這樣使用findKeys()它：
value, ok := findKeys(name, localeInfo, localeInfoUS, localeInfoGB)
if !ok {
errs.Newf("Unable to find locale for ProductId: %d", id)
}
return value, ok
另請注意，如果您使用的是非通用版本，因為無論如何它都將僅限于您的地圖類型，您可以將更多代碼移入其中：它可以處理密鑰創建，因此您只需傳遞string語言環境值（和clientId）：
func findKeys(names map[models.LocaleInfo]string, clientId int, locales ...string) (v string, ok bool) {
key := models.LocaleInfo{GroupID: clientId}
for _, locale := range locales {
key.Locale = locale
if v, ok = names[key]; ok {
return
}
}
return
}
調用它就像：
value, ok := findKeys(names, int(clientId), code, "US", "GB")
- 1 回答
- 0 關注
- 139 瀏覽
添加回答
舉報
0/150
提交
取消