From a6d3fca7c968dc69156beb40b85d365ac6aa865d Mon Sep 17 00:00:00 2001 From: Mahad Zaryab Date: Thu, 24 Oct 2024 23:06:49 -0400 Subject: [PATCH] Revert To Old Flow For Legacy Signed-off-by: Mahad Zaryab --- cmd/query/app/server.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cmd/query/app/server.go b/cmd/query/app/server.go index 14c855fef78..3b325f7fc3b 100644 --- a/cmd/query/app/server.go +++ b/cmd/query/app/server.go @@ -285,17 +285,17 @@ func createHTTPServerLegacy( ) (*httpServer, error) { r := createHTTPRouter(querySvc, metricsQuerySvc, queryOpts, tm, telset) var handler http.Handler = r + handler = responseHeadersHandler(handler, queryOpts.HTTP.ResponseHeaders) if queryOpts.BearerTokenPropagation { handler = bearertoken.PropagationHandler(telset.Logger, handler) } - recoveryHandler := recoveryhandler.NewRecoveryHandler(telset.Logger, true) - handler = recoveryHandler(handler) - handler = responseHeadersHandler(handler, queryOpts.HTTP.ResponseHeaders) handler = handlers.CompressHandler(handler) + recoveryHandler := recoveryhandler.NewRecoveryHandler(telset.Logger, true) + errorLog, _ := zap.NewStdLogAt(telset.Logger, zapcore.ErrorLevel) server := &httpServer{ Server: &http.Server{ - Handler: handler, + Handler: recoveryHandler(handler), ErrorLog: errorLog, ReadHeaderTimeout: 2 * time.Second, },