CR comments

Signed-off-by: Pedro Tanaka <pedro.tanaka@shopify.com>
thanos-io · Oct 31, 2024 · b69c9bb · b69c9bb
1 parent 413e96d
commit b69c9bb
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/internal/cortex/querier/queryrange/stats_middleware.go b/internal/cortex/querier/queryrange/stats_middleware.go
@@ -33,11 +33,8 @@ func (s statsMiddleware) Do(ctx context.Context, r Request) (Response, error) {
 	}
 
 	if resp.GetStats() != nil {
-		sts := stats.FromContext(ctx)
-		if sts != nil {
-			if sts.LoadPeakSamples() < resp.GetStats().Samples.PeakSamples {
-				sts.SetPeakSamples(resp.GetStats().Samples.PeakSamples)
-			}
+		if sts := stats.FromContext(ctx); sts != nil {
+			sts.SetPeakSamples(max(sts.LoadPeakSamples(), resp.GetStats().Samples.PeakSamples))
 			sts.AddTotalSamples(resp.GetStats().Samples.TotalQueryableSamples)
 		}
 	}