Golang如何实现WebSocket客户端_Golang WebSocket Client通信实践-Golang-PHP中文网

Golang如何实现WebSocket客户端_Golang WebSocket Client通信实践

P粉602998670

发布： 2025-11-29 12:07:35

原创

326人浏览过

Go语言通过gorilla/websocket库可轻松实现WebSocket客户端，1. 安装github.com/gorilla/websocket；2. 使用websocket.Dial建立连接；3. 通过WriteMessage和ReadMessage收发消息；4. 用ticker定时发送Ping实现心跳；5. 捕获错误并实现带延迟的重连机制，确保连接稳定。

golang如何实现websocket客户端_golang websocket client通信实践

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 main
<p>import (
"fmt"
"log"
"net/http"
"github.com/gorilla/websocket"
)</p><p>var upgrader = websocket.Upgrader{} // 用于服务端，客户端不需要</p><p>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()</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">fmt.Printf("成功连接到服务器，状态码: %d\n", resp.StatusCode)

// 后续消息处理...

登录后复制

}

DefaultDialer 提供了默认配置，你也可以自定义超时、TLS设置等。

3. 发送与接收消息

连接建立后，就可以通过 WriteMessage 和 ReadMessage 收发数据。

发送文本消息：

Magic Write

Canva旗下AI文案生成器

查看详情

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
    }
<pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">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))
<p>// 启动心跳协程
go func() {
ticker := time.NewTicker(30 * time.Second)
defer ticker.Stop()</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">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 WebSocket Client通信实践的详细内容，更多请关注php中文网其它相关文章！