Golang实现WebSocket实时通信教程

Golang结合gorilla/websocket可高效实现WebSocket通信。首先通过go get安装库，然后创建升级HTTP连接的服务器，处理消息收发；接着用map和channel管理客户端连接并广播消息；前端通过JavaScript建立连接测试；最后建议添加心跳机制与错误监控以提升稳定性。

在构建现代 Web 应用时，实时通信是一个常见需求。Golang 凭借其高并发和轻量级 Goroutine 的优势，非常适合用于实现 WebSocket 服务端。通过 gorilla/websocket 这个广泛使用的第三方库，可以快速搭建稳定高效的 WebSocket 实时通信系统。

1. 安装 gorilla/websocket 库

在项目中使用 WebSocket 前，需要先引入 gorilla 提供的 websocket 包：

该库封装了 WebSocket 协议的底层细节，提供简洁的 API 来处理连接、读取消息和发送数据。

2. 创建 WebSocket 服务端

一个基础的 WebSocket 服务端包括 HTTP 升级处理和消息收发逻辑。以下是一个简单的示例：

func main() {
  http.HandleFunc("/ws", handleWebSocket)
  log.Println("服务启动在 :8080")
  log.Fatal(http.ListenAndServe(":8080", nil))
}

上述代码启动了一个监听 /ws 路径的 WebSocket 服务，接受客户端连接并回显收到的消息。

3. 管理多个连接与广播消息

实际应用中通常需要管理多个客户端连接，并支持广播消息。可以通过维护一个连接池来实现：

// 广播协程 func broadcastMessages() {
  for {
    msg := <-broadcast
    for client := range clients {
      err := client.WriteMessage(websocket.TextMessage, msg)
      if err != nil {
        client.Close()
        delete(clients, client)
      }
    }
  }
}

在 main 函数中启动广播协程：
go broadcastMessages()
这样所有在线客户端都能接收到广播消息。

4. 前端连接测试

使用浏览器 JavaScript 测试连接：

确保服务运行后打开页面或使用工具（如 wscat）测试通信是否正常。

基本上就这些。Golang 配合 gorilla/websocket 可以高效实现 WebSocket 实时通信，适用于聊天室、通知推送、协同编辑等场景。关键是合理管理连接生命周期和并发安全。不复杂但容易忽略心跳机制和错误处理，建议在生产环境中加入 ping/pong 心跳检测和日志监控。