如何在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中实现起来不难，但细节处理到位才能保证稳定性和扩展性。只要掌握了基本结构和常见问题处理方式，就能轻松构建实时通信功能。
