Go 中使用结构体嵌入简化自定义 Reader 实现

通过结构体嵌入 `*bytes.reader`,可直接复用其所有方法(如 `read`),无需手动代理,同时支持动态替换底层字节切片,适用于需复用 `io.reader` 接口但需灵活更新数据的场景(如配合 `json.decoder`)。

在 Go 中,若希望封装 bytes.Reader 并支持运行时更换底层 []byte,最简洁、符合 Go 惯用法的方式是结构体嵌入(embedding),而非手动实现方法代理。嵌入 *bytes.Reader 后,Go 编译器会自动将 bytes.Reader 的所有导出方法(如 Read, Seek, Len, Reset 等)提升为 EZReader 的方法,无需重复声明。

以下是推荐实现:

type EZReader struct {
    *bytes.Reader
}

// Replace 替换底层数据,重置读取位置到开头
func (r *EZReader) Replace(data []byte) {
    r.Reader = bytes.NewReader(data)
}

// 可选:提供 Reset 方法(语义更清晰,且兼容 io.Seeker)
func (r *EZReader) Reset(data []byte) {
    r.Reader = bytes.NewReader(data)
}

使用示例:

reader := &EZReader{bytes.NewReader([]byte(`{"name":"Alice"}`))}
decoder := json.NewDecoder(reader)

var person struct{ Name string }
if err := decoder.Decode(&person); err != nil {
    log.Fatal(err)
}
fmt.Println(person.Name) // "Alice"

// 动态替换数据,无需重建 decoder 或 reader 实例
reader.Replace([]byte(`{"name":"Bob"}`))
if err := decoder.Decode(&person); err != nil {
    log.Fatal(err)
}
fmt.Println(person.Name) // "Bob"

优势说明

  • 零冗余代码:无需手写 Read, Len, Seek 等方法;
  • 完全兼容 io.Reader:EZReader 自动满足 io.Reader, io.Seeker, io.ByteReader 等接口;
  • 内存高效:仅持有一个指针,无额外字段开销;
  • 语义清晰:Replace 或 Reset 明确表达“切换数据源”意图。

⚠️ 注意事项

  • 嵌入后 *bytes.Reader 字段是公开的(即 reader.Reader 可被外部访问),若需严格封装,应避免嵌入而改用组合 + 显式方法代理(但本场景中暴露无害,且符合标准库设计风格,如 bufio.Reader 也公开其 rd io.Reader 字段);
  • bytes.NewReader() 总是将读取位置重置为 0,因此 Replace 天然具备“重播”能力;
  • 若需保留原读取位置(如从中间继续读),应改用 r.Seek(0, io.SeekStart) 配合 r.Reset(data),或自行管理偏移量。

总之,结构体嵌入是 Go 中实现“轻量级行为继承”的核心机制。它让 EZReader 既是 bytes.Reader 的增强版,又是真正的 io.Reader,完美契合“一次创建、多次复用、动态更新”的需求。