修复websocket获取上游连接失败导致崩溃的问题

This commit is contained in:
Liujian
2022-11-18 10:38:28 +08:00
parent 4b1780ee67
commit e25a49059d

View File

@@ -26,6 +26,11 @@ var upgrader = websocket.FastHTTPUpgrader{}
func (w *WebsocketContext) Upgrade() error {
err := upgrader.Upgrade(w.fastHttpRequestCtx, func(conn *websocket.Conn) {
if w.upstreamConn == nil {
// 上游连接失败,直接返回
log.Error("fail to connect upstream")
return
}
defer conn.Close()
defer w.upstreamConn.Close()
wg := &sync.WaitGroup{}