使用gorilla/websocket库可实现Go语言中的WebSocket通信。1. 通过go get github.com/gorilla/websocket安装依赖；2. 创建服务端，利用upgrader.Upgrade将HTTP升级为WebSocket，通过ReadMessage和WriteMessage处理消息收发；3. 前端使用new WebSocket("ws://localhost:8080/ws")连接并测试通信；4. Go也可作为客户端，使用websocket.DefaultDialer.Dial发起连接；5. 注意设置读写超时、并发安全管理连接及异步写入优化。
在Golang中实现WebSocket通信，主要依赖第三方库 gorilla/websocket，这是目前最常用且稳定的选择。Go标准库本身不包含WebSocket支持，因此需要引入这个库来完成握手、消息读写等操作。
1. 安装 gorilla/websocket 库
使用 go mod 管理依赖，在项目根目录执行：
go get github.com/gorilla/websocket
这会自动下载并添加到 go.mod 文件中。
2. 创建 WebSocket 服务端
以下是一个简单的 WebSocket 服务器示例，支持客户端连接、接收消息并回显：
立即学习“go语言免费学习笔记（深入）”；
package main
<p>import (
"log"
"net/http"
"github.com/gorilla/websocket"
)</p><p>var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true // 允许跨域连接，生产环境应限制具体来源
},
}</p><p>func echoHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Print("升级失败:", err)
return
}
defer conn.Close()</p><pre class='brush:php;toolbar:false;'>for {
messageType, message, err := conn.ReadMessage()
if err != nil {
log.Print("读取消息失败:", err)
break
}
log.Printf("收到: %s", message)
if err := conn.WriteMessage(messageType, message); err != nil {
log.Print("发送消息失败:", err)
break
}
}}
func main() { http.HandleFunc("/ws", echoHandler) log.Println("服务启动在 :8080") log.Fatal(http.ListenAndServe(":8080", nil)) }
说明：
- upgrader.Upgrade 将HTTP请求升级为WebSocket连接。
- CheckOrigin 设置为true允许所有跨域请求，实际部署时建议验证Origin头。
- ReadMessage 阻塞等待客户端消息，返回消息类型和字节数据。
- WriteMessage 向客户端发送消息，保持类型一致（如文本或二进制）。
3. 编写前端测试页面
创建一个HTML页面测试连接：
<!DOCTYPE html>
<html>
<body>
<script>
const ws = new WebSocket("ws://localhost:8080/ws");
<p>ws.onopen = function() {
console.log("连接建立");
ws.send("Hello, Server!");
};</p><p>ws.onmessage = function(event) {
console.log("收到:", event.data);
};
</script>
</body>
</html></p>打开浏览器访问该页面，可在控制台看到来回通信的日志。
4. 实现客户端（可选）
Go也可以作为WebSocket客户端：
conn, _, err := websocket.DefaultDialer.Dial("ws://localhost:8080/ws", nil)
if err != nil {
log.Fatal("连接失败:", err)
}
defer conn.Close()
<p>conn.WriteMessage(websocket.TextMessage, []byte("Hi from client"))
<em>, msg, </em> := conn.ReadMessage()
log.Printf("响应: %s", msg)</p>这种方式适合微服务间通信或自动化测试。
5. 注意事项与优化建议
- 每个连接使用独立的goroutine处理，避免阻塞其他客户端。
- 设置读写超时防止连接长时间挂起：
conn.SetReadDeadline(time.Now().Add(60 * time.Second)) conn.SetWriteDeadline(time.Now().Add(10 * time.Second))
- 管理活跃连接可用全局map+互斥锁，或结合Redis做集群状态同步。
- 对高频消息场景，考虑使用缓冲通道进行异步写入，避免WriteMessage阻塞。
基本上就这些。用 gorilla/websocket 能快速搭建可靠的WebSocket服务，结构清晰，易于扩展。
