用于作为函数参数的任意结构的切片的接口(golang)

时间:2018-04-18 13:26:48

标签: go struct reflection interface slice

我的应用中有两种不同类型的结构。

我将它显示为一个简化示例:

type typeA struct {
    fieldA1 int
    fieldA2 string
}

type typeB struct {
    fieldB1 float32
    fieldB2 bool
}

首先我初始化它们,然后我想将它们存储在DB中。

a := []typeA{
    {10, "foo"},
    {20, "boo"},
}
b := []typeB{
    {2.5, true},
    {3.5, false},
}

我的第一次尝试是迭代第一个切片,然后是第二个切片。它工作正常,但看起来不DRY。代码明显重复:

printBothArrays(a, b)

// ...

func printBothArrays(dataA []typeA, dataB []typeB) {
    // Not DRY
    for i, row := range dataA {
        fmt.Printf("printBothArrays A row %d: %v\n", i, row)
    }
    for i, row := range dataB {
        fmt.Printf("printBothArrays B row %d: %v\n", i, row)
    }
}

使其成为DRY的错误方法是将其拆分为2个函数:

printArrayA(a)
printArrayB(b)

// ...

func printArrayA(data []typeA) {
    // Not DRY too, because the code is just split between 2 funcs
    for i, row := range data {
        fmt.Printf("printArrayA row %d: %v\n", i, row)
    }
}

func printArrayB(data []typeB) {
    // Not DRY too, because the code is just split between 2 funcs
    for i, row := range data {
        fmt.Printf("printArrayB row %d: %v\n", i, row)
    }
}

这两个功能'签名是不同的,但代码是一样的!

我想到了一个通用函数,它可以接受任何[]结构并只存储它。由于我的商店功能可以采用任何接口{},我想到了这个:

func printArrayAny(data [](interface{})) {
    for i, row := range data {
        fmt.Printf("printArrayAny row %d: %v\n", i, row)
    }
}

但我尝试过不同的方法,但我无法匹配任何共享界面。我收到的错误如下:

cannot use a (type []typeA) as type []interface {} in argument to printArrayAny

我真的不想做任何繁重的事情,比如把它转换为[]map[string]interface,或者使用reflect,因为两个切片都非常大。

有没有办法修改printArrayAny,以便它可以接收和迭代任意[]struct

去游乐场链接:https://play.golang.org/p/qHzcQNUtLIX

1 个答案:

答案 0 :(得分:1)

使用reflect包迭代任意切片类型:

func printArrayAny(data interface{}) {
    v := reflect.ValueOf(data)
    for i := 0; i < v.Len(); i++ {
        fmt.Printf("printArrayAny row %d: %v\n", i, v.Index(i).Interface())
    }
}

Playground Example