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

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

package main

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

const (
    // 定义一个合理的缓冲区大小。
    // 对于大多数局域网和互联网，MTU通常在1500字节左右。
    // 如果你的应用可能接收更大的数据报，可以适当调大，最大不超过65535。
    bufferSize = 2048
)

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

    conn, err := net.ListenUDP("udp", addr)
    if err != nil {
        fmt.Println("Error listening UDP:", err)
        return
    }
    defer conn.Close()
    fmt.Printf("UDP server listening on %s\n", conn.LocalAddr().String())

    // 2. 创建一个缓冲区
    // 这个缓冲区可以重用，无需每次接收都重新分配
    buffer := make([]byte, bufferSize)

    for {
        // 3. 使用 ReadFromUDP 接收数据报
        // n 是实际读取的字节数，addr 是发送方地址
        n, remoteAddr, err := conn.ReadFromUDP(buffer)
        if err != nil {
            if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
                // 处理可能的超时错误 (如果设置了读超时)
                fmt.Println("Read timeout:", err)
                continue
            }
            fmt.Println("Error reading from UDP:", err)
            continue
        }

        // 4. 处理接收到的数据
        // 仅处理 buffer[:n] 部分，即实际接收到的数据
        receivedData := buffer[:n]
        fmt.Printf("Received %d bytes from %s: %s\n", n, remoteAddr.String(), string(receivedData))

        // 5. (可选) 发送响应
        response := []byte("ACK: " + string(receivedData))
        _, err = conn.WriteToUDP(response, remoteAddr)
        if err != nil {
            fmt.Println("Error writing to UDP:", err)
        }
    }
}

// 如何测试：
// 可以在另一个终端使用 netcat 或 Go 客户端发送UDP数据：
// echo "Hello UDP" | nc -u -w1 127.0.0.1 8080
// 或者编写一个简单的Go UDP客户端：
/*
package main

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

func main() {
    conn, err := net.Dial("udp", "127.0.0.1:8080")
    if err != nil {
        fmt.Println("Error dialing UDP:", err)
        return
    }
    defer conn.Close()

    message := "Hello from Go UDP client!"
    _, err = conn.Write([]byte(message))
    if err != nil {
        fmt.Println("Error sending data:", err)
        return
    }
    fmt.Println("Sent:", message)

    // 等待响应
    buffer := make([]byte, 1024)
    conn.SetReadDeadline(time.Now().Add(5 * time.Second)) // 设置读超时
    n, err := conn.Read(buffer)
    if err != nil {
        if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
            fmt.Println("Read timeout for response:", err)
        } else {
            fmt.Println("Error reading response:", err)
        }
        return
    }
    fmt.Printf("Received response: %s\n", string(buffer[:n]))
}
*/