fix GetActiveAndPartialBlockIDs panic (thanos-io#7621)

Signed-off-by: Ben Ye <[email protected]>
harry671003 · Aug 12, 2024 · 2375b59 · 2375b59
1 parent c9500df
commit 2375b59
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/block/fetcher.go b/pkg/block/fetcher.go
@@ -257,7 +257,11 @@ func (f *ConcurrentLister) GetActiveAndPartialBlockIDs(ctx context.Context, ch c
 					mu.Unlock()
 					continue
 				}
-				ch <- uid
+				select {
+				case <-ctx.Done():
+					return ctx.Err()
+				case ch <- uid:
+				}
 			}
 			return nil
 		})