Golang UDP多线程接收与发送示例

Go中UDP通信通过net.ListenPacket结合多Goroutine实现高并发,服务端用channel协调多个worker接收数据,客户端并发发送消息,共享的*net.UDPConn支持安全的并发读写,无需额外锁机制。

golang udp多线程接收与发送示例

在Go语言中,UDP通信本身是并发安全的,通过net.ListenPacket创建一个UDP监听套接字后,可以利用Goroutine实现多线程接收与发送。虽然Go的Goroutine不是传统意义上的“线程”,但其轻量特性非常适合高并发网络编程。下面是一个简单的Golang UDP多Goroutine接收与发送示例。

1. UDP服务端:多Goroutine接收数据

服务端启动后,监听指定端口,并使用多个Goroutine并发处理接收到的数据包。

package main

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

const numWorkers = 3 // 启动3个接收协程

func handlePacket(conn net.UDPConn, data []byte, addr net.UDPAddr) {fmt.Printf("来自 %s 的消息: %sn", addr.String(), string(data))

// 模拟响应response := "收到: " + string(data)conn.WriteToUDP([]byte(response), addr)

}

立即学习“go语言免费学习笔记(深入)”;

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 读取错误: %vn", id, err)        continue    }    // 复制数据,避免后续读取覆盖    data := make([]byte, n)    copy(data, buffer[:n])    go handlePacket(conn, data, addr)}

}

立即学习“go语言免费学习笔记(深入)”;

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)// 启动多个workerfor i := 0; i < numWorkers; i++ {    go worker(i, conn, packets)}fmt.Println("UDP服务器正在运行在 :8080...")// 持续发送事件到任意worker(轮询方式)for {    packets <- nil // 触发任一worker去调用ReadFromUDP    time.Sleep(time.Microsecond) // 避免忙等}

}

立即学习“go语言免费学习笔记(深入)”;

2. UDP客户端:并发发送消息

客户端使用多个Goroutine向服务端发送UDP消息,模拟高并发场景。

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 连接失败: %vn", 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 读取失败: %vn", id, err)    } else {        fmt.Printf("客户端-%d 收到响应: %sn", id, string(buffer[:n]))    }    time.Sleep(100 * time.Millisecond)}

}

立即学习“go语言免费学习笔记(深入)”;

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("所有客户端发送完成")

}

立即学习“go语言免费学习笔记(深入)”;

3. 关键说明与注意事项

该设计展示了如何在Go中实现UDP的“多线程”通信模式,实际使用中需注意以下几点:

UDP无连接:每个WriteToUDP都需要指定目标地址,适合一对一或广播场景。Goroutine调度:多个Goroutine调用ReadFromUDP时,Go运行时会自动处理并发,但不能多个同时阻塞读同一个连接。共享Socket:服务端只有一个*net.UDPConn,多个worker通过channel协调读取,避免竞争。数据拷贝:在Goroutine中处理前应复制buffer内容,防止主缓冲区被覆盖。错误处理:UDP不保证送达,程序需容忍丢包、乱序等情况。

4. 简化单协程接收 + 多协程发送

更常见做法是服务端用单个循环接收,再分发给多个处理协程,结构更清晰:

// 简化版服务端接收逻辑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处理}

基本上就这些。Go的UDP编程结合Goroutine非常简洁高效,无需复杂锁机制即可实现高性能并发通信。关键是理解*net.UDPConn是线程安全的读写操作,合理分配任务即可。

以上就是Golang UDP多线程接收与发送示例的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1412753.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
Golang如何使用strconv进行类型转换
上一篇 2025年12月16日 06:36:46
数据库连接池性能调优实践
下一篇 2025年12月16日 06:37:02

相关推荐

发表回复

登录后才能评论
关注微信