Go语言中高效接收完整UDP数据报的实践指南

func (c *UDPConn) ReadFromUDP(b []byte) (n int, addr *UDPAddr, err error)

package main

import (
    "fmt"
    "net"
    "time"
)

const (
    // 定义一个合理的缓冲区大小。
    // 对于大多数局域网环境，1500字节（以太网MTU）是一个常见的选择。
    // 如果需要处理可能的最大UDP数据报，可以设置为65507。
    bufferSize = 1500 
)

func main() {
    // 1. 监听UDP地址
    udpAddr, err := net.ResolveUDPAddr("udp", ":8080")
    if err != nil {
        fmt.Println("Error resolving UDP address:", err)
        return
    }

    conn, err := net.ListenUDP("udp", udpAddr)
    if err != nil {
        fmt.Println("Error listening UDP:", err)
        return
    }
    defer conn.Close()

    fmt.Printf("UDP服务器正在监听 %s\n", udpAddr.String())

    // 2. 循环接收数据报
    buffer := make([]byte, bufferSize) // 分配一个缓冲区
    for {
        // 使用 ReadFromUDP 接收数据报
        n, remoteAddr, err := conn.ReadFromUDP(buffer)
        if err != nil {
            fmt.Println("Error reading from UDP:", err)
            continue
        }

        // n 就是实际接收到的字节数
        receivedData := buffer[:n] // 根据 n 截取有效数据

        fmt.Printf("[%s] 收到来自 %s 的 %d 字节数据: %s\n",
            time.Now().Format("15:04:05"),
            remoteAddr.String(),
            n,
            string(receivedData), // 假设是文本数据
        )
    }
}

package main

import (
    "fmt"
    "net"
    "time"
)

func main() {
    // 1. 解析目标UDP地址
    serverAddr, err := net.ResolveUDPAddr("udp", "127.0.0.1:8080")
    if err != nil {
        fmt.Println("Error resolving server address:", err)
        return
    }

    // 2. 创建一个UDP连接（客户端通常不需要绑定特定端口，系统会自动分配）
    conn, err := net.DialUDP("udp", nil, serverAddr)
    if err != nil {
        fmt.Println("Error dialing UDP:", err)
        return
    }
    defer conn.Close()

    fmt.Printf("UDP客户端已连接到 %s\n", serverAddr.String())

    // 3. 发送多条消息
    messages := []string{
        "Hello, Go UDP!",
        "This is a test message.",
        "Another short message.",
        "A slightly longer message to demonstrate buffer handling in the server.",
        "This message is specifically designed to be longer than the previous ones but still fit within a typical MTU.",
    }

    for i, msg := range messages {
        data := []byte(msg)
        _, err := conn.Write(data) // 或者 conn.WriteToUDP(data, serverAddr)
        if err != nil {
            fmt.Printf("Error sending message %d: %v\n", i, err)
            continue
        }
        fmt.Printf("发送消息 (%d 字节): %s\n", len(data), msg)
        time.Sleep(500 * time.Millisecond) // 稍作延迟
    }

    fmt.Println("所有消息发送完毕。")
}

// ... (之前的代码)
for {
    n, remoteAddr, err := conn.ReadFromUDP(buffer)
    if err != nil {
        fmt.Println("Error reading from UDP:", err)
        continue
    }

    // 拷贝数据以避免竞态条件，因为 buffer 会在下一次循环中被重用
    dataCopy := make([]byte, n)
    copy(dataCopy, buffer[:n])

    go func(data []byte, addr *net.UDPAddr) {
        // 在这里处理接收到的数据
        fmt.Printf("[Goroutine] 收到来自 %s 的 %d 字节数据: %s\n",
            addr.String(),
            len(data),
            string(data),
        )
        // 进行业务逻辑处理...
    }(dataCopy, remoteAddr)
}