Go语言并发TCP服务器：正确处理net.Conn连接对象

// 假设 con 是 listener.Accept() 返回的 net.Conn
// go handleClient(&con) // 错误！试图传递 net.Conn 的地址
// func handleClient(c *net.Conn) { ... } // 错误！参数类型不匹配

// 假设 con 是 listener.Accept() 返回的 net.Conn
go handleClient(con) // 正确！直接传递 net.Conn 接口值
func handleClient(c net.Conn) { // 正确！参数类型为 net.Conn 接口
    // 现在可以在 c 上调用 Read, Write, RemoteAddr 等方法
    // 例如：fmt.Println("Client connected from:", c.RemoteAddr().String())
}

package main

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

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

// handleClient 处理单个客户端连接
func handleClient(conn net.Conn) {
    // 确保连接在函数退出时关闭
    defer func() {
        log.Printf("Closing connection from %s", conn.RemoteAddr().String())
        conn.Close()
    }()

    log.Printf("New client connected: %s", conn.RemoteAddr().String())

    // 设置读取超时，防止恶意客户端或僵尸连接
    conn.SetReadDeadline(time.Now().Add(5 * time.Minute))

    // 缓冲区用于读取客户端数据
    buffer := make([]byte, 1024)
    for {
        // 读取客户端发送的数据
        n, err := conn.Read(buffer)
        if err != nil {
            if err == io.EOF {
                log.Printf("Client %s disconnected.", conn.RemoteAddr().String())
            } else {
                log.Printf("Error reading from client %s: %v", conn.RemoteAddr().String(), err)
            }
            return // 退出循环，关闭连接
        }

        // 将读取到的数据转换为字符串并打印
        receivedData := string(buffer[:n])
        log.Printf("Received from %s: %s", conn.RemoteAddr().String(), receivedData)

        // 向客户端发送响应
        response := fmt.Sprintf("Server received: %s (at %s)\n", receivedData, time.Now().Format("15:04:05"))
        _, err = conn.Write([]byte(response))
        if err != nil {
            log.Printf("Error writing to client %s: %v", conn.RemoteAddr().String(), err)
            return // 退出循环，关闭连接
        }

        // 每次成功读写后，重置读取超时时间
        conn.SetReadDeadline(time.Now().Add(5 * time.Minute))
    }
}

func main() {
    log.Printf("Starting %s server on %s:%s", SERVER_TYPE, SERVER_HOST, SERVER_PORT)

    // 监听TCP端口
    listener, err := net.Listen(SERVER_TYPE, SERVER_HOST+":"+SERVER_PORT)
    if err != nil {
        log.Fatalf("Error listening: %v", err)
    }
    // 确保监听器在main函数退出时关闭
    defer listener.Close()

    log.Println("Server listening...")

    for {
        // 接受新的客户端连接
        conn, err := listener.Accept()
        if err != nil {
            log.Printf("Error accepting connection: %v", err)
            continue // 继续接受下一个连接
        }

        // 为每个新连接启动一个goroutine进行处理
        go handleClient(conn)
    }
}