util: Check context every N iterations (#6250)

cortexproject · Oct 3, 2024 · 3f18862 · 3f18862
1 parent 53556fe
commit 3f18862
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/util/strings.go b/pkg/util/strings.go
@@ -129,8 +129,10 @@ func MergeSortedSlices(ctx context.Context, a ...[]string) ([]string, error) {
 
 	r := make([]string, 0, sumLengh*2/10)
 	var current string
+	cnt := 0
 	for lt.Next() {
-		if ctx.Err() != nil {
+		cnt++
+		if cnt%CheckContextEveryNIterations == 0 && ctx.Err() != nil {
 			return nil, ctx.Err()
 		}
 		if lt.At() != current {