实现 websocket 服务的关键在于使用 gorilla/websocket 库完成连接升级与消息处理。具体步骤如下：1. 安装依赖 go get github.com/gorilla/websocket；2. 编写服务端代码，定义 upgrader 实例并设置 /ws 路由，通过 handlewebsocket 函数处理连接、读取消息与回写；3. 使用 wscat 或浏览器测试服务运行状态；4. 常见优化包括结构化消息格式、并发连接管理及性能调优如缓冲区设置与心跳机制。整个流程清晰且易于扩展。
实现一个 Golang 的 WebSocket 服务其实并不难，尤其是借助 gorilla/websocket 这个社区广泛使用的库。它封装了底层的握手、消息处理等细节，让开发者可以快速搭建 WebSocket 服务。
下面通过一个简单的实战案例，带你一步步搭建一个基础但完整的 WebSocket 服务。
准备工作：安装依赖
在开始之前，确保你已经安装了 Go，并且创建了一个项目目录。然后使用以下命令安装 gorilla/websocket：
立即学习“go语言免费学习笔记（深入）”；
go get github.com/gorilla/websocket
这个包会自动加入到你的 go.mod 文件中。
编写 WebSocket 服务端代码
我们先从最核心的部分入手：如何建立一个 WebSocket 服务器。
package main
import (
"fmt"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
// 允许跨域请求（开发阶段可开启）
CheckOrigin: func(r *http.Request) bool {
return true
},
}
func handleWebSocket(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
fmt.Println("升级失败:", err)
return
}
fmt.Println("客户端已连接")
for {
messageType, p, err := conn.ReadMessage()
if err != nil {
fmt.Println("读取消息失败:", err)
break
}
fmt.Printf("收到消息: %s\n", p)
// 回传给客户端
if err := conn.WriteMessage(messageType, p); err != nil {
fmt.Println("发送消息失败:", err)
break
}
}
}
func main() {
http.HandleFunc("/ws", handleWebSocket)
fmt.Println("WebSocket 服务启动于 ws://localhost:8080/ws")
err := http.ListenAndServe(":8080", nil)
if err != nil {
panic(err)
}
}这段代码做了几件事：
- 定义了一个 Upgrader 实例，用于将 HTTP 请求升级为 WebSocket 连接。
- 设置了
/ws路由来处理 WebSocket 请求。 - 在
handleWebSocket中循环读取和回写消息。
测试 WebSocket 是否正常运行
你可以使用浏览器控制台或者工具如 wscat 来测试服务是否正常。
例如，用 wscat 测试：
npx wscat -c ws://localhost:8080/ws
连接成功后，输入任意文字，服务端会原样返回。
常见问题与优化建议
消息类型处理不够灵活
默认情况下，上面的例子只是简单地把收到的消息再发回去。实际场景中，可能需要根据不同的消息类型做不同处理。
可以这样做：
- 使用 JSON 格式传输结构化数据
- 添加字段区分消息种类（如 chat、ping、auth 等）
示例结构体定义：
type Message struct {
Type string `json:"type"`
Body string `json:"body"`
}然后在读取时解析 JSON：
var msg Message
if err := conn.ReadJSON(&msg); err != nil {
// 错误处理
}并发连接管理
如果要支持多个客户端通信，比如聊天室功能，就需要一个连接池来保存所有活跃的连接。
可以这样设计：
- 定义全局变量存储连接列表
- 每次新连接加入列表
- 每次断开连接移除列表
- 收到消息后广播给所有连接
注意并发安全，可以用 sync.Mutex 或者使用 channel 来协调。
性能与错误处理
- 设置合理的缓冲区大小（Read/WriteBufferSize）
- 处理连接超时、心跳机制（可用 gorilla/websocket 提供的 Ping/Pong 方法）
- 对写操作加锁，避免并发写冲突
基本上就这些。只要掌握了基本流程，剩下的就是根据业务逻辑扩展功能。WebSocket 本身不复杂，但容易忽略的是连接管理和错误处理部分。
