Echo框架怎么实现WebSocket即时通讯-Golang-PHP中文网

WebSocket连接必须用Upgrade处理，Echo不支持直接GET路由挂websocket.Handler；正确做法是在e.GET("/ws", func(c echo.Context) error { ... })中调用upgrader.Upgrade()，升级前完成鉴权、设置头，升级后通过*websocket.Conn读写，禁用c.Response()/c.Request()，并手动管理超时、心跳、并发锁与连接清理。

echo框架怎么实现websocket即时通讯

WebSocket连接必须用 Upgrade 处理，Echo 不支持直接 `GET` 路由挂 `websocket.Handler`

Echo 本身不内置 WebSocket 服务端逻辑，它只提供 HTTP 路由和中间件能力。想用 WebSocket，得靠 gorilla/websocket 或 gobwas/ws 这类库接管升级流程。常见错误是直接在 e.GET("/ws", handler) 里传一个 http.HandlerFunc，结果浏览器报 400 Bad Request 或 Connection closed before receiving a handshake response —— 因为没走 Upgrade 流程。

正确做法是：用 e.GET("/ws", func(c echo.Context) error { ... })，在函数体内调用 upgrader.Upgrade()，并显式设置响应头、禁用 Echo 的默认写入器。关键点：

upgrader.CheckOrigin 默认拒绝所有跨域请求，开发时可临时设为 func(r *http.Request) bool { return true }，上线必须收紧
必须在 Upgrade() 前调用 c.Response().Header().Set("Sec-Websocket-Protocol", "...")（如需子协议）
调用 Upgrade() 后，c.Response() 和 c.Request() 不再可用，后续读写必须通过返回的 *websocket.Conn

消息收发要自己管理连接生命周期，Echo 不参与 `Conn.ReadMessage()` 循环

建立连接后，Echo 的请求上下文就结束了。你得在 goroutine 里手动调用 conn.ReadMessage() 和 conn.WriteMessage()，否则连接会空闲超时断开，或者消息堆积阻塞。

典型结构：

go func() {
    defer conn.Close()
    for {
        _, msg, err := conn.ReadMessage()
        if err != nil {
            // 检查是否是 websocket.CloseMessage 或 net.ErrClosed
            break
        }
        // 处理业务逻辑，比如广播给其他连接
        broadcast(msg)
    }
}()

注意：

不要在主线程里同步 WriteMessage()，容易因对方网络卡顿导致整个 goroutine 阻塞；建议用带缓冲 channel 投递消息
conn.SetReadDeadline() 和 conn.SetWriteDeadline() 必须手动设，Echo 不帮你管这些
多个 goroutine 并发写同一连接会 panic，必须加锁或用 conn.WriteJSON() + 单独写协程

广播场景下不能遍历原始 `*websocket.Conn` 切片，要防并发读写和已断连

很多人把连接存进 map[string]*websocket.Conn 或 []*websocket.Conn，然后循环调用 WriteMessage()。问题立刻出现：write tcp: use of closed network connection 或 concurrent write to websocket connection。

Echo框架 5.1.0

Echo框架 5.1.0 版本源码包下载，适合关注 RealIP 行为变化、StartConfig.Listener、NewDefaultFS 和观测性中间件入口的开发团队。

下载

安全做法是：

用 sync.Map 存连接，key 是连接 ID（比如 c.Request().RemoteAddr 或 JWT 用户 ID），value 是自定义结构体，含 *websocket.Conn 和 sync.Mutex
每次写之前先 conn.WriteMutex.Lock()，写完 defer Unlock()；读消息时也锁，避免读写冲突
广播前用 conn.WriteMessage(websocket.PingMessage, nil) 探活，失败则从 map 中删掉该连接
务必监听 conn.SetCloseHandler()，主动清理断连资源

生产环境必须配好超时、心跳和反向代理转发头

Nginx 或 Traefik 默认会中断空闲长连接。如果只写了基础代码，部署后可能 60 秒就自动断开，用户毫无感知。

必须配置：

Nginx：加 proxy_read_timeout 300;、proxy_send_timeout 300;、proxy_http_version 1.1;、proxy_set_header Upgrade $http_upgrade;、proxy_set_header Connection "upgrade";
代码侧：设 upgrader.HandshakeTimeout = 5 * time.Second，conn.SetPongHandler() 响应心跳，conn.SetPingHandler() 主动发 ping
客户端 JS 也要实现重连逻辑，WebSocket 对象断开后不会自动恢复