Go语言中读取TCP连接所有字节的实用指南

package main

import (
    "bytes"
    "fmt"
    "io"
    "log"
    "net" // 实际TCP连接场景需要
    "time" // 用于模拟TCP连接的延迟关闭
)

func main() {
    // 示例一：使用 bytes.Reader 模拟数据流
    fmt.Println("--- 示例一：使用 bytes.Reader 模拟数据流 ---")
    mockData := "Hello, Go!\r\nThis is a multi-line message.\r\nEnd of stream."
    reader := bytes.NewReader([]byte(mockData))

    allBytes, err := io.ReadAll(reader)
    if err != nil {
        log.Fatalf("读取模拟数据时发生错误: %v", err)
    }

    fmt.Printf("成功读取所有模拟字节:\n%s\n", string(allBytes))

    // 示例二：使用 TCP 连接 (概念性演示，需要一个简单的服务器来配合)
    fmt.Println("\n--- 示例二：使用 TCP 连接 (概念性演示) ---")
    // 启动一个简单的TCP服务器，在发送数据后关闭连接
    go func() {
        listener, err := net.Listen("tcp", ":8080")
        if err != nil {
            log.Fatalf("无法启动服务器: %v", err)
        }
        defer listener.Close()
        fmt.Println("服务器已启动，监听 :8080")

        conn, err := listener.Accept()
        if err != nil {
            log.Printf("服务器接受连接失败: %v", err)
            return
        }
        defer conn.Close()
        fmt.Println("服务器已接受连接")

        // 模拟发送数据
        _, err = conn.Write([]byte("Data from server.\r\nAnother line.\r\n"))
        if err != nil {
            log.Printf("服务器写入数据失败: %v", err)
        }
        // 关键：服务器关闭连接，客户端的 io.ReadAll 才能收到 EOF
        time.Sleep(100 * time.Millisecond) // 确保数据已发送
        fmt.Println("服务器关闭连接。")
    }()

    // 客户端连接到服务器并读取所有数据
    time.Sleep(50 * time.Millisecond) // 等待服务器启动
    clientConn, err := net.Dial("tcp", "localhost:8080")
    if err != nil {
        log.Fatalf("客户端连接服务器失败: %v", err)
    }
    defer clientConn.Close()
    fmt.Println("客户端已连接到服务器。")

    // 从TCP连接读取所有字节
    tcpBytes, err := io.ReadAll(clientConn)
    if err != nil && err != io.EOF { // io.EOF 是正常结束，无需作为错误处理
        log.Fatalf("从TCP连接读取所有字节时发生错误: %v", err)
    }

    fmt.Printf("从TCP连接接收到所有数据:\n%s\n", string(tcpBytes))
    fmt.Println("客户端读取完成。")
}

--- 示例一：使用 bytes.Reader 模拟数据流 ---
成功读取所有模拟字节:
Hello, Go!
This is a multi-line message.
End of stream.

--- 示例二：使用 TCP 连接 (概念性演示) ---
服务器已启动，监听 :8080
客户端已连接到服务器。
服务器已接受连接
服务器关闭连接。
从TCP连接接收到所有数据:
Data from server.
Another line.

客户端读取完成。