WebSocket路由不能用app.Get或app.Use直接注册,必须用app.Get("/ws", websocket.New(handler)),因websocket.New已封装Upgrade头校验、101响应及Hijack逻辑;其他方式易导致握手失败或panic。

WebSocket路由不能用 app.Get 或 app.Use 直接注册
Fiber 的 app.Get("/ws", handler) 本质是 HTTP 路由,它返回的是 200 OK 响应体,而 WebSocket 握手必须返回 101 Switching Protocols。直接把 websocket.New() 塞进 app.Get 里看似能跑,但实际会漏掉关键的 Upgrade 头校验和连接劫持(Hijack)逻辑,导致客户端收到 200 后静默失败。
正确做法是用 app.Get 绑定路径,但 handler 必须是 websocket.New() 返回的函数——它内部已封装好 Upgrade 判定、Header 设置和 http.Hijacker 调用。其他方式(比如自己写中间件调 c.Next() 再手动升级)极易出错。
-
app.Get("/ws", websocket.New(handler))✅ 正确,websocket.New是专用包装器 -
app.Use("/ws", myUpgradeMiddleware)❌ 错误,Use不处理协议切换,只做前置拦截 -
app.Get("/ws", func(c *fiber.Ctx) { ... upgrader.Upgrade(...) })❌ 危险,绕过 Fiber 的响应生命周期,可能 panic 或丢头
路径参数和 StrictRouting 会影响 WebSocket 连接建立
如果路由带参数,比如 app.Get("/ws/:room/:id", websocket.New(h)),Fiber 默认开启 StrictRouting: true,这意味着 /ws/room1/123/(末尾多斜杠)会被视为不同路径,直接 404。而浏览器或某些客户端 SDK(如 Socket.IO 客户端)可能自动补尾斜杠,导致握手失败却查不到原因。
别关 StrictRouting:关掉后 /ws/room1/123/abc 可能被 /ws/:room/:id 错误匹配,参数解析错乱。
- 统一约定:前端发起 WebSocket 连接时,URL 严格不带尾部斜杠,例如
ws://localhost:3000/ws/chat/101 - 后端注册路由时,也只写一种形式,不重复注册
/ws/:room/:id和/ws/:room/:id/ - 可在全局加一个重定向中间件,对非根路径的尾斜杠请求 301 跳转:
c.Redirect(strings.TrimSuffix(c.Path(), "/"), 301)
中间件必须放在 websocket.New() 之前,且只能读取请求头和 URL
WebSocket 连接建立前,你只有一次机会做鉴权、日志、参数提取等操作。这些必须在 websocket.New() 包装的 handler 执行前完成,因为一旦进入 *websocket.Conn 状态,原始 *fiber.Ctx 就不可用了——Header、Query、Body 都已无法访问。
使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器,提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象,支持命名依赖等
常见错误是想在 websocket.New() 内部用 c.QueryParam("token") 或 c.Locals 传参,但此时 c 已不是 Fiber 上下文,而是原生 net/http 的 ResponseWriter 和 Request。
- 正确姿势:用
app.Use("/ws", authMiddleware)提前校验,再通过c.Locals存数据,websocket.New()的 handler 函数里拿不到,得换方案 - 更稳做法:把必要信息(如用户 ID、房间名)塞进路径参数,例如
/ws/:room/:user_id,然后在websocket.New()的 handler 里用c.Params("room")提取 - Token 鉴权必须在 Upgrade 前完成,不能等到
conn.ReadMessage()阶段再校验——连接已建,拒绝成本高
Nginx 反向代理必须透传 Upgrade 头,否则 WebSocket 握手必断
本地开发直连 go run main.go 没问题,但上生产走 Nginx,90% 的 WebSocket 失败都卡在这儿。Nginx 默认不透传 Upgrade 和 Connection 头,导致后端收不到合法握手请求,返回 200 或 400,客户端报 ERR_CONNECTION_REFUSED 或秒断(code 1006)。
必须在 location 块里显式配置:
location /ws/ {
proxy_pass http://backend;
proxy_http_version 1.1;
proxy_set_header Upgrade $http_upgrade;
proxy_set_header Connection "upgrade";
proxy_set_header Host $host;
}
-
proxy_http_version 1.1是硬性要求,HTTP/1.0 不支持 Upgrade -
$http_upgrade必须小写,Nginx 变量名区分大小写;写成$HTTP_UPGRADE就失效 - 不要加
proxy_buffering off之类无关配置,它不影响握手,只影响后续消息流 - 如果用 Cloudflare 或 ALB,同样要确认它们开启了 WebSocket 支持(Cloudflare 需 Enterprise 计划)
WebSocket 连接建立后,所有状态管理(如广播、心跳、用户上下线)必须脱离单个进程内存,否则滚动更新或 K8s 重启就会丢连接。这不是路由配置能解决的,是架构层面的事。
golang免费学习笔记(深入):立即使用
在学习笔记中,你将探索golang的核心概念和高级技巧!










