go - 如何仅使用一种方法删除不同类型的 slice

Question

我有 2 个函数，如下所示

 func removeL2McEntry(a []api.L2McEntry, index int) []api.L2McEntry {
    a = append(a[:index], a[index+1:]...) 
    element
    return a[:len(a)]
 }

 func removeVlagBasedGroup(a []api.VlanPortBased, index int) []api.VlanPortBased {
    a = append(a[:index], a[index+1:]...) 
    return a[:len(a)]
 }

如您所见，这两个函数都在做同样的工作。但我需要将它们分开，因为函数的输出和输入是不同类型的。

我试过:

func removeSlice(a interface{}, idx int) interface{} {
    switch v := a.(type) { 
    case []string:
        v = append(v[:idx], v[idx+1:]...) 
        fmt.Println("is ary", v)
        return v[:len(v)]
    case []int:
        v = append(v[:idx], v[idx+1:]...) 
        fmt.Println("is ary", v)
        return v[:len(v)]
    default:

    }
    return nil
}

但是这种方式重复代码太多。有没有什么办法让它只做一个功能，减少重复代码？

提前致谢。

Answer 1

正如 Adrian 指出的那样，从 slice 中删除一个元素通常是一行代码:

a = append(a[:i], a[i+1]...)
// or
a = a[:i+copy(a[i:], a[i+1:])]

真的不值得为它写一个函数，只要在需要的地方使用这个代码片段就可以了。

如果您确实需要创建一个可以处理任何 slice 类型的函数，可以使用反射来创建它。但是在使用它时，您必须对结果使用类型断言，因为该函数只能返回 interface{} 的静态类型。它也比在您的具体 slice 值上使用上面的代码片段要慢!

可以使用 reflect 来“复制”上述删除步骤包裹。 slice 是 Value.Slice()方法，附加操作是reflect.AppendSlice()功能。

这是它的样子(省略了类型和绑定(bind)检查):

func remove(s interface{}, i int) interface{} {
    v := reflect.ValueOf(s)
    return reflect.AppendSlice(v.Slice(0, i), v.Slice(i+1, v.Len())).Interface()
}

测试它:

is := []int{0, 1, 2, 3}
is = remove(is, 2).([]int)
fmt.Printf("%#v\n", is)

ss := []string{"0", "1", "2", "3"}
ss = remove(ss, 2).([]string)
fmt.Printf("%#v\n", ss)

输出(在 Go Playground 上尝试):

[]int{0, 1, 3}
[]string{"0", "1", "3"}

但再次声明:我不建议任何人使用此(尽管有效)代码，只需直接删除带有原始代码段的元素即可。