怎样用Golang开发TCP服务器 实现简单网络通信案例

net

package main

import (
    "bufio"
    "fmt"
    "log"
    "net"
    "strings"
    "time" // 引入time包，用于设置读取超时

    // 简单模拟一个请求处理逻辑，避免过于复杂的业务代码
    "strconv"
)

// handleConnection 处理每个客户端连接的逻辑
func handleConnection(conn net.Conn) {
    // 确保连接在函数退出时关闭，这很重要，避免资源泄露
    defer func() {
        log.Printf("Closing connection from %s", conn.RemoteAddr().String())
        conn.Close()
    }()

    log.Printf("New connection from %s", conn.RemoteAddr().String())

    // 设置一个读取超时，防止客户端长时间不发送数据导致goroutine阻塞
    // 生产环境中这个值可能需要根据业务场景精细调整
    conn.SetReadDeadline(time.Now().Add(60 * time.Second)) // 60秒无数据则超时

    reader := bufio.NewReader(conn)

    for {
        // 尝试读取一行数据，直到遇到换行符或达到超时
        message, err := reader.ReadString('\n')
        if err != nil {
            // 如果是EOF，说明客户端关闭了连接
            if err.Error() == "EOF" {
                log.Printf("Client %s disconnected.", conn.RemoteAddr().String())
            } else if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
                log.Printf("Read timeout for client %s, closing connection.", conn.RemoteAddr().String())
            } else {
                log.Printf("Error reading from client %s: %v", conn.RemoteAddr().String(), err)
            }
            return // 发生错误或超时，退出处理循环，连接将被defer关闭
        }

        // 移除消息末尾的换行符和回车符
        message = strings.TrimSpace(message)
        log.Printf("Received from %s: %s", conn.RemoteAddr().String(), message)

        // 简单的业务逻辑：如果消息是数字，就返回它的两倍
        // 否则，就返回一个通用的“Echo”消息
        response := ""
        if num, parseErr := strconv.Atoi(message); parseErr == nil {
            response = fmt.Sprintf("Double of %s is %d\n", message, num*2)
        } else {
            response = fmt.Sprintf("Echo: %s\n", message)
        }

        // 写回数据给客户端
        _, 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(60 * time.Second))
    }
}

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

    log.Printf("TCP server listening on %s", port)

    for {
        // 接受一个新的连接
        conn, err := listener.Accept()
        if err != nil {
            log.Printf("Error accepting connection: %v", err)
            // 这里可以根据错误类型决定是否继续循环，
            // 例如，如果是临时网络问题，可以短暂等待再重试
            continue
        }

        // 为每个新连接启动一个goroutine来处理
        go handleConnection(conn)
    }
}

net.Listen

listener.Accept()

go handleConnection(conn)

go handleConnection(conn)

net

defer conn.Close()

handleConnection

defer

conn.SetReadDeadline

conn.SetWriteDeadline