使用bufio.Reader Peek后，如何将其转换回io.ReadWriteCloser？

你遇到的问题核心在于bufio.NewReader的预读机制：它会提前从原io.ReadWriteCloser读取数据到内部缓冲区（默认4096字节），哪怕你只调用Peek(1)，也会触发预读操作，直接推进原对象的读取指针。这就导致后续再操作原对象时，数据已经被bufio.Reader“消耗”，自然返回EOF。

bufio.Reader无法直接转换回原io.ReadWriteCloser，因为它没有提供将缓存数据回吐给原对象的能力，除非原对象本身支持指针回退（比如实现了io.Seeker接口）。下面是几种可行的解决思路：

1. 使用Go 1.21+新增的io.PeekReader（推荐） #

io.PeekReader是专门为“不推进原读取器指针的peek操作”设计的，它不会预读多余数据，peek后原对象的读取位置完全不受影响。

示例代码：

import "io"

// i是你的io.ReadWriteCloser实例
pr := io.NewPeekReader(i)
// Peek指定字节数，不会改变原i的读取指针
data, err := pr.Peek(1)
if err != nil {
    // 处理错误逻辑
}

// 后续直接使用原i即可，读取指针未被改动

2. 自行实现带缓存的包装器（兼容低版本Go） #

如果你的Go版本低于1.21，可以自己封装一个读取器，把peek到的数据暂存到缓存中，后续读取优先返回缓存内容，避免直接操作原对象导致指针偏移。

示例代码：

import "io"

type PeekableRW struct {
    src io.ReadWriteCloser
    buf []byte
}

// Read 优先返回缓存数据，缓存为空时再读取原对象
func (p *PeekableRW) Read(b []byte) (n int, err error) {
    if len(p.buf) > 0 {
        n = copy(b, p.buf)
        p.buf = p.buf[n:]
        return n, nil
    }
    return p.src.Read(b)
}

// Peek 读取指定字节数到缓存，不推进读取指针
func (p *PeekableRW) Peek(n int) ([]byte, error) {
    if len(p.buf) >= n {
        return p.buf[:n], nil
    }

    need := n - len(p.buf)
    temp := make([]byte, need)
    readCnt, err := p.src.Read(temp)
    p.buf = append(p.buf, temp[:readCnt]...)
    
    if err != nil {
        return p.buf, err
    }
    if len(p.buf) < n {
        return p.buf, io.EOF
    }
    return p.buf[:n], nil
}

// 直接转发Write和Close方法给原对象
func (p *PeekableRW) Write(b []byte) (n int, err error) {
    return p.src.Write(b)
}

func (p *PeekableRW) Close() error {
    return p.src.Close()
}

// 使用方式
prw := &PeekableRW{src: i}
data, err := prw.Peek(1)
// 后续统一使用prw进行读写操作，避免直接操作原i

3. 针对支持io.Seeker的对象回退指针 #

如果你的io.ReadWriteCloser同时实现了io.Seeker接口（比如本地文件os.File），可以在peek前记录当前位置，操作后再回退指针：

示例代码：

import "io"

// 确保i实现了io.Seeker
pos, err := i.Seek(0, io.SeekCurrent)
if err != nil {
    // 处理错误
}

bi := bufio.NewReader(i)
bi.Peek(1)

// 将原对象的读取指针回退到之前的位置
_, err = i.Seek(pos, io.SeekStart)
if err != nil {
    // 处理错误
}

// 现在原i可以正常复用

注意：这个方案只适用于支持Seek的对象，像TCP连接、管道这类无法回退指针的对象无法使用。

内容的提问来源于stack exchange，提问作者user892960