如何使用 Gorilla Websockets 和 alexedwards/scs/v2 实现 http.Hijacker

Question

最近将 Go Web 应用程序中出色的 alexedwards/scs/v2 从 2.5.0 升级到 2.7.0，以允许 Go 1.20+ 支持 http.NewResponseController()。随后允许扩展处理程序特定的服务器写入超时以进行大文件上传。这很好。 此必要升级的一个不幸后果是 websocket 功能的丢失，目前使用 Gorilla websockets v1.5.1（最新）。我在尝试连接升级时遇到常见错误... websocket：响应未实现http.Hijacker 我已经为此研究了几个线索，其中一些是关于 SO 的，但我还无法解决任何特定于本案的问题。 我已经尝试了一些尝试与我的其他中间件代码一起并在我的 websocket 端点处理程序中实现 http.Hijacker （注释掉，附加“误导”示例以供说明）...我只是认为我对此理解得不够好。 ..

type WebsocketConnection struct {
    *websocket.Conn
}

type WebsocketPayload struct {
    Action      string              `json:"action"`
    Message     string              `json:"message"`
    AdminName   string              `json:"admin_name"`
    MessageType string              `json:"message_type"`
    AdminID     int                 `json:"admin_id"`
    Connection  WebsocketConnection `json:"-"`
}

type WebsocketJSONResponse struct {
    Action        string   `json:"action"`
    Message       string   `json:"message"`
    AdminName     string   `json:"admin_name"`
    AdminID       int      `json:"admin_id"`
    CurrentLogins []string `json:"current_logins"`
}

var upgradeConnection = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    CheckOrigin:     func(r *http.Request) bool { return true },
}

var webClients = make(map[WebsocketConnection]string)

var websocketChannel = make(chan WebsocketPayload)

func (app *application) WebsocketEndPoint(w http.ResponseWriter, r *http.Request) {
    // rc := http.NewResponseController(w)
    // netConn, _, err := rc.Hijack()
    // if err != nil {
    //  app.errorLog.Println(err)
    //  return
    // }
    // defer netConn.Close()

    app.debugLog.Printf("w's type is %T\n", w)

    ws, err := upgradeConnection.Upgrade(w, r, nil)
    if err != nil {
        app.errorLog.Println(err)  <= ERROR HERE
        return
    }

    app.infoLog.Printf("Web client connected from %s", r.RemoteAddr)
    var response WebsocketJSONResponse
    response.Message = "Connected to server..."

    err = ws.WriteJSON(response)
    if err != nil {
        app.errorLog.Println(err)
        return
    }

    conn := WebsocketConnection{Conn: ws}
    webClients[conn] = ""

    go app.ListenForWebsocket(&conn)
}

“w”在本例中是 *scs.sessionResponseWriter。 如何在响应中实现 http.Hijacker 而不破坏连接并进行升级？请问正确的做法是什么？

Answer 1

alexedwards/scs/v2 软件包的 2.5.0 版本支持 Hijacker 接口。 2.7.0 版本不支持 Hijacker 接口，但支持支持响应控制器。 gorilla websocket 包不使用响应控制器。

选项1

使用 alexedwards/scs/v2 软件包的 2.5.0。

选项2

通过在应用程序代码中解开响应编写器来解决 gorilla 包的缺陷：

func (app *application) WebsocketEndPoint(w http.ResponseWriter, r *http.Request) {

    wupgrade := w
    if u, ok := w.(interface{ Unwrap() http.ResponseWriter }); ok {
        wupgrade = u.Unwrap()
    }

    app.debugLog.Printf("w's type is %T\n", w)

    ws, err := upgradeConnection.Upgrade(wupgrade, r, nil)
    if err != nil {
        app.errorLog.Println(err)
        return
    }

    app.infoLog.Printf("Web client connected from %s", r.RemoteAddr)
    var response WebsocketJSONResponse
    response.Message = "Connected to server..."

    err = ws.WriteJSON(response)
    if err != nil {
        app.errorLog.Println(err)
        return
    }

    conn := WebsocketConnection{Conn: ws}
    webClients[conn] = ""

    go app.ListenForWebsocket(&conn)
}