Update pkg/distributor/distributor_test.go

Co-authored-by: Arve Knudsen <[email protected]>
grafana · Jan 21, 2025 · 61dcc6d · 61dcc6d
1 parent 4127722
commit 61dcc6d
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/pkg/distributor/distributor_test.go b/pkg/distributor/distributor_test.go
@@ -6548,9 +6548,14 @@ loop:
 	for _, ts := range i.timeseries {
 		if match(ts.Labels, matchers) {
 			for _, lbl := range ts.Labels {
-				if _, ok := labelsSet[lbl.Name]; !ok {
-					labelsSet[lbl.Name] = struct{}{}
-					response.LabelNames = append(response.LabelNames, lbl.Name)
+				if _, ok := labelsSet[lbl.Name]; ok {
+					continue
+				}
+
+				labelsSet[lbl.Name] = struct{}{}
+				response.LabelNames = append(response.LabelNames, lbl.Name)
+				if hints != nil && hints.Limit > 0 && len(labelsSet) >= hints.Limit {
+					break loop
 				}
 			}
 		}