mirror of
https://github.com/prometheus/prometheus
synced 2024-12-24 23:42:32 +00:00
Add benchmark for Querier.Select over blocks and head.
Signed-off-by: Brian Brazil <brian.brazil@robustperception.io>
This commit is contained in:
parent
b90b5736b9
commit
1d1732bc25
@ -14,6 +14,7 @@
|
|||||||
package tsdb
|
package tsdb
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
"strconv"
|
"strconv"
|
||||||
@ -123,3 +124,56 @@ func benchmarkPostingsForMatchers(b *testing.B, ir IndexReader) {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func BenchmarkQuerierSelect(b *testing.B) {
|
||||||
|
h, err := NewHead(nil, nil, nil, 1000)
|
||||||
|
testutil.Ok(b, err)
|
||||||
|
defer h.Close()
|
||||||
|
app := h.Appender()
|
||||||
|
numSeries := 1000000
|
||||||
|
for i := 0; i < numSeries; i++ {
|
||||||
|
app.Add(labels.FromStrings("foo", "bar", "i", fmt.Sprintf("%d%s", i, postingsBenchSuffix)), int64(i), 0)
|
||||||
|
}
|
||||||
|
testutil.Ok(b, app.Commit())
|
||||||
|
|
||||||
|
bench := func(b *testing.B, br BlockReader) {
|
||||||
|
matcher := labels.MustNewMatcher(labels.MatchEqual, "foo", "bar")
|
||||||
|
for s := 1; s <= numSeries; s *= 10 {
|
||||||
|
b.Run(fmt.Sprintf("%dof%d", s, numSeries), func(b *testing.B) {
|
||||||
|
q, err := NewBlockQuerier(br, 0, int64(s-1))
|
||||||
|
testutil.Ok(b, err)
|
||||||
|
|
||||||
|
b.ResetTimer()
|
||||||
|
for i := 0; i < b.N; i++ {
|
||||||
|
ss, err := q.Select(matcher)
|
||||||
|
testutil.Ok(b, err)
|
||||||
|
for ss.Next() {
|
||||||
|
}
|
||||||
|
testutil.Ok(b, ss.Err())
|
||||||
|
}
|
||||||
|
q.Close()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
b.Run("Head", func(b *testing.B) {
|
||||||
|
bench(b, h)
|
||||||
|
})
|
||||||
|
|
||||||
|
tmpdir, err := ioutil.TempDir("", "test_benchquerierselect")
|
||||||
|
testutil.Ok(b, err)
|
||||||
|
defer func() {
|
||||||
|
testutil.Ok(b, os.RemoveAll(tmpdir))
|
||||||
|
}()
|
||||||
|
|
||||||
|
blockdir := createBlockFromHead(b, tmpdir, h)
|
||||||
|
block, err := OpenBlock(nil, blockdir, nil)
|
||||||
|
testutil.Ok(b, err)
|
||||||
|
defer func() {
|
||||||
|
testutil.Ok(b, block.Close())
|
||||||
|
}()
|
||||||
|
|
||||||
|
b.Run("Block", func(b *testing.B) {
|
||||||
|
bench(b, block)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user