为了账号安全,请及时绑定邮箱和手机立即绑定

如何检查 golang 映射中的三个不同的结构键?

如何检查 golang 映射中的三个不同的结构键?

Go
元芳怎么了 2022-11-23 15:56:31

我有一个下面的方法,我创建了三个不同的结构对象来检查我的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贡献1484条经验 获得超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")


查看完整回答
反对 回复 2022-11-23

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信