如何在 golang 映射中查找三个不同的结构键？

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《如何在 golang 映射中查找三个不同的结构键？》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

我有一个下面的方法，我创建三个不同的结构对象来检查我的 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 块看起来也非常嵌套，我认为这也可以改进很多。

正确答案

特殊的“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")

查看相关问题：Check if key exists in multiple maps in one condition

终于介绍完啦！小伙伴们，这篇关于《如何在 golang 映射中查找三个不同的结构键？》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！