diff --git a/pkg/infra/httpclient/httpclientprovider/datasource_metrics_middleware.go b/pkg/infra/httpclient/httpclientprovider/datasource_metrics_middleware.go index 2709ed6ab76..bfb722ef4cd 100644 --- a/pkg/infra/httpclient/httpclientprovider/datasource_metrics_middleware.go +++ b/pkg/infra/httpclient/httpclientprovider/datasource_metrics_middleware.go @@ -96,7 +96,7 @@ func executeMiddleware(next http.RoundTripper, datasourceLabel prometheus.Labels return nil, err } - if res != nil { + if res != nil && res.StatusCode != http.StatusSwitchingProtocols { res.Body = httpclient.CountBytesReader(res.Body, func(bytesRead int64) { responseSizeSummary.Observe(float64(bytesRead)) }) diff --git a/pkg/infra/httpclient/httpclientprovider/response_limit_middleware.go b/pkg/infra/httpclient/httpclientprovider/response_limit_middleware.go index c17c70b3516..97d8285d4cb 100644 --- a/pkg/infra/httpclient/httpclientprovider/response_limit_middleware.go +++ b/pkg/infra/httpclient/httpclientprovider/response_limit_middleware.go @@ -21,7 +21,10 @@ func ResponseLimitMiddleware(limit int64) sdkhttpclient.Middleware { return nil, err } - res.Body = httpclient.MaxBytesReader(res.Body, limit) + if res != nil && res.StatusCode != http.StatusSwitchingProtocols { + res.Body = httpclient.MaxBytesReader(res.Body, limit) + } + return res, nil }) })