zishuo/apinto
1
0
Fork 0
mirror of https://github.com/eolinker/apinto synced 2025-10-31 03:56:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Liujian
2022-11-18 10:38:28 +08:00
parent 4b1780ee67
commit e25a49059d
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
node/http-context/websocket-context.go
View File

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