From 701d13abf9fc6fc8ab7ebfb0c112102f8de66529 Mon Sep 17 00:00:00 2001
From: Alex Le <leqiyue@amazon.com>
Date: Tue, 8 Apr 2025 01:04:23 -0700
Subject: [PATCH] Make sure LeveledCompactor respect context cancellation
 during the time opening blocks (#16407)

Signed-off-by: Alex Le <leqiyue@amazon.com>
---
 tsdb/compact.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tsdb/compact.go b/tsdb/compact.go
index 0e2c0485d9..2e9e01e852 100644
--- a/tsdb/compact.go
+++ b/tsdb/compact.go
@@ -470,6 +470,12 @@ func (c *LeveledCompactor) CompactWithBlockPopulator(dest string, dirs []string,
 	start := time.Now()
 
 	for _, d := range dirs {
+		select {
+		case <-c.ctx.Done():
+			return nil, c.ctx.Err()
+		default:
+		}
+
 		meta, _, err := readMetaFile(d)
 		if err != nil {
 			return nil, err