Fix tracking websocket connections (#26943)

Этот коммит содержится в:
Claudio Costa
2024-05-08 10:12:50 -06:00
коммит произвёл GitHub
родитель 8fa6757949
Коммит be2ffbcd0c
2 изменённых файлов: 7 добавлений и 6 удалений

Просмотреть файл

@@ -417,8 +417,15 @@ func (wc *WebConn) Pump() {
func (wc *WebConn) readPump() {
defer func() {
if metrics := wc.Platform.metricsIFace; metrics != nil {
metrics.DecrementHTTPWebSockets(wc.originClient)
}
wc.WebSocket.Close()
}()
if metrics := wc.Platform.metricsIFace; metrics != nil {
metrics.IncrementHTTPWebSockets(wc.originClient)
}
wc.WebSocket.SetReadLimit(model.SocketMaxMessageSizeKb)
wc.WebSocket.SetReadDeadline(time.Now().Add(pongWaitTime))
wc.WebSocket.SetPongHandler(func(string) error {