Add response compression wrapper for query api (cortexproject#6844)

afhassan · web-flow · commit 16a2f54dde3a · 2025-06-26T17:28:06.000-07:00
* add response compression wrapper to query api

Signed-off-by: Ahmed Hassan &lt;afayekhassan@gmail.com&gt;

* fix formatting

Signed-off-by: Ahmed Hassan &lt;afayekhassan@gmail.com&gt;

---------

Signed-off-by: Ahmed Hassan &lt;afayekhassan@gmail.com&gt;
diff --git a/pkg/api/queryapi/query_api.go b/pkg/api/queryapi/query_api.go
@@ -188,7 +188,7 @@ func (q *QueryAPI) InstantQueryHandler(r *http.Request) (result apiFuncResult) {
 }
 
 func (q *QueryAPI) Wrap(f apiFunc) http.HandlerFunc {
-	return func(w http.ResponseWriter, r *http.Request) {
+	hf := func(w http.ResponseWriter, r *http.Request) {
 		httputil.SetCORS(w, q.CORSOrigin, r)
 
 		result := f(r)
@@ -207,6 +207,10 @@ func (q *QueryAPI) Wrap(f apiFunc) http.HandlerFunc {
 		}
 		w.WriteHeader(http.StatusNoContent)
 	}
+
+	return httputil.CompressionHandler{
+		Handler: http.HandlerFunc(hf),
+	}.ServeHTTP
 }
 
 func (q *QueryAPI) respond(w http.ResponseWriter, req *http.Request, data interface{}, warnings annotations.Annotations, query string) {

Original file line number	Diff line number	Diff line change
`@@ -188,7 +188,7 @@ func (q QueryAPI) InstantQueryHandler(r http.Request) (result apiFuncResult) {`
`188`	`188`	`}`
`189`	`189`
`190`	`190`	`func (q *QueryAPI) Wrap(f apiFunc) http.HandlerFunc {`
`191`		`- return func(w http.ResponseWriter, r *http.Request) {`
	`191`	`+ hf := func(w http.ResponseWriter, r *http.Request) {`
`192`	`192`	`httputil.SetCORS(w, q.CORSOrigin, r)`
`193`	`193`
`194`	`194`	`result := f(r)`
`@@ -207,6 +207,10 @@ func (q *QueryAPI) Wrap(f apiFunc) http.HandlerFunc {`
`207`	`207`	`}`
`208`	`208`	`w.WriteHeader(http.StatusNoContent)`
`209`	`209`	`}`
	`210`	`+`
	`211`	`+ return httputil.CompressionHandler{`
	`212`	`+ Handler: http.HandlerFunc(hf),`
	`213`	`+ }.ServeHTTP`
`210`	`214`	`}`
`211`	`215`
`212`	`216`	`func (q QueryAPI) respond(w http.ResponseWriter, req http.Request, data interface{}, warnings annotations.Annotations, query string) {`