mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-05 01:01:09 +01:00
c0b852d50d
Callers of InitFromFilePart log the error and exit. It is better to log the error with the path to the part and the call stack directly inside the MustInitFromFilePart() function. This simplifies the code at callers' side while leaving the same level of debuggability.
58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
package storage
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
func BenchmarkBlockStreamReaderBlocksWorstCase(b *testing.B) {
|
|
benchmarkBlockStreamReader(b, benchInmemoryPartWorstCase, false)
|
|
}
|
|
|
|
func BenchmarkBlockStreamReaderBlocksBestCase(b *testing.B) {
|
|
benchmarkBlockStreamReader(b, benchInmemoryPartBestCase, false)
|
|
}
|
|
|
|
func BenchmarkBlockStreamReaderRowsWorstCase(b *testing.B) {
|
|
benchmarkBlockStreamReader(b, benchInmemoryPartWorstCase, true)
|
|
}
|
|
|
|
func BenchmarkBlockStreamReaderRowsBestCase(b *testing.B) {
|
|
benchmarkBlockStreamReader(b, benchInmemoryPartBestCase, true)
|
|
}
|
|
|
|
func benchmarkBlockStreamReader(b *testing.B, mp *inmemoryPart, readRows bool) {
|
|
b.ReportAllocs()
|
|
b.SetBytes(int64(mp.ph.RowsCount))
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
var bsr blockStreamReader
|
|
blockNum := 0
|
|
for pb.Next() {
|
|
bsr.MustInitFromInmemoryPart(mp)
|
|
for bsr.NextBlock() {
|
|
if !readRows {
|
|
continue
|
|
}
|
|
if err := bsr.Block.UnmarshalData(); err != nil {
|
|
panic(fmt.Errorf("unexpected error when unmarshaling rows on block %d: %w", blockNum, err))
|
|
}
|
|
for bsr.Block.nextRow() {
|
|
}
|
|
}
|
|
if err := bsr.Error(); err != nil {
|
|
panic(fmt.Errorf("unexpected error when reading block %d: %w", blockNum, err))
|
|
}
|
|
blockNum++
|
|
}
|
|
})
|
|
}
|
|
|
|
var benchInmemoryPartWorstCase = newTestInmemoryPart(benchRawRowsWorstCase)
|
|
var benchInmemoryPartBestCase = newTestInmemoryPart(benchRawRowsBestCase)
|
|
|
|
func newTestInmemoryPart(rows []rawRow) *inmemoryPart {
|
|
var mp inmemoryPart
|
|
mp.InitFromRows(rows)
|
|
return &mp
|
|
}
|