Golang在WebSocket实时通信中的实践详解gorilla/websocket库的使用-Golang-PHP中文网

Golang在WebSocket实时通信中的实践详解gorilla/websocket库的使用

P粉602998670

发布： 2025-08-04 09:11:01

原创

636人浏览过

gorilla/websocket 是 golang 中用于实现 websocket 实时通信的常用库，适合构建聊天系统、实时通知等场景。1. 它基于 tcp 协议实现双向通信，相比 http 轮询更高效；2. 使用前需安装并导入包 github.com/gorilla/websocket；3. 服务端通过 upgrader.upgrade() 方法升级连接，客户端使用 websocket.dialer 拨号连接；4. 通过 writemessage 和 readmessage 实现消息收发，并建议在独立 goroutine 中处理读写；5. 心跳机制通过定时发送 ping/pong 消息维持连接状态；6. 注意跨域设置、并发控制、资源释放及错误处理等问题。掌握这些要点即可快速搭建高效的 websocket 应用。

Golang在WebSocket实时通信中的实践详解gorilla/websocket库的使用

在使用 Golang 实现 WebSocket 实时通信时，

gorilla/websocket

登录后复制

是一个非常常用且功能完善的库。它不仅性能稳定，而且 API 设计简洁清晰，适合构建聊天系统、实时通知、在线协作等场景。

Golang在WebSocket实时通信中的实践详解gorilla/websocket库的使用

1. WebSocket 基本原理与 Golang 支持

WebSocket 是一种基于 TCP 的协议，允许客户端和服务器之间建立持久连接，并进行双向通信。相比传统的 HTTP 轮询方式，WebSocket 显得更高效、响应更快。

Golang 标准库中并没有完整的 WebSocket 实现，但

gorilla/websocket

登录后复制

弥补了这一空白。它是 Gorilla Web Toolkit 的一部分，专为 Go 编写，兼容性强，支持标准的 WebSocket 协议。

立即学习“go语言免费学习笔记（深入）”；

要使用它，先安装：

go get github.com/gorilla/websocket

登录后复制

然后导入：

import "github.com/gorilla/websocket"

登录后复制

2. 使用 gorilla/websocket 建立连接

建立 WebSocket 连接主要分为两个部分：服务端升级连接、客户端拨号连接。

服务端升级连接

在服务端处理 WebSocket 请求的核心是调用

Upgrader.Upgrade()

登录后复制

方法将 HTTP 请求“升级”为 WebSocket 连接：

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 类型，可用于读写消息
}

登录后复制

客户端拨号连接

客户端使用

websocket.Dialer

登录后复制

来拨号连接服务器：

dialer := websocket.DefaultDialer
conn, _, err := dialer.Dial("ws://localhost:8080/ws", nil)
if err != nil {
    log.Fatal("Dial error:", err)
}

登录后复制

3. 消息收发机制与常见操作

WebSocket 连接建立后，就可以通过

Conn

登录后复制

对象进行消息的发送和接收。

Kotlin Android 中文开发帮助文档 PDF版

这本书并不是一本语言参考书，但它是一个Android开发者去学习Kotlin并且使用在自己项目中的一个工具。我会通过使用一些语言特性和有趣的工具和库来解决很多我们在日常生活当中都会遇到的典型问题。这本书是非常具有实践性的，所以我建议你在电脑面前跟着我的例子和代码实践。无论何时你都可以在有一些想法的时候深入到实践中去。这本书适合你吗？写这本书是为了帮助那些有兴趣使用Kotlin语言来进行开发的Android开发者。如果你符合下面这些情况，那这本书是适合你的：你有相关Android开发和Andro

查看详情

发送消息

使用

WriteMessage

登录后复制

方法发送消息，第一个参数是消息类型（如

websocket.TextMessage

登录后复制

或

websocket.BinaryMessage

登录后复制

）：

conn.WriteMessage(websocket.TextMessage, []byte("Hello Server"))

登录后复制

接收消息

使用

ReadMessage

登录后复制

阻塞等待消息到来：

_, message, err := conn.ReadMessage()
if err != nil {
    log.Println("Read error:", err)
    return
}
fmt.Printf("Received: %s\n", message)

登录后复制

实际开发中，通常会在一个循环中持续监听消息：

for {
    _, msg, err := conn.ReadMessage()
    if err != nil {
        break
    }
    fmt.Println("Got:", string(msg))
}

登录后复制

注意：建议为每个连接启动一个独立的 goroutine 处理读取或写入，避免阻塞主流程。

4. 心跳机制与连接保持

WebSocket 连接可能因为网络波动、超时等原因断开，因此需要心跳机制来维持连接状态。

实现方式一般是定时向对方发送 ping 消息，并在收到 pong 后确认连接正常。

设置心跳参数

可以通过

SetPingHandler

登录后复制

和

SetPongHandler

登录后复制

自定义处理逻辑：

conn.SetPingHandler(func(appData string) error {
    // 收到 ping 时可以回复 pong
    return conn.WriteControl(websocket.PongMessage, []byte(appData), time.Now().Add(time.Second))
})

登录后复制

主动发送 ping

也可以定期主动发送 ping 消息：

ticker := time.NewTicker(30 * time.Second)
go func() {
    for {
        select {
        case <-ticker.C:
            err := conn.WriteControl(websocket.PingMessage, []byte{}, time.Now().Add(time.Second))
            if err != nil {
                log.Println("Ping error:", err)
                return
            }
        }
    }
}()

登录后复制

这样可以在连接异常时及时发现并重连。