Golang实现WebSocket通信教程

本文详细介绍了如何使用Go语言和gorilla/websocket库构建一个健壮、实用的WebSocket客户端，涵盖从依赖安装、连接建立、双向消息收发，到关键的心跳保活机制与智能错误重连策略的完整实现流程，代码简洁清晰、注重生产环境可用性，让开发者轻松掌握在Go中高效、稳定地集成实时双向通信能力的核心要点。

Go语言实现WebSocket客户端其实并不复杂，标准库虽然没有直接支持WebSocket，但借助第三方库如 gorilla/websocket 可以轻松完成。本文带你一步步构建一个功能完整的WebSocket客户端，涵盖连接、收发消息、心跳机制和错误处理。

1. 安装依赖：使用 gorilla/websocket

在项目中引入 gorilla 的 WebSocket 包：

这是目前Go中最流行、最稳定的WebSocket库，服务端和客户端都适用。

2. 建立WebSocket连接

使用 websocket.Dial 连接到WebSocket服务器。以下是一个基础连接示例：

package main
import (
"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客户端，关键在于连接管理、消息循环和异常恢复。实际项目中可将其封装成结构体，加入上下文控制和并发安全处理。不复杂但容易忽略细节。

本篇关于《Golang实现WebSocket通信教程》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！