讲师中心 微信公众号
AI工具推荐 视频效率加速

如何使用Golang实现WebSocket通信_WebSocket实现流程解析

风瑶大大_5497

风瑶大大_5497

发布时间:2026-01-24 16:42:08

|

884人浏览过

|

来源于php中文网

原创

直接用 net/http 搭配 gorilla/websocket 是最稳妥的选择,因其稳定、维护活跃、自动处理握手/帧解析/心跳/关闭等细节,且避免了手动实现易出错和过时库的风险。

如何使用golang实现websocket通信_websocket实现流程解析

为什么直接用 net/http 搭配 gorilla/websocket 是最稳妥的选择

Go 标准库没有原生 WebSocket 实现,net/http 只能处理升级请求(Upgrade),真正握手、帧解析、心跳、连接管理都得靠第三方库。目前生产环境几乎统一用 gorilla/websocket,它稳定、文档清晰、维护活跃,且不依赖其他重型框架。

别碰 gobwas/ws 或自己基于 net/http 手搓——前者已归档,后者极易在掩码、分片、关闭码、ping/pong 超时等细节上出错。你不是在写 RFC 实现,是在做业务通信。

  • gorilla/websocket 的 Upgrader 自动处理 Sec-WebSocket-Key 验证和 101 切换,不用手动写 header
  • 所有读写方法(WriteMessage、ReadMessage)默认带缓冲和错误重试逻辑,比裸 conn.Read/Write 安全得多
  • 连接关闭时会自动发送 CloseMessage 帧,避免客户端卡在 “closing” 状态

如何正确升级 HTTP 连接并避免 400 / 426 错误

常见错误是把 WebSocket 路由和普通 HTTP 路由混写,或没设置 Upgrader.CheckOrigin,导致浏览器发来的 Origin 头被拒绝,返回 400;或者服务端没响应 Upgrade: websocket,返回 426 Upgrade Required。

关键点在于:必须用 http.HandlerFunc 显式处理升级,不能塞进中间件链里(除非中间件明确支持 http.Hijacker)。

立即学习“go语言免费学习笔记(深入)”;

Miller CSV TSV JSON 数据处理器
Miller CSV TSV JSON 数据处理器

Miller (mlr) 是一个命令行工具,用于查询、整形和重新格式化名称索引数据,如 CSV、TSV、JSON 和 JSON Lines。它将 awk、sed、cut、join 和 sort 的功能整合到一个专为结构化数据处理而构建的单一工具中。

下载
func wsHandler(w http.ResponseWriter, r *http.Request) {
	upgrader := websocket.Upgrader{
		CheckOrigin: func(r *http.Request) bool {
			return true // 生产请校验 Origin,例如 r.Header.Get("Origin") == "https://myapp.com"
		},
	}
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		http.Error(w, "Upgrade error", http.StatusBadRequest)
		return
	}
	defer conn.Close()

	// 后续读写在此进行
	for {
		_, msg, err := conn.ReadMessage()
		if err != nil {
			break
		}
		if err := conn.WriteMessage(websocket.TextMessage, msg); err != nil {
			break
		}
	}
}
  • CheckOrigin 默认返回 false,必须显式设为 true 或实现白名单逻辑,否则跨域请求必失败
  • Upgrade 必须在 response body 写入前调用,任何 fmt.Fprintf(w, ...) 或 json.NewEncoder(w).Encode(...) 都会导致 panic 或 500
  • 不要对 *http.ResponseWriter 做任何其他操作(如设置 Header().Set),升级后它就失效了

如何安全地并发读写并避免 websocket: write deadline exceeded

WebSocket 连接是全双工的,但 gorilla/websocket 的底层连接不允许多 goroutine 同时调用 WriteMessage —— 会 panic 报 concurrent write to connection。而读写超时(write deadline exceeded)通常是因为写操作阻塞太久,比如客户端断连但服务端还在发消息。

标准解法是:一个 goroutine 专责读(处理业务逻辑 + 关闭信号),另一个 goroutine 专责写(从 channel 拉消息,带超时控制)。

func handleConnection(conn *websocket.Conn) {
	// 读协程
	go func() {
		defer conn.Close()
		for {
			_, msg, err := conn.ReadMessage()
			if err != nil {
				return
			}
			// 解析 msg,投递到业务 channel 或广播池
		}
	}()

	// 写协程
	writer := func() {
		defer conn.Close()
		ticker := time.NewTicker(pingPeriod)
		defer ticker.Stop()
		for {
			select {
			case message, ok := <-broadcastChan:
				if !ok {
					return
				}
				if err := conn.SetWriteDeadline(time.Now().Add(writeWait)); err != nil {
					return
				}
				if err := conn.WriteMessage(websocket.TextMessage, message); err != nil {
					return
				}
			case <-ticker.C:
				if err := conn.SetWriteDeadline(time.Now().Add(writeWait)); err != nil {
					return
				}
				if err := conn.WriteMessage(websocket.PingMessage, nil); err != nil {
					return
				}
			}
		}
	}()
	go writer()
}
  • 每个连接必须有自己的 writeWait(建议 10s)和 pingPeriod(建议 30s),不能全局共用
  • 写操作前务必调用 conn.SetWriteDeadline,否则网络卡住时会永久阻塞 goroutine
  • 不要用 conn.SetDeadline 同时管读写——读写超时需求不同,混合设置反而容易误关连接

客户端断连时,服务端怎么知道并清理资源

WebSocket 没有 TCP 那样的 FIN 包保证,客户端静默掉线(比如关机、断网)时,服务端可能数分钟甚至更久才收到 EOF。不能只靠 ReadMessage 返回 error 就清理——那太慢,也漏掉“假在线”连接。

必须主动探测:服务端定期发 Ping,客户端必须回 Pong;若连续几次没收到 Pong,就关连接。

  • Upgrader 提供 EnableCompression: true 和 HandshakeTimeout,但这两个和断连检测无关,别混淆
  • conn.SetPongHandler 必须在 Upgrade 后立刻注册,且 handler 内不要做耗时操作(比如查 DB),否则会阻塞整个连接的读循环
  • 真实项目中,连接要绑定用户 ID 或 session,并注册到 map[string]*websocket.Conn 中;关闭前记得 delete,否则内存泄漏

最易忽略的是:Ping/Pong 是 WebSocket 协议层机制,不是业务消息,ReadMessage 永远不会读到 PongMessage —— 它被库自动处理了,你只需设 handler 更新最后活跃时间即可。

热门AI工具

更多
立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

479

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

576

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

976

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

751

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

1446

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

3994

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

1774

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

3746

2025.06.17

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

60

2026.09.23

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
WEB前端教程【HTML5+CSS3+JS】
WEB前端教程【HTML5+CSS3+JS】

共101课时 | 20.6万人学习

JS进阶与BootStrap学习
JS进阶与BootStrap学习

共39课时 | 4.7万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn