计算结构字段中的切片长度

count slice length in struct field

提问人:Barry Marples 提问时间:11/30/2022 更新时间:12/1/2022 访问量:61

问:

我想我需要一个比我的大小写开关更好的解决方案,因为结构体获得了更多的字段,我的函数将变得冗长。有没有办法将我的开关换成环路?

我有以下代码

type Things struct {
    StreetNames    []string `json:"streetNames"`
    Letters        []string `json:"letters"`
    MaleNames      []string `json:"maleNames"`
}

func CheckCategories(data *Things, filePath string) error {
    errMsg := "list has no values or is a missing category in File: " + filePath
    categories := []string{"street_name", "letter", "male_first_name"}
    for _, value := range categories {
        switch value {
        case "street_name":
            if len(data.StreetNames) == 0 {
                return errors.New("street_name " + errMsg)
            }
        case "letter":
            if len(data.Letters) == 0 {
                return errors.New("letter " + errMsg)
            }
        case "male_first_name":
            if len(data.MaleNames) == 0 {
                return errors.New("male_first_name " + errMsg)
            }
        }
    }
    return nil
}

这对我有用,但真正的结构包含 12 个字段,这使得我的函数冗长且重复。

我试过了

for _, value := range categories {
        if len("data." + value) == 0 {
            return errors.New(value + errMsg)
}

但是当我运行代码时,我花了一点时间注意到它没有按预期工作,我得到了字符串的长度。我尝试过 data[value],但也没有用。

go struct 切片

评论

0赞 Roman Kiselenko 11/30/2022
有许多很好的验证库,例如 github.com/go-playground/validatorgithub.com/go-ozzo/ozzo-validation
1赞 Adrian 11/30/2022
如果不使用反射,则无法动态访问结构字段。

答:

1赞 mkopriva 12/1/2022 #1

有没有办法将我的开关换成环路?

您可以执行以下操作:

type Things struct {
    StreetNames []string `json:"streetNames"`
    Letters     []string `json:"letters"`
    MaleNames   []string `json:"maleNames"`
}

func CheckCategories(data *Things, filePath string) error {
    errMsg := "list has no values or is a missing category in File: " + filePath
    categories := []struct{
        name  string
        slice []string
    }{
        {"street_name", data.StreetNames},
        {"letter", data.Letters},
        {"male_first_name", data.MaleNames},
    }

    for _, v := range categories {
        if len(v.slice) == 0 {
            return errors.New(v.name + " " + errMsg)
        }
    }
    return nil
}