Golang服务端推送消息首选WebSocket实现双向实时通信,需用gorilla/websocket库升级HTTP连接、构建Hub管理连接及goroutine收发消息;SSE则适用于单向推送,通过设置响应头、按data:格式推送并维护客户端channel实现。

在Golang中实现服务端推送消息,核心是建立一种服务器能主动向客户端发送数据的通信机制。HTTP协议本身是请求-响应模式,无法满足实时性要求,因此需要采用长连接技术。目前最主流和高效的方法是使用WebSocket,它支持全双工通信,允许服务端随时推送消息。Server-Sent Events(SSE)是另一种轻量级方案,适用于只需要服务端单向推送的场景。
使用WebSocket进行双向实时推送
WebSocket是实现实时应用的首选技术,特别适合聊天室、在线游戏或需要双向交互的系统。
-
引入依赖库:Go标准库不包含WebSocket支持,推荐使用成熟的第三方库github.com/gorilla/websocket。通过
go get github.com/gorilla/websocket安装。 -
升级HTTP连接:服务端监听一个特定路径(如/ws),当收到客户端的Upgrade请求时,使用Upgrader.Upgrade()方法将普通的HTTP连接转换为持久的WebSocket连接。记得设置
CheckOrigin: func(r *http.Request) bool { return true }以处理跨域问题(生产环境需更严格校验)。 - 管理客户端连接:为避免并发读写map导致的panic,设计一个中心化的“Hub”或管理器。它通常包含一个存储所有活跃连接的map、一个用于注册/注销连接的通道(chan)、以及一个用于广播消息的通道。通过一个for-select循环在主线程中安全地处理这些操作。
-
处理消息收发:每个新连接都启动两个独立的goroutine。一个负责readPump,持续调用
conn.ReadMessage()监听客户端发来的消息,并将其转发到广播通道。另一个负责writePump,从广播通道接收消息,然后调用conn.WriteMessage()推送给对应的客户端。
使用SSE实现服务端单向消息推送
如果应用场景只需要服务端向客户端推送通知或状态更新(如股票行情、新闻推送),SSE是一个更简单优雅的选择,它基于HTTP协议,天然兼容性好。
-
建立SSE连接:客户端通过JavaScript的EventSource API发起请求。服务端接收到请求后,首先设置关键的响应头:
Content-Type: text/event-stream、Cache-Control: no-cache和Connection: keep-alive,以保持长连接并禁用缓存。 -
推送数据格式:服务端推送的数据必须遵循事件流格式。每次发送数据时,以
data:开头,后跟实际内容,最后以\n\n结束。例如:fmt.Fprintf(c.Writer, "data: %s\n\n", message)。发送后立即调用c.Writer.Flush()强制将数据刷到客户端,而不是等待缓冲区满。 -
管理客户端列表:与WebSocket类似,需要一个全局的map来存储所有活跃的SSE客户端。但由于SSE是单向的,客户端列表中的元素通常是带缓冲的channel(如
chan string)。当有新消息要广播时,遍历这个map,将消息发送到每个客户端的channel中。同时,在处理SSE路由的函数里,通过for msg := range clientChan监听该channel,并将收到的消息按SSE格式写回响应。 - 触发推送:可以提供一个独立的HTTP接口(如POST /send),接收要推送的内容,然后由服务端逻辑将此内容发送到所有客户端的channel中,从而触发对所有连接的广播。










