Go语言通过gorilla/websocket库实现WebSocket通信，首先使用go get安装依赖，然后创建Upgrader实例将HTTP连接升级为WebSocket，示例代码展示了服务端接收并回显消息的过程，客户端可用JavaScript测试连接，关键点包括允许跨域、读写消息及连接关闭，适用于实时通信场景。
Go语言实现WebSocket通信主要依赖第三方库 gorilla/websocket，这是目前最常用且稳定的选择。下面介绍如何用Golang搭建一个简单的WebSocket服务端，完成数据的接收与发送。
1. 安装 gorilla/websocket
在项目中引入依赖：
go get github.com/gorilla/websocket
2. 创建WebSocket服务端
使用 gorilla/websocket 提供的 Upgrader 将HTTP连接升级为WebSocket连接。
示例代码：
立即学习“go语言免费学习笔记（深入）”；
package mainimport ( "log" "net/http" "github.com/gorilla/websocket")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.Printf("升级失败: %v", err) return } defer conn.Close() // 循环读取消息 for { msgType, message, err := conn.ReadMessage() if err != nil { log.Printf("读取失败: %v", err) break } log.Printf("收到消息: %s", message) // 回显消息 if err := conn.WriteMessage(msgType, message); err != nil { log.Printf("发送失败: %v", err) break } }}func main() { http.HandleFunc("/ws", wsHandler) log.Println("服务器启动，地址: :8080") log.Fatal(http.ListenAndServe(":8080", nil))}
3. 客户端测试（JavaScript）
可以用浏览器控制台测试连接：
const ws = new WebSocket("ws://localhost:8080/ws");ws.onopen = () => { ws.send("你好，Golang WebSocket！");};ws.onmessage = (event) => { console.log("收到:", event.data);};ws.onclose = () => { console.log("连接关闭");};
4. 关键点说明
Upgrader.CheckOrigin：默认拒绝跨域请求，开发时可设为true，上线前应限制合法来源。 ReadMessage()：阻塞读取客户端消息，返回消息类型（text/binary）和内容。 WriteMessage()：向客户端发送数据，保持与读取相同的消息类型更稳妥。 defer conn.Close()：确保连接关闭，防止资源泄漏。
基本上就这些。通过 gorilla/websocket 可以轻松实现全双工通信，适合实时聊天、通知推送等场景。不复杂但容易忽略错误处理和连接管理，建议结合 context 或 hub 模式做更完善的控制。
以上就是Golang如何实现WebSocket数据收发的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1414054.html
微信扫一扫 
支付宝扫一扫 
