Golang UDP多线程收发示例详解

package main
import (
"fmt"
"net"
"time"
)
const numWorkers = 3 // 启动3个接收协程
func handlePacket(conn net.UDPConn, data []byte, addr net.UDPAddr) {
fmt.Printf("来自 %s 的消息: %s\n", addr.String(), string(data))
// 模拟响应
response := "收到: " + string(data)
conn.WriteToUDP([]byte(response), addr)
}
func worker(id int, conn net.UDPConn, packets chan net.UDPAddr) {
buffer := make([]byte, 1024)
fmt.Printf("worker-%d 开始监听\n", id)
for range packets {
    n, addr, err := conn.ReadFromUDP(buffer)
    if err != nil {
        fmt.Printf("worker-%d 读取错误: %v\n", id, err)
        continue
    }

    // 复制数据，避免后续读取覆盖
    data := make([]byte, n)
    copy(data, buffer[:n])

    go handlePacket(conn, data, addr)
}
}
func main() {
addr, _ := net.ResolveUDPAddr("udp", ":8080")
conn, err := net.ListenUDP("udp", addr)
if err != nil {
panic(err)
}
defer conn.Close()
// 使用channel通知worker继续接收
packets := make(chan *net.UDPAddr, 100)

// 启动多个worker
for i := 0; i < numWorkers; i++ {
    go worker(i, conn, packets)
}

fmt.Println("UDP服务器正在运行在 :8080...")

// 持续发送事件到任意worker（轮询方式）
for {
    packets <- nil // 触发任一worker去调用ReadFromUDP
    time.Sleep(time.Microsecond) // 避免忙等
}
}

package main
import (
"fmt"
"net"
"sync"
"time"
)
func sendMsg(id int, serverAddr net.UDPAddr, wg sync.WaitGroup) {
defer wg.Done()
conn, err := net.DialUDP("udp", nil, serverAddr)
if err != nil {
    fmt.Printf("客户端 %d 连接失败: %v\n", id, err)
    return
}
defer conn.Close()

for i := 0; i < 5; i++ {
    msg := fmt.Sprintf("客户端-%d 发送消息-%d", id, i+1)
    conn.Write([]byte(msg))

    // 接收响应
    buffer := make([]byte, 1024)
    n, err := conn.Read(buffer)
    if err != nil {
        fmt.Printf("客户端 %d 读取失败: %v\n", id, err)
    } else {
        fmt.Printf("客户端-%d 收到响应: %s\n", id, string(buffer[:n]))
    }

    time.Sleep(100 * time.Millisecond)
}
}
func main() {
serverAddr, _ := net.ResolveUDPAddr("udp", "127.0.0.1:8080")
var wg sync.WaitGroup
// 启动10个并发客户端
for i := 0; i < 10; i++ {
    wg.Add(1)
    go sendMsg(i+1, serverAddr, &wg)
    time.Sleep(10 * time.Millisecond) // 避免瞬间连接风暴
}

wg.Wait()
fmt.Println("所有客户端发送完成")
}

// 简化版服务端接收逻辑
for {
    buffer := make([]byte, 1024)
    n, addr, err := conn.ReadFromUDP(buffer)
    if err != nil {
        continue
    }
    data := make([]byte, n)
    copy(data, buffer[:n])
    go handlePacket(conn, data, addr) // 分发给worker处理
}