使用 JSON 实现 io.WriterTo接口

你在学习Golang相关的知识吗？本文《使用 JSON 实现 io.WriterTo接口》，主要介绍的内容就涉及到，如果你想提升自己的开发能力，就不要错过这篇文章，大家要知道编程理论基础和实战操作都是不可或缺的哦！

我最近发现了这个很酷的界面，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 之前在内存中复制了对象。是否可以直接写入，而且还知道写入了多少字节？

正确答案

要直接写入，请创建一个 io.writer 包装器来计算写入的字节数：

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 方法以编码为编写器，其中编写器是参数的包装器。完成后返回字节数和错误。

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

今天关于《使用 JSON 实现 io.WriterTo接口》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！