用Golang实现WebSocket服务需借助gorilla/websocket库,通过upgrader.Upgrade()升级HTTP连接,用goroutine读写消息,维护线程安全的连接池,妥善处理错误与心跳,确保连接生命周期管理健壮。
用 Golang 实现一个简单的 WebSocket 服务并不复杂,核心是借助
这个广泛使用的第三方库——它稳定、文档清晰、社区支持好,比
标准库
更直接易用。
安装依赖并初始化 WebSocket 连接处理
先通过 go mod 管理依赖:
然后定义 HTTP handler,把升级 WebSocket 请求的逻辑封装进去。关键点是:不能直接用
写响应,必须用
把连接“升格”为 WebSocket 连接。
常见写法示例:
立即学习
“
go语言免费学习笔记(深入)
”;
定义全局
,设置允许
跨域
(开发时常用):
在 handler 中调用
,成功后得到
实例,后续收发消息都靠它
接收和广播消息(基础通信逻辑)
每个连接对应一个 goroutine,持续读取消息;服务端可维护一个连接池(比如用
),配合互斥锁安全增删。
go语言参考手册 中文CHM版
Go 是一个开源的编程语言,它能让构造简单、可靠且高效的软件变得容易。本文给大家带来Go参考手册,需要的可以来下载! Go是从2007年末由Robert Griesemer, Rob Pike, Ken Thompson主持开发,后来还加入了Ian Lance Taylor, Russ Cox等人,并最终于2009年11月开源,在2012年早些时候发布了Go 1稳定版本。现在Go的开发已经是完全开放的,并且拥有一个活跃的社区。 Go 语言特色 简洁、快速、安全 并行、有趣、开源 内存管理、v数组安全、编译
下载
典型结构:
客户端发来消息 → 调用
获取
简单回传:直接
广播给所有人:遍历连接池,对每个 conn 调用
(注意加锁读 map,且需检查 conn 是否已关闭)
建议统一用
类型,避免二进制解析复杂度
处理连接生命周期与错误退出
WebSocket 连接不是永久的,网络抖动、页面关闭、心跳超时都会导致断连。不能忽略错误,否则 goroutine 泄漏。
关键做法:
和
都可能返回 error,最常见的是
或
,此时应主动
并从连接池中移除
给每个连接启动两个 goroutine:一个读,一个写(尤其广播场景下,避免写阻塞读)
可选加 ping/pong 心跳:设置
和
,定期调用
前端
简单测试连接(验证通路)
不用框架,纯 HTML + JS 就能快速验证服务是否跑通:
只要控制台打出 "Received: Hello from browser",说明握手成功、双向通信正常。后续可扩展 JSON 消息格式、用户标识、房间分组等。
基本上就这些。不复杂但容易忽略错误处理和并发安全——把连接管理、读写分离、异常清理做扎实,一个健壮的基础 WebSocket 服务就出来了。
gorilla/websocketgo mod init example.com/ws-servergo get github.com/gorilla/websockethttp.ResponseWriterupgrader.Upgrade()upgradervar upgrader = websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}upgrader.Upgrade(w, r, nil)*websocket.Connmap[*websocket.Conn]boolconn.ReadMessage()messageType, data []byteconn.WriteMessage(messageType, data)WriteMessageTextMessageReadMessageWriteMessagewebsocket.CloseMessageio.EOFconn.Close()conn.SetPingHandler()conn.SetPongHandler()conn.WriteMessage(websocket.PingMessage, nil)