如何在golang中实现websocket通信？使用标准库net/http配合gorilla/websocket包可快速搭建高性能websocket服务端和客户端。1. 建立连接需定义upgrader配置并通过upgrade方法升级http连接；2. 收发消息通过readmessage和writemessage方法实现，建议分别在goroutine中处理；3. 保持连接稳定可通过添加心跳机制、设置超时时间和自动重连机制实现；4. 客户端连接使用websocket.dial发起请求，并同样处理收发消息逻辑。
在golang中实现WebSocket通信，实现实时双向数据传输其实并不复杂。标准库net/http配合第三方库比如gorilla/websocket可以非常方便地搭建一个高性能的WebSocket服务端和客户端。本文会从几个关键点入手，让你快速掌握如何用Go构建WebSocket应用。
如何建立WebSocket连接？
WebSocket是一种基于TCP协议的全双工通信协议，它通过HTTP握手升级为WebSocket连接。在Go中，最常用的是gorilla/websocket包，它封装了握手、消息读写等流程。
要建立一个WebSocket服务器，首先需要一个HTTP处理函数：
立即学习“go语言免费学习笔记（深入）”；
var upgrader = websocket.Upgrader{ ReadBufferSize: 1024, WriteBufferSize: 1024, CheckOrigin: func(r *http.Request) bool { return true // 允许跨域，生产环境建议配置具体域名 }, } func wsHandler(w http.ResponseWriter, r *http.Request) { conn, _ := upgrader.Upgrade(w, r, nil) // conn 是 *websocket.Conn 类型，可用于后续通信 }
然后注册路由并启动服务：
http.HandleFunc("/ws", wsHandler) log.Fatal(http.ListenAndServe(":8080", nil))
这样就完成了基础的WebSocket连接建立。
如何收发消息？
一旦连接建立完成，就可以使用conn.ReadMessage()和conn.WriteMessage()进行消息的收发。这两个方法都是阻塞式的，适合放在goroutine中运行。
接收消息
接收消息通常在一个循环中持续监听：
for { _, message, err := conn.ReadMessage() if err != nil { log.Println("read error:", err) break } fmt.Println("收到消息：", string(message)) }
发送消息
发送消息也很简单，只需要调用WriteMessage，并指定消息类型（文本或二进制）：
err := conn.WriteMessage(websocket.TextMessage, []byte("Hello Client!")) if err != nil { log.Println("write error:", err) }
小技巧：如果希望同时支持收发，可以用两个goroutine分别处理读写逻辑，避免阻塞。
如何保持连接稳定？
WebSocket连接容易因为网络波动、心跳缺失等问题断开。为了提高稳定性，有几点建议：
- 添加心跳机制：定期发送ping/pong消息维持连接活跃状态。
- 设置超时时间：可以通过SetReadDeadline来设置读取超时，防止长时间无响应导致程序卡住。
- 自动重连机制（客户端）：如果是客户端连接，建议在网络异常后尝试重新连接。
示例代码（服务端心跳）：
conn.SetReadLimit(512) conn.SetReadDeadline(time.Now().Add(60 * time.Second)) conn.SetPongHandler(func(appData string) error { log.Println("收到pong") return nil }) ticker := time.NewTicker(30 * time.Second) defer ticker.Stop() go func() { for { select { case <-ticker.C: err := conn.WriteControl(websocket.PingMessage, []byte{}, time.Now().Add(10*time.Second)) if err != nil { log.Println("ping error:", err) conn.Close() return } } } }()
客户端怎么连接？
除了服务端，你也可以用Go写一个WebSocket客户端用于测试或者集成到其他系统中。
使用websocket.Dial即可发起连接：
url := "ws://localhost:8080/ws" conn, _, err := websocket.DefaultDialer.Dial(url, nil) if err != nil { log.Fatal("dial error:", err) } // 收消息 go func() { for { _, msg, err := conn.ReadMessage() if err != nil { log.Println("client read error:", err) return } fmt.Println("客户端收到：", string(msg)) } }() // 发消息 conn.WriteMessage(websocket.TextMessage, []byte("Hello Server!"))
这种方式非常适合做本地调试、微服务间通信或自动化测试。
基本上就这些。WebSocket在Go中实现起来不难，但细节处理到位才能保证稳定性和扩展性。只要掌握了基本结构和常见问题处理方式，就能轻松构建实时通信功能。
