使用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 mainimport ("log""net/http""github.com/gorilla/websocket")
var upgrader = websocket.Upgrader{CheckOrigin: func(r *http.Request) bool {return true // 允许跨域连接，生产环境应限制具体来源},}
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()
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页面测试连接：
const ws = new WebSocket("ws://localhost:8080/ws");ws.onopen = function() {console.log("连接建立");ws.send("Hello, Server!");};
ws.onmessage = function(event) {console.log("收到:", event.data);};
打开浏览器访问该页面，可在控制台看到来回通信的日志。
4. 实现客户端（可选）
Go也可以作为WebSocket客户端：
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 client")), msg, := conn.ReadMessage()log.Printf("响应: %s", msg)
这种方式适合微服务间通信或自动化测试。
5. 注意事项与优化建议
每个连接使用独立的goroutine处理，避免阻塞其他客户端。设置读写超时防止连接长时间挂起：
conn.SetReadDeadline(time.Now().Add(60 * time.Second))conn.SetWriteDeadline(time.Now().Add(10 * time.Second))
管理活跃连接可用全局map+互斥锁，或结合Redis做集群状态同步。对高频消息场景，考虑使用缓冲通道进行异步写入，避免WriteMessage阻塞。
基本上就这些。用 gorilla/websocket 能快速搭建可靠的WebSocket服务，结构清晰，易于扩展。
以上就是如何在Golang中实现WebSocket通信的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1411333.html
微信扫一扫 
支付宝扫一扫 
