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

用Go语言做WebSocket服务时,如果想高效处理大量消息,特别是需要异步处理、保证不丢消息或对接数据库、第三方API,结合消息队列是个很实用的做法。核心思路是把WebSocket接收到的消息先推到队列里,再由专门的消费者去处理,避免阻塞实时通信。
为什么用消息队列?
直接在WebSocket的读写循环里处理业务逻辑,比如存数据库、调外部接口,一旦耗时稍长,就可能拖慢整个连接,甚至导致超时断开。引入消息队列后:
- 生产者(WebSocket Handler)只负责快速接收消息并投递到队列
- 消费者(后台Worker)从队列取出消息,慢慢处理,不影响前端体验
- 即使处理服务暂时挂了,消息还在队列里,不会丢失
基础实现:WebSocket + 内存通道模拟队列
对于轻量级应用,可以用Go的channel作为简单的内存队列。下面是一个基于gorilla/websocket和channel的例子:
package mainimport ( "log" "net/http" "encoding/json" "github.com/gorilla/websocket" )
var upgrader = websocket.Upgrader{CheckOrigin: func(r http.Request) bool { return true }} var broadcast = make(chan []byte, 100) // 带缓冲的广播通道,相当于队列 var clients = make(map[websocket.Conn]bool)
// 消费者:从队列读取并广播 func handleMessages() { for { msg := <-broadcast for client := range clients { err := client.WriteMessage(websocket.TextMessage, msg) if err != nil { client.Close() delete(clients, client) } } } }
// 生产者: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()
clients[conn] = true for { _, msg, err := conn.ReadMessage() if err != nil { delete(clients, conn) break } // 收到消息后,快速扔进队列 broadcast <- msg }}
立即学习“go语言免费学习笔记(深入)”;
func main() { go handleMessages() // 启动消费者 http.HandleFunc("/ws", wsHandler) log.Println("Server on :8080") log.Fatal(http.ListenAndServe(":8080", nil)) }
进阶方案:集成RabbitMQ/Kafka
生产环境推荐使用RabbitMQ、Kafka这类专业消息队列。以RabbitMQ为例,你需要:
- 安装amqp库:go get github.com/rabbitmq/amqp091-go
- 在wsHandler里,把broadcast 换成向RabbitMQ发送消息
- 另起一个或多个worker进程,监听RabbitMQ队列,执行实际业务逻辑
这样系统更稳定,支持多实例水平扩展,也方便做消息持久化、重试等。
基本上就这些,关键是把“接收”和“处理”拆开,让WebSocket保持轻快。










