Go语言中如何完整读取TCP连接上的所有字节流

func ReadAll(r Reader) ([]byte, error)

package main

import (
    "bytes"
    "fmt"
    "io" // 在Go 1.16+版本中，推荐使用io.ReadAll
    "net"
    "time"
)

func main() {
    // 场景1: 从一个bytes.Buffer读取，模拟一个已知结束的数据流
    fmt.Println("--- 场景1: 从bytes.Buffer读取 ---")
    dataWithCRLF := []byte("Hello\r\nWorld!\r\nThis is a test.\r\n")
    bufferReader := bytes.NewReader(dataWithCRLF)

    // 使用 io.ReadAll 读取所有字节
    allBytes, err := io.ReadAll(bufferReader)
    if err != nil {
        fmt.Printf("从bytes.Buffer读取错误: %v\n", err)
        return
    }
    fmt.Printf("读取到的所有字节 (%d bytes):\n%s\n", len(allBytes), string(allBytes))
    fmt.Println("---------------------------------")

    // 场景2: 模拟TCP连接读取，需要服务端关闭连接才能触发EOF
    fmt.Println("\n--- 场景2: 模拟TCP连接读取 (需要服务端关闭) ---")
    listener, err := net.Listen("tcp", "127.0.0.1:8080")
    if err != nil {
        fmt.Printf("启动服务器失败: %v\n", err)
        return
    }
    defer listener.Close()
    fmt.Println("服务器已启动，监听 127.0.0.1:8080")

    // 启动一个Goroutine作为服务器端
    go func() {
        conn, err := listener.Accept()
        if err != nil {
            fmt.Printf("服务器接受连接失败: %v\n", err)
            return
        }
        defer conn.Close() // 确保连接关闭，从而发送EOF给客户端
        fmt.Println("服务器: 客户端已连接")
        conn.Write([]byte("TCP data line 1\r\n"))
        time.Sleep(50 * time.Millisecond) // 模拟数据传输延迟
        conn.Write([]byte("TCP data line 2\r\n"))
        fmt.Println("服务器: 数据发送完毕，关闭连接以发送EOF")
        // conn.Close() 将在defer语句中执行，发送EOF
    }()

    // 客户端连接服务器并读取
    clientConn, err := net.Dial("tcp", "127.0.0.1:8080")
    if err != nil {
        fmt.Printf("客户端连接失败: %v\n", err)
        return
    }
    defer clientConn.Close()
    fmt.Println("客户端: 已连接服务器")
    fmt.Println("客户端: 尝试读取所有数据...")

    // 关键点：io.ReadAll 会阻塞直到服务器关闭连接（发送EOF）
    // 或者发生读取错误
    allClientBytes, err := io.ReadAll(clientConn) // clientConn 实现了 io.Reader 接口
    if err != nil {
        fmt.Printf("客户端读取错误: %v\n", err)
        return
    }
    fmt.Printf("客户端: 读取到的所有字节 (%d bytes):\n%s\n", len(allClientBytes), string(allClientBytes))
    fmt.Println("---------------------------------")
}