Go:为一组单独的结果实现ManyDecode

huangapple go评论83阅读模式
英文:

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

问题

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

工作中的解码方法:

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

func Decode(v interface{}) error {
   buf, _ := DoSomething() // func DoSomething() ([]byte, error)
   // 省略错误处理
   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}
      
      // 使用g.Buf作为单个的"interface{}"
      // 希望像这样:
      var v interface{} /* 如何创建单个vv类型的实例? */
      _ = gob.NewDecoder(bytes.NewReader(g.Buf)).Decode(v)
      vv = append(vv, v)
   }
   return
}

除了无法编译之外,上述代码还有以下错误:

无法将&vv(类型为*[]MyType的值)作为DecodeMany[]interface{}类型的参数使用。

英文:

I have implemented a very simple Decode method (using gob.Decoder for now) - this works well for single responses - it would even work well for slices, but I need to implement a DecodeMany method where it is able to decode a set of individual responses (not a slice).

Working Decode method:

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)
}

What I'm trying to do for a DecodeMany method is to deal with a response that isn't necessarily a slice:

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
}

Besides not compiling the above also has the error of:
> cannot use &vv (value of type *[]MyType) as type []interface{} in argument to DecodeMany

答案1

得分: 1

如果你想修改传递的切片,它必须是一个指针,否则你必须返回一个新的切片。另外,如果函数声明为具有类型[]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}]

当返回一个切片时

如果你选择返回切片,你不需要传递任何东西,但你需要分配结果:

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上尝试这个例子。

英文:

If you want to modify the passed slice, it must be a pointer, else you must return a new slice. Also if the function is declared to have a param of type []interface{}, you can only pass a value of type []interface{} and no other slice types... Unless you use generics...

This is a perfect example to start using generics introduced in Go 1.18.

Change DecodeMany() to be generic, having a T type parameter being the slice element type:

When taking a pointer

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

Here's a simple app to test it:

type MyType struct {
	S int64
}

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

func DoSomething() (result []struct{ Buf []byte }) {
	for i := 3; i &lt; 6; i++ {
		buf := &amp;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
}

This outputs (try it on the Go Playground):

[{3} {4} {5}]

When returning a slice

If you choose to return the slice, you don't have to pass anything, but you need to assign the result:

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(&amp;v); err != nil {
			return result, err
		}
		result = append(result, v)
	}
	return result, nil
}

Using it:

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

Try this one on the Go Playground.

huangapple
  • 本文由 发表于 2022年4月6日 00:36:20
  • 转载请务必保留本文链接:https://go.coder-hub.com/71755407.html
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定