Beego框架即时通讯系统开发实战-Golang-PHP中文网

beego 做即时通讯可行但需绕开其 http 生命周期：websocket 必须用 http.handlefunc 直接注册，避开 beego.router；需手动实现心跳、超时、连接清理和广播优化，框架本身不参与 websocket 管理。

beego框架即时通讯系统开发实战

Beego 做即时通讯系统是可行的，但必须绕开它默认的 HTTP 请求生命周期模型——WebSocket 连接不能走 beego.Controller 的常规 Get() 流程，否则连接会在响应返回后立即关闭。

为什么直接在 Controller 的 Get 方法里调用 Upgrade 会失败

常见错误现象是：前端发起 WebSocket 连接后，浏览器控制台报 WebSocket connection to 'ws://localhost:8080/ws' failed: Error during WebSocket handshake: Unexpected response code: 200。这是因为 Beego 默认把所有请求都当作 HTTP 处理，即使你写了 Get() 方法，框架仍会自动写入 HTTP 状态码 200 和 Content-Type，并结束 ResponseWriter。而 gorilla/websocket.Upgrader.Upgrade() 要求底层连接未被写入、未被关闭，否则升级必然失败。

解决方式不是“修复”这个流程，而是彻底跳过 Beego 的 HTTP 响应封装：

不要让 WebSocket 路由注册进 beego.Router 或 beego.Get；
改用 beego.BeeApp.Handlers 直接挂载自定义的 http.HandlerFunc；
或在 main.go 中，在 beego.Run() 之前，用 http.Handle() 显式接管路径（如 /ws），确保该路径完全脱离 Beego 路由中间件链。

如何正确注册 WebSocket 处理器并保持连接存活

Beego 自身不提供原生 WebSocket 控制器抽象，所以你要自己管理连接生命周期。推荐做法是在 main.go 里注册裸 handler：

package main

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
    "github.com/astaxie/beego"
)

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool { return true }, // 生产环境需校验 Origin
}

func wsHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Printf("upgrade error: %v", err)
        return
    }
    defer conn.Close()

    // 此处开始读写循环，例如：
    for {
        _, msg, err := conn.ReadMessage()
        if err != nil {
            log.Printf("read error: %v", err)
            break
        }
        if err := conn.WriteMessage(websocket.TextMessage, msg); err != nil {
            log.Printf("write error: %v", err)
            break
        }
    }
}

func main() {
    // 先接管 /ws 路径，避开 Beego 路由系统
    http.HandleFunc("/ws", wsHandler)

    // 再启动 Beego（它会监听其他路径）
    beego.Run()
}

注意：http.HandleFunc 必须在 beego.Run() 之前调用，否则 Beego 启动后会独占 http.DefaultServeMux，你的 handler 将无效。