mirror of
https://github.com/eolinker/apinto
synced 2025-10-05 08:47:04 +08:00
修复websocket获取上游连接失败导致崩溃的问题
This commit is contained in:
@@ -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{}
|
||||
|
Reference in New Issue
Block a user