Golang WebSocket消息队列处理示例-Golang-PHP中文网

Golang WebSocket消息队列处理示例

P粉602998670

发布： 2025-10-11 18:36:01

原创

886人浏览过

使用消息队列可避免阻塞WebSocket通信，提升系统可靠性；通过Go的channel或RabbitMQ/Kafka实现生产者-消费者模式，确保消息不丢失并支持异步处理。

golang websocket消息队列处理示例

用Go语言做WebSocket服务时，如果想高效处理大量消息，特别是需要异步处理、保证不丢消息或对接数据库、第三方API，结合消息队列是个很实用的做法。核心思路是把WebSocket接收到的消息先推到队列里，再由专门的消费者去处理，避免阻塞实时通信。

为什么用消息队列？

直接在WebSocket的读写循环里处理业务逻辑，比如存数据库、调外部接口，一旦耗时稍长，就可能拖慢整个连接，甚至导致超时断开。引入消息队列后：

生产者（WebSocket Handler）只负责快速接收消息并投递到队列
消费者（后台Worker）从队列取出消息，慢慢处理，不影响前端体验
即使处理服务暂时挂了，消息还在队列里，不会丢失

基础实现：WebSocket + 内存通道模拟队列

对于轻量级应用，可以用Go的channel作为简单的内存队列。下面是一个基于gorilla/websocket和channel的例子：

package main
<p>import (
"log"
"net/http"
"encoding/json"
"github.com/gorilla/websocket"
)</p><p>var upgrader = websocket.Upgrader{CheckOrigin: func(r <em>http.Request) bool { return true }}
var broadcast = make(chan []byte, 100) // 带缓冲的广播通道，相当于队列
var clients = make(map[</em>websocket.Conn]bool)</p><p>// 消费者：从队列读取并广播
func handleMessages() {
for {
msg := <-broadcast
for client := range clients {
err := client.WriteMessage(websocket.TextMessage, msg)
if err != nil {
client.Close()
delete(clients, client)
}
}
}
}</p><p>// 生产者：WebSocket连接处理
func wsHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("Upgrade error:", err)
return
}
defer conn.Close()</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">clients[conn] = true

for {
    _, msg, err := conn.ReadMessage()
    if err != nil {
        delete(clients, conn)
        break
    }
    // 收到消息后，快速扔进队列
    broadcast <- msg
}

登录后复制

}

立即学习“go语言免费学习笔记（深入）”；

Alkaid.art

专门为Phtoshop打造的AIGC绘画插件

153

查看详情

func main() { go handleMessages() // 启动消费者 http.HandleFunc("/ws", wsHandler) log.Println("Server on :8080") log.Fatal(http.ListenAndServe(":8080", nil)) }