跳转到主内容
新派编程网 - 新潮编程理念,全新技术课堂!

如何使用Golang实现简单的WebSocket Server_GolangWebSocket Server通信说明

用Golang实现WebSocket服务需借助gorilla/websocket库,通过upgrader.Upgrade()升级HTTP连接,用goroutine读写消息,维护线程安全的连接池,妥善处理错误与心跳,确保连接生命周期管理健壮。 用 Golang 实现一个简单的 WebSocket 服务并不复杂,核心是借助
gorilla/websocket
这个广泛使用的第三方库——它稳定、文档清晰、社区支持好,比 标准库 更直接易用。 安装依赖并初始化 WebSocket 连接处理 先通过 go mod 管理依赖:
go mod init example.com/ws-server
go get github.com/gorilla/websocket
然后定义 HTTP handler,把升级 WebSocket 请求的逻辑封装进去。关键点是:不能直接用
http.ResponseWriter
写响应,必须用
upgrader.Upgrade()
把连接“升格”为 WebSocket 连接。 常见写法示例: 立即学习 “ go语言免费学习笔记(深入) ”; 定义全局
upgrader
,设置允许 跨域 (开发时常用):
var upgrader = websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}
在 handler 中调用
upgrader.Upgrade(w, r, nil)
,成功后得到
*websocket.Conn
实例,后续收发消息都靠它 接收和广播消息(基础通信逻辑) 每个连接对应一个 goroutine,持续读取消息;服务端可维护一个连接池(比如用
map[*websocket.Conn]bool
),配合互斥锁安全增删。 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.ReadMessage()
获取
messageType, data []byte
简单回传:直接
conn.WriteMessage(messageType, data)
广播给所有人:遍历连接池,对每个 conn 调用
WriteMessage
(注意加锁读 map,且需检查 conn 是否已关闭) 建议统一用
TextMessage
类型,避免二进制解析复杂度 处理连接生命周期与错误退出 WebSocket 连接不是永久的,网络抖动、页面关闭、心跳超时都会导致断连。不能忽略错误,否则 goroutine 泄漏。 关键做法:
ReadMessage
和
WriteMessage
都可能返回 error,最常见的是
websocket.CloseMessage
或
io.EOF
,此时应主动
conn.Close()
并从连接池中移除 给每个连接启动两个 goroutine:一个读,一个写(尤其广播场景下,避免写阻塞读) 可选加 ping/pong 心跳:设置
conn.SetPingHandler()
和
conn.SetPongHandler()
,定期调用
conn.WriteMessage(websocket.PingMessage, nil)
前端 简单测试连接(验证通路) 不用框架,纯 HTML + JS 就能快速验证服务是否跑通:
只要控制台打出 "Received: Hello from browser",说明握手成功、双向通信正常。后续可扩展 JSON 消息格式、用户标识、房间分组等。 基本上就这些。不复杂但容易忽略错误处理和并发安全——把连接管理、读写分离、异常清理做扎实,一个健壮的基础 WebSocket 服务就出来了。

相关文章