如何在不调用Read()的情况下检测io.ReadWriteCloser是否有可读数据？

io.ReadWriteCloser 的 Read() 方法会阻塞直至有数据可读。我希望在不实际调用 Read() 的前提下，检测它是否有可读数据，因为需要在以下两个环节之间执行其他处理：

• 检测到存在可读数据
• 调用 io.Copy(thisReadWriteCloser, anotherReadWriteCloser)

Go 标准库没有提供通用方法直接检测任意 io.ReadWriteCloser 是否有可读数据（不调用 Read()），因为这个接口仅定义了基础读写关闭能力，无状态查询方法。需根据具体实现类型针对性处理：

1. 针对网络连接（net.Conn 类型） #

通过设置极短读超时尝试读取，根据返回结果判断是否有数据。注意读取到的字节需要缓存，后续合并到 io.Copy 的输入中：

conn, ok := rwc.(net.Conn)
if ok {
    // 设置1毫秒读超时
    err := conn.SetReadDeadline(time.Now().Add(1 * time.Millisecond))
    if err != nil {
        // 处理超时设置错误
    }

    buf := make([]byte, 1)
    n, err := conn.Read(buf)
    if err == os.ErrDeadlineExceeded {
        // 无可读数据
    } else if err != nil {
        // 处理其他读取错误
    } else {
        // 检测到数据，执行自定义处理逻辑
        // 合并缓存字节与原连接，传给io.Copy
        reader := io.MultiReader(bytes.NewReader(buf[:n]), conn)
        io.Copy(anotherReadWriteCloser, reader)
    }

    // 恢复默认读超时（若之前有设置）
    conn.SetReadDeadline(time.Time{})
}

2. 针对文件（os.File 类型） #

通过对比文件当前偏移量与文件大小判断是否有未读数据：

file, ok := rwc.(*os.File)
if ok {
    stat, err := file.Stat()
    if err != nil {
        // 处理文件状态获取错误
    }

    offset, err := file.Seek(0, io.SeekCurrent)
    if err != nil {
        // 处理偏移量获取错误
    }

    if offset < stat.Size() {
        // 有可读数据，执行自定义处理逻辑
        io.Copy(anotherReadWriteCloser, file)
    }
}

3. 自定义包装器（通用场景） #

如果无法断言到具体类型，可实现一个包装器，在读写过程中记录是否有未读数据：

import "bytes"

type RWCPeeker struct {
    io.ReadWriteCloser
    buf     bytes.Buffer
    hasData bool
}

func (r *RWCPeeker) Read(p []byte) (n int, err error) {
    if r.buf.Len() > 0 {
        n, err = r.buf.Read(p)
        if err == io.EOF {
            err = nil
        }
        r.hasData = r.buf.Len() > 0
        return
    }

    n, err = r.ReadWriteCloser.Read(p)
    r.hasData = err == nil && n > 0
    return
}

// HasData 检测是否有可读数据
func (r *RWCPeeker) HasData() bool {
    return r.hasData || r.buf.Len() > 0
}

使用时将原始 io.ReadWriteCloser 包装为 RWCPeeker，调用 HasData() 即可检测，不影响后续 io.Copy 逻辑。

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