Golang WebSocket通信如何实现实时双向数据传输教程-Golang-PHP中文网

Golang WebSocket通信如何实现实时双向数据传输教程

P粉602998670

发布： 2025-07-07 12:43:02

原创

659人浏览过

如何在golang中实现websocket通信？使用标准库net/http配合gorilla/websocket包可快速搭建高性能websocket服务端和客户端。1. 建立连接需定义upgrader配置并通过upgrade方法升级http连接；2. 收发消息通过readmessage和writemessage方法实现，建议分别在goroutine中处理；3. 保持连接稳定可通过添加心跳机制、设置超时时间和自动重连机制实现；4. 客户端连接使用websocket.dial发起请求，并同样处理收发消息逻辑。

Golang WebSocket通信如何实现实时双向数据传输教程

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

以上就是Golang WebSocket通信如何实现实时双向数据传输教程的详细内容，更多请关注php中文网其它相关文章！