首先使用 gorilla/websocket 库实现 WebSocket 服务，定义 Upgrader 并处理连接请求，在 wsHandler 中读取和回显消息，最后通过 http.ListenAndServe 启动服务，并用 JavaScript 或 Go 客户端测试通信。
WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，相比传统的 HTTP 轮询，它能显著降低延迟并提升实时性。Golang 凭借其高并发和轻量级 Goroutine 的优势，非常适合构建高性能的 WebSocket 服务。本文将详细介绍如何使用 Go 实现 WebSocket 通信，并提供完整可运行的实践示例。
使用 gorilla/websocket 库实现 WebSocket 服务
Go 标准库不直接支持 WebSocket，但社区广泛使用的 gorilla/websocket 库提供了完整的实现。首先安装该库：
go get github.com/gorilla/websocket接下来创建一个基础的 WebSocket 服务器：
定义升级器（Upgrader），用于将 HTTP 连接升级为 WebSocket 连接：
立即学习“go语言免费学习笔记（深入）”；
var upgrader = websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }, // 允许跨域
}
处理连接请求：
func wsHandler(w http.ResponseWriter, r *http.Request) {conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("Upgrade error:", err)
return
}
defer conn.Close()
for {
_, msg, err := conn.ReadMessage()
if err != nil {
log.Println("Read error:", err)
break
}
log.Printf("Received: %s", msg)
err = conn.WriteMessage(websocket.TextMessage, msg)
if err != nil {
log.Println("Write error:", err)
break
}
}
}
启动 HTTP 服务：
func main() {http.HandleFunc("/ws", wsHandler)
log.Println("Server started on :8080")
log.Fatal(http.ListenAndServe(":8080", nil))
}
客户端连接与消息收发
使用 JavaScript 浏览器客户端测试服务：
你也可以用 Go 编写客户端：
func main() {
conn, _, err := websocket.DefaultDialer.Dial("ws://localhost:8080/ws", nil)
if err != nil {
log.Fatal(err)
}
defer conn.Close()
conn.WriteMessage(websocket.TextMessage, []byte("Hi from Go client"))
_, msg, _ := conn.ReadMessage()
fmt.Printf("Response: %s\n", msg)
}
实现广播机制与连接管理
实际项目中通常需要管理多个连接并支持广播。可以使用 map 和互斥锁来保存客户端连接：
var clients = make(map[*websocket.Conn]bool)var broadcast = make(chan []byte)
var mutex = sync.Mutex{}
在处理函数中注册连接：
mutex.Lock()clients[conn] = true
mutex.Unlock()
启动广播监听协程：
go func() {for {
msg := mutex.Lock()
for client := range clients {
err := client.WriteMessage(websocket.TextMessage, msg)
if err != nil {
client.Close()
delete(clients, client)
}
}
mutex.Unlock()
}
}()
收到消息后推送到广播通道：
broadcast心跳检测与连接健康维护
长时间空闲可能导致连接中断。通过 Ping/Pong 机制维持活跃：
conn.SetReadDeadline(time.Now().Add(60 * time.Second))conn.SetPongHandler(func(string) error {
conn.SetReadDeadline(time.Now().Add(60 * time.Second))
return nil
})
服务端定期发送 Ping 消息：
go func() {ticker := time.NewTicker(54 * time.Second)
defer ticker.Stop()
for {
select {
case if err := conn.WriteMessage(websocket.PingMessage, nil); err != nil {
return
}
}
}
}()
基本上就这些。通过 gorilla/websocket，Go 能轻松实现稳定高效的 WebSocket 服务。关键在于连接管理、并发安全和心跳机制的设计。实际部署时建议结合 TLS 使用 wss 协议保障安全。不复杂但容易忽略细节，比如关闭连接和资源清理。
