Golang WebSocket实时通信实现教程
首先通过gorilla/websocket库升级HTTP连接为WebSocket，再用map和互斥锁管理客户端连接，最后通过广播机制实现消息实时推送。
首先通过gorilla/websocket库升级HTTP连接为WebSocket，再用map和互斥锁管理客户端连接，最后通过广播机制实现消息实时推送。
WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，非常适合实现实时数据交互。在 Go 语言中，可以通过 gorilla/websocket 库快速实现 WebSocket 实时通信功能。以下是基于 Golang 实现 WebSocket 通信的核心步骤和代码示例。
建立 WebSocket 服务端
使用 gorilla/websocket 包可以轻松创建 WebSocket 服务。首先需要安装依赖：
go get github.com/gorilla/websocket然后编写一个简单的 WebSocket 服务端，用于接收和广播消息：
初始化 WebSocket 升级器：HTTP 请求需要通过升级器（Upgrader）转换为 WebSocket 连接。
func (upgrader *websocket.Upgrader) 定义：upgrader := &websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool { return true }, // 允许跨域
}
处理客户端连接：将 HTTP 请求升级为 WebSocket，并启动读写协程。
func handleConnection(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()
// 启动读取消息循环
for {
_, msg, err := conn.ReadMessage()
if err != nil {
break
}
log.Printf("Received: %s", msg)
// 回显消息或广播给其他客户端
broadcastMessage(msg)
}
}
管理客户端连接与广播消息
为了实现多用户实时通信，需要维护所有活跃的连接，并支持消息广播。
使用全局 map 存储连接：可以用 map 保存所有 WebSocket 连接，配合互斥锁保证并发安全。
var clients = make(map[*websocket.Conn]bool)var broadcast = make(chan []byte)
var mutex = sync.Mutex{}
注册与注销连接：在建立和断开时更新 clients 集合。
// 注册mutex.Lock()
clients[conn] = true
mutex.Unlock()
// 注销（在读循环退出后）
mutex.Lock()
delete(clients, conn)
mutex.Unlock()
广播消息到所有客户端：启动一个独立的 goroutine 处理广播。
go func() {for msg := range broadcast {
mutex.Lock()
for conn := range clients {
err := conn.WriteMessage(websocket.TextMessage, msg)
if err != nil {
conn.Close()
delete(clients, conn)
}
}
mutex.Unlock()
}
}()
前端建立 WebSocket 连接
前端使用原生 WebSocket API 连接 Go 后端服务。
const socket = new WebSocket("ws://localhost:8080/ws");socket.onopen = function(event) {
console.log("Connected to server");
};
socket.onmessage = function(event) {
console.log("Received:", event.data);
};
// 发送消息
socket.send("Hello, Server!");
确保后端路由 /ws 能正确处理 WebSocket 升级请求。
启动 HTTP 服务并监听端口
将 WebSocket 处理函数挂载到指定路由，并启动服务。
func main() {http.HandleFunc("/ws", handleConnection)
log.Println("Server started on :8080")
err := http.ListenAndServe(":8080", nil)
if err != nil {
log.Fatal("ListenAndServe error:", err)
}
}
运行程序后，前端即可通过 ws://localhost:8080/ws 建立连接，实现双向实时通信。
基本上就这些。核心是升级 HTTP 到 WebSocket、管理连接状态、安全地广播消息。gorilla/websocket 简洁高效，适合构建聊天室、通知系统等实时应用。注意处理连接错误和并发访问，避免资源泄漏。不复杂但容易忽略细节。
火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力，依托字节自研豆包多模态大模型，凭借火山方舟一站式平台，提供模型精调、推理部署、智能应用搭建等全链路服务，结合高性能AI算力，助力各行业快速落地大模型数字化与智能化应用。
AirBuddy 3 是面向 macOS 的蓝牙外设增强工具，可集中查看 AirPods、Beats 与苹果设备电量，提供快速连接、播放控制、电量提醒、桌面小组件和多台 Mac 外设接力等功能。
AirServer macOS版可把 Mac 转换为无线投屏接收端，通过 AirPlay 或 Google Cast 接收手机、平板和电脑画面，并支持多设备同时连接、高清镜像及会话录制，适合家庭娱乐、课堂教学、会议演示和应用展示。
火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力，依托字节自研豆包多模态大模型，凭借火山方舟一站式平台，提供模型精调、推理部署、智能应用搭建等全链路服务，结合高性能AI算力，助力各行业快速落地大模型数字化与智能化应用。
PixTV 是一款在浏览器中使用的 AI 视觉内容创作平台，提供图片、视频、音频、文字生成、AI 分镜、局部重绘、提示词反推和无限画布等工具，适合短剧、漫剧、广告及社交媒体内容制作。
Autodesk Maya 是面向影视、电视、动画和游戏行业的专业三维制作软件，集多边形与 NURBS 建模、角色绑定、关键帧和动作捕捉动画、Bifrost 特效模拟、材质开发及 Arnold 渲染于一体，