Go:为 "set" 个单独结果实施 ManyDecode

Go: Implementing a ManyDecode for a "set" of individual results

我已经实现了一个非常简单的 Decode 方法(现在使用 gob.Decoder)——这对单个响应很有效——它甚至对切片也很有效,但我需要在它所在的位置实现一个 DecodeMany 方法能够解码一组单独的响应(不是切片)。

工作解码方法:

var v MyType
_ = Decode(&v)
...

func Decode(v interface{}) error {
   buf, _ := DoSomething() // func DoSomething() ([]byte, error)
   // error handling omitted for brevity
   return gob.NewDecoder(bytes.NewReader(buf)).Decode(v)
}

我试图为 DecodeMany 方法做的是处理不一定是切片的响应:

var vv []MyType
_ = DecodeMany(&vv)
...

func DecodeMany(vv []interface{}) error {
   for _, g := range DoSomething() { // func DoSomething() []struct{Buf []bytes}
      
      // Use g.Buf as an individual "interface{}"
      // want something like:
      var v interface{} /* Somehow create instance of single vv type? */
      _ = gob.NewDecoder(bytes.NewReader(g.Buf)).Decode(v)
      vv = append(vv, v)
   }
   return
}

上面除了没有编译还有错误:

cannot use &vv (value of type *[]MyType) as type []interface{} in argument to DecodeMany

如果你想修改传递的切片,它必须是一个指针,否则你必须return一个新的切片。此外,如果函数被声明为具有 []interface{} 类型的参数,则只能传递 []interface{} 类型的值而不能传递其他切片类型...除非您使用泛型...

这是开始使用 Go 1.18 中引入的泛型的完美示例。

DecodeMany() 更改为通用的,将 T 类型参数作为切片元素类型:

取指点时

func DecodeMany[T any](vv *[]T) error {
    for _, g := range DoSomething() {
        var v T
        if err := gob.NewDecoder(bytes.NewReader(g.Buf)).Decode(&v); err != nil {
            return err
        }
        *vv = append(*vv, v)
    }
    return nil
}

这是一个简单的应用程序来测试它:

type MyType struct {
    S int64
}

func main() {
    var vv []MyType
    if err := DecodeMany(&vv); err != nil {
        panic(err)
    }
    fmt.Println(vv)
}

func DoSomething() (result []struct{ Buf []byte }) {
    for i := 3; i < 6; i++ {
        buf := &bytes.Buffer{}
        v := MyType{S: int64(i)}
        if err := gob.NewEncoder(buf).Encode(v); err != nil {
            panic(err)
        }
        result = append(result, struct{ Buf []byte }{buf.Bytes()})
    }
    return
}

此输出(在 Go Playground 上尝试):

[{3} {4} {5}]

当 return切片时

如果选择return分片,则不需要传递任何东西,但需要对结果进行赋值:

func DecodeMany[T any]() ([]T, error) {
    var result []T
    for _, g := range DoSomething() {
        var v T
        if err := gob.NewDecoder(bytes.NewReader(g.Buf)).Decode(&v); err != nil {
            return result, err
        }
        result = append(result, v)
    }
    return result, nil
}

使用它:

vv, err := DecodeMany[MyType]()
if err != nil {
    panic(err)
}
fmt.Println(vv)

Go Playground 上试试这个。