我有一個(gè)下面的方法，我創(chuàng)建了三個(gè)不同的結(jié)構(gòu)對(duì)象來檢查我的name地圖中的鍵。如果我能找到該值，我會(huì)返回，否則我會(huì)返回空字符串和狀態(tài)。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}有什么方法可以改進(jìn)上面的代碼，我不必為每個(gè)美國(guó)和英國(guó)區(qū)域設(shè)置信息保留單獨(dú)的行。我的 if 塊看起來也很嵌套，我認(rèn)為這也可以改進(jìn)很多。
1 回答
翻閱古今
TA貢獻(xiàn)1780條經(jīng)驗(yàn) 獲得超5個(gè)贊
特殊的“comma-ok”形式只能在作業(yè)中使用，所以你不能讓它更緊湊。
你可以做的是編寫一個(gè)實(shí)用函數(shù)，它使用 for 循環(huán)來檢查任意數(shù)量的鍵。
使用 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
}
測(cè)試它：
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
另請(qǐng)注意，如果您使用的是非通用版本，因?yàn)闊o論如何它都將僅限于您的地圖類型，您可以將更多代碼移入其中：它可以處理密鑰創(chuàng)建，因此您只需傳遞string語言環(huán)境值（和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
}
調(diào)用它就像：
value, ok := findKeys(names, int(clientId), code, "US", "GB")
- 1 回答
- 0 關(guān)注
- 148 瀏覽
添加回答
舉報(bào)
0/150
提交
取消