go - 将未知长度的 slice 中的值分配给 Go 中的结构体？

Question

我从github上的一些有用的包中找到了下面的案例，它看起来是如此丑陋和愚蠢。我认为更好的代码设计可以避免这种情况，但是如果遇到这种情况，是不是可以写得更简洁呢？

有没有更好的方法来代替连续检查 slice 长度？

package main

type NumberSet struct {
    SetName string
    Number1 int
    Number2 int
    Number3 int
}

func main() {
    set := &NumberSet{SetName: "test"}
    var numbers []int
    // assume that we don't know the actual length of slice
    numbers = append(numbers, 1)
    numbers = append(numbers, 2)

    length := len(numbers)
    if length > 0 {
        set.Number1 = numbers[0]
    }
    if length > 1 {
        set.Number2 = numbers[1]
    }
    if length > 2 {
        set.Number3 = numbers[2]
    }
}

Answer 1

您可以使用for range来范围化数字，这样您就可以省略长度检查，并使用switch进行字段分配，例如:

for i, v := range numbers {
    switch i {
    case 0:
        set.Number1 = v
    case 1:
        set.Number2 = v
    case 2:
        set.Number3 = v
    }
}

在 Go Playground 上尝试一下.

您还可以列出另一个 slice 中的字段地址，并使用单个循环来进行分配:

ps := []*int{&set.Number1, &set.Number2, &set.Number3}
for i, v := range numbers {
    *ps[i] = v
}

在 Go Playground 上尝试一下.

如果性能并不重要，并且分配的字段是连续的，则可以使用反射来进行分配，而无需枚举字段，例如:

rv := reflect.ValueOf(set).Elem()
for i, v := range numbers {
    rv.Field(1 + i).Set(reflect.ValueOf(v)) // 1 is the offset of Number1
}

在 Go Playground 上尝试一下.