mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-02-02 08:31:11 -08:00
Code Issues Actions 55 Packages Projects Releases Wiki Activity

Merge pull request #12755 from bboreham/rangequery-benchmark-mmap

Browse source 
promql: force mmap of head chunks in BenchmarkRangeQuery
This commit is contained in:
Bryan Boreham 2023-08-26 15:56:52 +01:00 committed by GitHub
parent 5d22d422ab 0d283effa8
commit c5671c6d97
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
promql/bench_test.go
View file

@ -66,6 +66,8 @@ func setupRangeQueryTestData(stor *teststorage.TestStorage, _ *Engine, interval,
return err return err
} }
} }
stor.DB.ForceHeadMMap() // Ensure we have at most one head chunk for every series.
stor.DB.Compact()
return nil return nil
} }
@ -222,6 +224,7 @@ func rangeQueryCases() []benchCase {
func BenchmarkRangeQuery(b *testing.B) { func BenchmarkRangeQuery(b *testing.B) {
stor := teststorage.New(b) stor := teststorage.New(b)
stor.DB.DisableCompactions() // Don't want auto-compaction disrupting timings.
defer stor.Close() defer stor.Close()
opts := EngineOpts{ opts := EngineOpts{
Logger: nil, Logger: nil,

5
tsdb/db.go
View file

@ -1797,6 +1797,11 @@ func (db *DB) EnableCompactions() {
level.Info(db.logger).Log("msg", "Compactions enabled") level.Info(db.logger).Log("msg", "Compactions enabled")
} }
// ForceHeadMMap is intended for use only in tests and benchmarks.
func (db *DB) ForceHeadMMap() {
db.head.mmapHeadChunks()
}
// Snapshot writes the current data to the directory. If withHead is set to true it // Snapshot writes the current data to the directory. If withHead is set to true it
// will create a new block containing all data that's currently in the memory buffer/WAL. // will create a new block containing all data that's currently in the memory buffer/WAL.
func (db *DB) Snapshot(dir string, withHead bool) error { func (db *DB) Snapshot(dir string, withHead bool) error {