Merge pull request thanos-io#1 from codesome/pracucci-prom-upgrade

Delete mmapped chunks after creating block from head
codesome · May 26, 2020 · 46e8802 · 46e8802
2 parents 1a0b648 + 9728c77
commit 46e8802
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/pkg/testutil/e2eutil/prometheus.go b/pkg/testutil/e2eutil/prometheus.go
@@ -411,11 +411,17 @@ func createBlock(
 	resolution int64,
 	tombstones bool,
 ) (id ulid.ULID, err error) {
-	h, err := tsdb.NewHead(nil, nil, nil, 10000000000, dir, nil, tsdb.DefaultStripeSize, nil)
+	chunksRootDir := filepath.Join(dir, "chunks")
+	h, err := tsdb.NewHead(nil, nil, nil, 10000000000, chunksRootDir, nil, tsdb.DefaultStripeSize, nil)
 	if err != nil {
 		return id, errors.Wrap(err, "create head block")
 	}
-	defer runutil.CloseWithErrCapture(&err, h, "TSDB Head")
+	defer func() {
+		runutil.CloseWithErrCapture(&err, h, "TSDB Head")
+		if e := os.RemoveAll(chunksRootDir); e != nil {
+			err = errors.Wrap(e, "delete chunks dir")
+		}
+	}()
 
 	var g errgroup.Group
 	var timeStepSize = (maxt - mint) / int64(numSamples+1)