Go语言TCP服务器：并发处理多个客户端连接的实践指南

// 错误示例：尝试传递 *net.Conn
// func handleClient(con *net.Conn) { ... }
// go handleClient(&con); // 这里的 &con 会导致类型不匹配或方法找不到

// 正确示例：直接传递 net.Conn 接口值
go handleClient(conn)

func handleClient(conn net.Conn) {
    // 在这里可以直接使用 conn.RemoteAddr(), conn.Read(), conn.Write() 等方法
    // 例如：log.Println("新连接来自:", conn.RemoteAddr().String())
}

package main

import (
    "fmt"
    "io"
    "log"
    "net"
    "time"
)

const (
    SERVER_HOST = "localhost"
    SERVER_PORT = "8080"
    SERVER_TYPE = "tcp"
)

func main() {
    // 1. 启动TCP监听器
    listener, err := net.Listen(SERVER_TYPE, SERVER_HOST+":"+SERVER_PORT)
    if err != nil {
        log.Fatalf("无法启动TCP监听器: %v", err)
    }
    defer listener.Close() // 确保在main函数退出时关闭监听器
    log.Printf("TCP服务器已启动，监听在 %s:%s", SERVER_HOST, SERVER_PORT)

    // 2. 循环接受客户端连接
    for {
        conn, err := listener.Accept()
        if err != nil {
            log.Printf("接受连接失败: %v", err)
            // 如果是临时错误，可以考虑短暂等待后重试，这里直接跳过
            continue
        }
        log.Printf("新连接来自: %s", conn.RemoteAddr().String())

        // 3. 为每个新连接启动一个独立的goroutine来处理
        // 关键点：直接传递 net.Conn 接口值
        go handleClient(conn)
    }
}

// handleClient 函数负责处理单个客户端连接
func handleClient(conn net.Conn) {
    // 确保在函数退出时关闭连接，释放资源
    defer func() {
        log.Printf("客户端 %s 连接已关闭。", conn.RemoteAddr().String())
        conn.Close()
    }()

    // 设置读取超时，防止客户端长时间不发送数据导致资源阻塞
    // conn.SetReadDeadline(time.Now().Add(5 * time.Minute))

    buf := make([]byte, 1024) // 创建一个缓冲区用于读写数据

    for {
        // 从连接中读取数据
        n, err := conn.Read(buf)
        if err != nil {
            if err == io.EOF {
                // io.EOF 表示客户端已关闭连接
                log.Printf("客户端 %s 已断开连接。", conn.RemoteAddr().String())
            } else {
                log.Printf("读取客户端 %s 数据失败: %v", conn.RemoteAddr().String(), err)
            }
            return // 发生错误或连接关闭，退出当前goroutine
        }

        receivedData := string(buf[:n])
        log.Printf("从 %s 接收到数据: %s", conn.RemoteAddr().String(), receivedData)

        // 将接收到的数据原样写回客户端（回显）
        _, err = conn.Write([]byte("服务器收到: " + receivedData))
        if err != nil {
            log.Printf("写入客户端 %s 数据失败: %v", conn.RemoteAddr().String(), err)
            return // 写入失败，退出当前goroutine
        }
    }
}

// --------------------------------------------------------------------------------
// 辅助：一个简单的TCP客户端，用于测试上述服务器
// 将以下代码保存为 client.go 并在另一个终端运行
/*
package main

import (
    "bufio"
    "fmt"
    "log"
    "net"
    "os"
    "time"
)

func main() {
    conn, err := net.Dial("tcp", "localhost:8080")
    if err != nil {
        log.Fatalf("连接服务器失败: %v", err)
    }
    defer conn.Close()
    fmt.Println("已连接到服务器。输入消息并按回车发送。")

    reader := bufio.NewReader(os.Stdin)
    for {
        fmt.Print(">> ")
        input, _ := reader.ReadString('\n') // 读取用户输入
        _, err := conn.Write([]byte(input)) // 发送数据到服务器
        if err != nil {
            log.Printf("发送数据失败: %v", err)
            return
        }

        // 读取服务器响应
        conn.SetReadDeadline(time.Now().Add(5 * time.Second)) // 设置读取超时，防止阻塞
        buf := make([]byte, 1024)
        n, err := conn.Read(buf)
        if err != nil {
            if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
                fmt.Println("读取服务器响应超时。")
            } else {
                log.Printf("读取服务器响应失败: %v", err)
            }
            return
        }
        fmt.Printf("<< %s", string(buf[:n])) // 打印服务器响应
    }
}
*/