Golang结合gorilla/websocket可高效实现WebSocket通信。首先通过go get安装库，然后创建升级HTTP连接的服务器，处理消息收发；接着用map和channel管理客户端连接并广播消息；前端通过JavaScript建立连接测试；最后建议添加心跳机制与错误监控以提升稳定性。
在构建现代 Web 应用时，实时通信是一个常见需求。Golang 凭借其高并发和轻量级 Goroutine 的优势，非常适合用于实现 WebSocket 服务端。通过 gorilla/websocket 这个广泛使用的第三方库，可以快速搭建稳定高效的 WebSocket 实时通信系统。
1. 安装 gorilla/websocket 库
在项目中使用 WebSocket 前，需要先引入 gorilla 提供的 websocket 包：
go get github.com/gorilla/websocket该库封装了 WebSocket 协议的底层细节，提供简洁的 API 来处理连接、读取消息和发送数据。
2. 创建 WebSocket 服务端
一个基础的 WebSocket 服务端包括 HTTP 升级处理和消息收发逻辑。以下是一个简单的示例：
立即学习“go语言免费学习笔记（深入）”；
package mainimport (
"log"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
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 {
log.Println("升级失败:", err)
return
}
defer conn.Close()
for {
mt, message, err := conn.ReadMessage()
if err != nil {
break
}
log.Printf("收到: %s", message)
// 回显消息
conn.WriteMessage(mt, []byte("echo: "+string(message)))
}
}
func main() {
http.HandleFunc("/ws", handleWebSocket)
log.Println("服务启动在 :8080")
log.Fatal(http.ListenAndServe(":8080", nil))
}
上述代码启动了一个监听 /ws 路径的 WebSocket 服务，接受客户端连接并回显收到的消息。
3. 管理多个连接与广播消息
实际应用中通常需要管理多个客户端连接，并支持广播消息。可以通过维护一个连接池来实现：
var clients = make(map[*websocket.Conn]bool)var broadcast = make(chan []byte)
func handleWebSocket(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}
defer conn.Close()
clients[conn] = true
go func() {
for {
_, message, err := conn.ReadMessage()
if err != nil {
delete(clients, conn)
break
}
broadcast
}
}()
}
// 广播协程
func broadcastMessages() {
for {
msg :=
for client := range clients {
err := client.WriteMessage(websocket.TextMessage, msg)
if err != nil {
client.Close()
delete(clients, client)
}
}
}
}
在 main 函数中启动广播协程：
go broadcastMessages()
这样所有在线客户端都能接收到广播消息。
4. 前端连接测试
使用浏览器 JavaScript 测试连接：
const ws = new WebSocket("ws://localhost:8080/ws");ws.onopen = () => {
ws.send("Hello Go WebSocket!");
};
ws.onmessage = (event) => {
console.log("收到:", event.data);
};
确保服务运行后打开页面或使用工具（如 wscat）测试通信是否正常。
基本上就这些。Golang 配合 gorilla/websocket 可以高效实现 WebSocket 实时通信，适用于聊天室、通知推送、协同编辑等场景。关键是合理管理连接生命周期和并发安全。不复杂但容易忽略心跳机制和错误处理，建议在生产环境中加入 ping/pong 心跳检测和日志监控。
