Go语言通过gorilla/websocket库可轻松实现WebSocket客户端，1. 安装github.com/gorilla/websocket；2. 使用websocket.Dial建立连接；3. 通过WriteMessage和ReadMessage收发消息；4. 用ticker定时发送Ping实现心跳；5. 捕获错误并实现带延迟的重连机制，确保连接稳定。
Go语言实现WebSocket客户端其实并不复杂，标准库虽然没有直接支持WebSocket，但借助第三方库如 gorilla/websocket 可以轻松完成。本文带你一步步构建一个功能完整的WebSocket客户端，涵盖连接、收发消息、心跳机制和错误处理。
1. 安装依赖：使用 gorilla/websocket
在项目中引入 gorilla 的 WebSocket 包：
go get github.com/gorilla/websocket这是目前Go中最流行、最稳定的WebSocket库，服务端和客户端都适用。
2. 建立WebSocket连接
使用 websocket.Dial 连接到WebSocket服务器。以下是一个基础连接示例：
立即学习“go语言免费学习笔记（深入）”；
package mainimport ( "fmt" "log" "net/http" "github.com/gorilla/websocket" )
var upgrader = websocket.Upgrader{} // 用于服务端，客户端不需要
func main() { url := "ws://localhost:8080/ws" // 替换为你的WebSocket地址 conn, resp, err := websocket.DefaultDialer.Dial(url, http.Header{}) if err != nil { log.Fatalf("连接失败: %v", err) } defer conn.Close()
fmt.Printf("成功连接到服务器，状态码: %d\n", resp.StatusCode) // 后续消息处理...}
DefaultDialer 提供了默认配置，你也可以自定义超时、TLS设置等。
3. 发送与接收消息
连接建立后，就可以通过 WriteMessage 和 ReadMessage 收发数据。
发送文本消息：
err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, Server!")) if err != nil { log.Printf("发送消息失败: %v", err) }持续接收消息：
for { messageType, message, err := conn.ReadMessage() if err != nil { log.Printf("读取消息出错: %v", err) break }if messageType == websocket.TextMessage { fmt.Printf("收到: %s\n", string(message)) }}
4. 添加心跳与Ping/Pong处理
长时间连接容易断开，需要实现心跳机制。gorilla/websocket 支持自动响应 Pong，你只需定期发送 Ping。
设置连接的读写deadline，或使用 goroutine 发送Ping：
// 设置读取消息时的超时（可选） conn.SetReadDeadline(time.Now().Add(60 * time.Second))// 启动心跳协程 go func() { ticker := time.NewTicker(30 * time.Second) defer ticker.Stop()
for range ticker.C { err := conn.WriteControl(websocket.PingMessage, []byte{}, time.Now().Add(5*time.Second)) if err != nil { log.Printf("发送Ping失败: %v", err) return } }}()
同时，建议在读取消息前设置读取截止时间，避免永久阻塞。
5. 错误处理与重连机制
网络不稳定时连接可能中断，应实现重连逻辑：
- 捕获 ReadMessage 或 WriteMessage 的错误
- 判断是否为网络错误（如 io.EOF）
- 断开后延迟重试，避免频繁连接
简单重连示例：
func connectWithRetry(url string) *websocket.Conn {
for {
conn, _, err := websocket.DefaultDialer.Dial(url, nil)
if err == nil {
return conn
}
log.Printf("连接失败，5秒后重试: %v", err)
time.Sleep(5 * time.Second)
}
}
基本上就这些。用 gorilla/websocket 写一个健壮的WebSocket客户端，关键在于连接管理、消息循环和异常恢复。实际项目中可将其封装成结构体，加入上下文控制和并发安全处理。不复杂但容易忽略细节。
