Set a header when we get a cache hit

Signed-off-by: milesbryant <[email protected]>
monzo · Nov 15, 2024 · f00612a · f00612a
1 parent 8a63bf3
commit f00612a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/internal/cortex/querier/queryrange/results_cache.go b/internal/cortex/querier/queryrange/results_cache.go
@@ -516,6 +516,15 @@ func (s resultsCache) handleHit(ctx context.Context, r Request, extents []Extent
 	}
 
 	response, err := s.merger.MergeResponse(r, responses...)
+	if err == nil {
+		if promResp, ok := response.(*PrometheusResponse); ok {
+			promResp.Headers = append(promResp.Headers, &PrometheusResponseHeader{
+				Name:   "X-Thanos-Results-Cache",
+				Values: []string{fmt.Sprintf("hit; extents=%d", len(extents))},
+			})
+		}
+	}
+
 	return response, mergedExtents, err
 }