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

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对象进行消息的发送和接收。
这本书并不是一本语言参考书,但它是一个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
}
}
}
}()这样可以在连接异常时及时发现并重连。
5. 常见问题与注意事项
-
跨域问题:默认情况下浏览器会限制跨域请求,需在 Upgrader 中设置
CheckOrigin
。 -
并发读写:
*websocket.Conn
不是并发安全的,多个 goroutine 写入需要加锁或者使用通道统一管理。 - 连接复用:WebSocket 是长连接,注意资源释放,避免内存泄漏。
- 错误处理:每次读写都要检查返回的 error,尤其是在生产环境中。
基本上就这些。gorilla/websocket 的使用并不复杂,但在实际项目中需要注意连接管理、错误处理和性能优化等方面。只要掌握基本结构和常见模式,就能快速搭建起高效的 WebSocket 应用。









