Go语言TCP服务器开发：解决Socket读取EOF错误

var buf = make([]byte, 1024)

package main

import (
    "fmt"
    "io"
    "net"
    "os"
    "strings"
)

// handleConnection 处理单个客户端连接
func handleConnection(conn net.Conn) {
    // 确保连接在函数退出时关闭
    defer conn.Close()
    fmt.Printf("接受来自 %s 的连接\n", conn.RemoteAddr().String())

    // 向客户端发送准备就绪消息
    readyMsg := "Ready to receive\n"
    writelen, err := conn.Write([]byte(readyMsg))
    if err != nil {
        fmt.Fprintf(os.Stderr, "写入到socket失败: %s\n", err.Error())
        return
    }
    fmt.Printf("已写入 %d 字节到socket\n", writelen)

    // 预分配一个1024字节的缓冲区用于读取数据
    // 这是解决EOF错误的关键
    buf := make([]byte, 1024)

    for {
        // 从连接中读取数据到缓冲区
        readlen, err := conn.Read(buf)
        if err != nil {
            // 如果错误是io.EOF，表示客户端关闭了连接
            if err == io.EOF {
                fmt.Printf("客户端 %s 关闭了连接\n", conn.RemoteAddr().String())
            } else {
                // 其他读取错误
                fmt.Fprintf(os.Stderr, "从socket读取时发生错误: %s\n", err.Error())
            }
            return // 退出循环，关闭此连接
        }

        if readlen == 0 {
            // readlen为0但err不是io.EOF通常表示连接被优雅关闭，
            // 但并非所有操作系统都如此，io.EOF是更可靠的判断方式。
            // 这里作为额外的安全检查。
            fmt.Printf("客户端 %s 发送了空数据包或连接已关闭\n", conn.RemoteAddr().String())
            return
        }

        // 将读取到的字节转换为字符串并打印
        // 注意：只转换实际读取的部分 (buf[:readlen])
        clientMsg := string(buf[:readlen])
        fmt.Printf("客户端 %s 说: '%s'\n", conn.RemoteAddr().String(), strings.TrimSpace(clientMsg))

        // 将收到的数据原样回写给客户端（Echo服务器逻辑）
        _, err = conn.Write(buf[:readlen])
        if err != nil {
            fmt.Fprintf(os.Stderr, "回写数据到socket失败: %s\n", err.Error())
            return
        }
    }
}

// listenAndServe 启动TCP监听服务
func listenAndServe(port string) {
    listener, err := net.Listen("tcp", ":"+port)
    if err != nil {
        fmt.Fprintf(os.Stderr, "无法在socket上监听: %s\n", err.Error())
        os.Exit(1)
    }
    defer listener.Close()
    fmt.Printf("服务器正在监听端口: %s\n", port)

    for {
        // 接受新的客户端连接
        conn, err := listener.Accept()
        if err != nil {
            fmt.Fprintf(os.Stderr, "接受连接失败: %s\n", err.Error())
            continue // 继续尝试接受下一个连接
        }
        // 为每个新连接启动一个goroutine进行处理
        go handleConnection(conn)
    }
}

func main() {
    if len(os.Args) < 2 {
        fmt.Println("用法: go run your_program.go <端口号>")
        os.Exit(1)
    }
    port := os.Args[1]
    listenAndServe(port)
}