mirror of
https://github.com/eolinker/apinto
synced 2025-11-02 21:04:01 +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