Go语言 用JSON实现io.WriterTo

dxpyg8gm  于 2022-12-30  发布在  Go
关注(0)|答案(2)|浏览(97)

我最近发现了这个很酷的界面io.WriterTo
https://godocs.io/io#WriterTo
我想为一些JSON对象实现它。我可以这样做:

package calendar

import (
   "bytes"
   "encoding/json"
   "io"
)

type date struct {
   Month int
   Day int
}

func (d date) WriteTo(w io.Writer) (int64, error) {
   buf := new(bytes.Buffer)
   err := json.NewEncoder(buf).Encode(d)
   if err != nil {
      return 0, err
   }
   return buf.WriteTo(w)
}

但我认为这并不理想,因为它会在内存中复制对象,然后发送到Writer。是否可以直接写入,但也知道写入了多少字节?

pqwbnv8z

pqwbnv8z1#

若要直接写入,请创建一个io.Writer Package 器来计算写入的字节数:

type countingWriter struct {
    n int64
    w io.Writer
}

func (cw *countingWriter) Write(p []byte) (int, error) {
    n, err := cw.w.Write(p)
    cw.n += int64(n)
    return n, err
}

更改WriteTo方法以编码为编写器,其中编写器是参数的 Package 。完成后返回字节计数和错误。

func (d date) WriteTo(w io.Writer) (int64, error) {
    cw := &countingWriter{w: w}
    err := json.NewEncoder(cw).Encode(d)
    return cw.n, err
}

Run an example on the the Go PlayGround

ohfgkhjo

ohfgkhjo2#

有没有可能直接写,而且还知道写了多少字节?
巧合的是,这也是《Go语言》一书中练习7.2的内容,这个练习包含了一个函数的实现过程,其签名为:

func CountingWriter(w io.Writer) (io.Writer, *int64)

返回的指向int64的指针必须随时包含写入返回的io.Writer的字节数。本质上,返回的io.Writerdecorator,因为它通过更新字节计数器增强了初始io.Writer的功能。
首先,让我们创建类型writerFunc,它与io.Writer的唯一方法Write具有相同的参数化:

type writerFunc func([]byte) (int, error)

然后,在writerFunc上定义方法Write(p []byte) (int, error)

func (wf writerFunc) Write(p []byte) (int, error) {
    return wf(p)
}

这样,writerFunc满足io.Writer,并且充当任何func([]byte) (int, error)io.Writer的适配器-也就是说,每当需要io.Writer时,我们可以将func([]byte) (int, error) Package 到writerFunc中。
最后,我们正在寻找的CountingWriter装饰函数:

func CountingWriter(w io.Writer) (io.Writer, *int64) {
    count := new(int64)
    writeAndCount := func(data []byte) (int, error) {
        bytes, err := w.Write(data)
        *count += int64(bytes)
        return bytes, err
    }
    return writerFunc(writeAndCount), count
}

注意最后一条return语句:闭包writeAndCount被 Package 在一个writerFunc中。这是有效的,因为闭包的类型也是func([]byte) (int, error)。正如我们在上面看到的,writerFunc满足io.Writer,这个函数的调用者最终会收到它。

相关问题