首页 > 文章列表 > 获取切片长度的结构体字段计算

获取切片长度的结构体字段计算

463 2024-02-05
问题内容

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

我有以下代码

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] 但也不起作用。


正确答案


有没有办法把我的开关换成循环?

您可以执行以下操作:

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
}