Go语言构建高性能异步TCP服务器教程

package main

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

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

func main() {
    fmt.Println("启动", SERVER_TYPE, "服务器在", SERVER_HOST, ":", SERVER_PORT)

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

    fmt.Println("服务器已启动，等待客户端连接...")

    // 2. 循环接受客户端连接
    for {
        conn, err := listener.Accept()
        if err != nil {
            log.Printf("接受连接失败: %s", err.Error())
            continue // 继续接受下一个连接
        }
        fmt.Println("新客户端连接来自:", conn.RemoteAddr().String())

        // 3. 为每个连接启动一个独立的Goroutine进行处理
        go handleConnection(conn)
    }
}

// handleConnection 函数负责处理单个客户端连接的请求
func handleConnection(conn net.Conn) {
    defer func() {
        fmt.Println("关闭客户端连接:", conn.RemoteAddr().String())
        conn.Close() // 确保在函数退出时关闭连接
    }()

    reader := bufio.NewReader(conn)
    writer := bufio.NewWriter(conn)

    for {
        // 4. 读取客户端发送的数据
        // ReadString('\n') 会阻塞直到读到换行符，或者连接关闭/出错
        netData, err := reader.ReadString('\n')
        if err != nil {
            // 通常是客户端断开连接，或者读取错误
            log.Printf("读取客户端数据失败: %s", err.Error())
            return // 退出此Goroutine
        }

        message := strings.TrimSpace(string(netData))
        fmt.Printf("收到来自 %s 的消息: %s\n", conn.RemoteAddr().String(), message)

        // 模拟异步计算或耗时操作
        // 在实际应用中，这里可能是数据库查询、复杂的计算、调用其他微服务等
        if message == "STOP" {
            fmt.Printf("客户端 %s 请求停止连接。\n", conn.RemoteAddr().String())
            break // 退出循环，关闭连接
        }

        // 模拟异步计算，例如延迟500毫秒
        time.Sleep(500 * time.Millisecond) 
        response := fmt.Sprintf("服务器已收到您的消息 '%s'，并在 %s 处理完毕。\n", message, time.Now().Format("15:04:05"))

        // 5. 将处理结果写回客户端
        _, err = writer.WriteString(response)
        if err != nil {
            log.Printf("写入数据到客户端失败: %s", err.Error())
            return
        }
        err = writer.Flush() // 确保数据被发送
        if err != nil {
            log.Printf("刷新写入缓冲区失败: %s", err.Error())
            return
        }
    }
}

package main

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

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

func main() {
    // 建立到服务器的连接
    conn, err := net.Dial(SERVER_TYPE, SERVER_HOST+":"+SERVER_PORT)
    if err != nil {
        log.Fatalf("连接服务器失败: %s", err.Error())
    }
    defer conn.Close() // 确保在main函数退出时关闭连接

    fmt.Println("已连接到服务器:", conn.RemoteAddr().String())

    reader := bufio.NewReader(os.Stdin)
    serverReader := bufio.NewReader(conn)

    for {
        fmt.Print("请输入消息 (输入'STOP'退出): ")
        input, _ := reader.ReadString('\n') // 从标准输入读取
        message := strings.TrimSpace(input)

        // 发送消息到服务器
        _, err = conn.Write([]byte(message + "\n")) // 确保加上换行符
        if err != nil {
            log.Printf("发送数据失败: %s", err.Error())
            return
        }

        if message == "STOP" {
            fmt.Println("客户端请求停止，即将退出。")
            break
        }

        // 读取服务器响应
        serverReply, err := serverReader.ReadString('\n')
        if err != nil {
            log.Printf("读取服务器响应失败: %s", err.Error())
            return
        }
        fmt.Printf("收到服务器响应: %s", serverReply)
        time.Sleep(100 * time.Millisecond) // 稍微等待一下，避免CPU空转
    }
}