From 751e79751acc374dc1cca6788545fb42cf3c5480 Mon Sep 17 00:00:00 2001 From: Ben Ye Date: Thu, 25 May 2023 16:07:18 -0700 Subject: [PATCH] check context cancel when doing posting batches Signed-off-by: Ben Ye --- pkg/store/bucket.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pkg/store/bucket.go b/pkg/store/bucket.go index 147d12609c..c3d740a8e0 100644 --- 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")