check context cancel when doing posting batches (thanos-io#6396)

Signed-off-by: Ben Ye <[email protected]>
databricks · Jun 27, 2023 · 2163faf · 2163faf
1 parent 69bd4bc
commit 2163faf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/store/bucket.go b/pkg/store/bucket.go
@@ -1013,6 +1013,9 @@ func (b *blockSeriesClient) nextBatch() error {
 
 	b.entries = b.entries[:0]
 	for i := 0; i < len(postingsBatch); i++ {
+		if err := b.ctx.Err(); err != nil {
+			return err
+		}
 		ok, err := b.indexr.LoadSeriesForTime(postingsBatch[i], &b.symbolizedLset, &b.chkMetas, b.skipChunks, b.mint, b.maxt)
 		if err != nil {
 			return errors.Wrap(err, "read series")