mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-30 07:40:06 +01:00
636c55b526
This also should reduce CPU time spent by GC, since inmemoryBlock.items don't have pointers now, so GC doesn't need visiting them.
82 lines
2.6 KiB
Go
82 lines
2.6 KiB
Go
package storage
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/encoding"
|
|
)
|
|
|
|
func TestMarshaledBlockHeaderSize(t *testing.T) {
|
|
// This test makes sure marshaled format isn't changed.
|
|
// If this test breaks then the storage format has been changed,
|
|
// so it may become incompatible with the previously written data.
|
|
expectedSize := 81
|
|
if marshaledBlockHeaderSize != expectedSize {
|
|
t.Fatalf("unexpected marshaledBlockHeaderSize; got %d; want %d", marshaledBlockHeaderSize, expectedSize)
|
|
}
|
|
}
|
|
|
|
func TestBlockHeaderMarshalUnmarshal(t *testing.T) {
|
|
var bh blockHeader
|
|
for i := 0; i < 1000; i++ {
|
|
bh.TSID.MetricID = uint64(i + 1)
|
|
bh.MinTimestamp = int64(-i*1e3 + 2)
|
|
bh.MaxTimestamp = int64(i*2e3 + 3)
|
|
bh.TimestampsBlockOffset = uint64(i*12345 + 4)
|
|
bh.ValuesBlockOffset = uint64(i*3243 + 5)
|
|
bh.TimestampsBlockSize = uint32((i*892 + 6) % maxBlockSize)
|
|
bh.ValuesBlockSize = uint32((i*894 + 7) % maxBlockSize)
|
|
bh.RowsCount = uint32(i*3 + 8)
|
|
bh.Scale = int16(i - 434 + 9)
|
|
bh.TimestampsMarshalType = encoding.MarshalType((i + 10) % 7)
|
|
bh.ValuesMarshalType = encoding.MarshalType((i + 11) % 7)
|
|
bh.PrecisionBits = 1 + uint8((i+12)%64)
|
|
|
|
testBlockHeaderMarshalUnmarshal(t, &bh)
|
|
}
|
|
}
|
|
|
|
func testBlockHeaderMarshalUnmarshal(t *testing.T, bh *blockHeader) {
|
|
t.Helper()
|
|
|
|
dst := bh.Marshal(nil)
|
|
if len(dst) != marshaledBlockHeaderSize {
|
|
t.Fatalf("unexpected dst size; got %d; want %d", len(dst), marshaledBlockHeaderSize)
|
|
}
|
|
var bh1 blockHeader
|
|
tail, err := bh1.Unmarshal(dst)
|
|
if err != nil {
|
|
t.Fatalf("cannot umarshal bh=%+v from dst=%x: %s", bh, dst, err)
|
|
}
|
|
if len(tail) > 0 {
|
|
t.Fatalf("unexpected tail left after unmarshaling of bh=%+v: %x", bh, tail)
|
|
}
|
|
if !reflect.DeepEqual(bh, &bh1) {
|
|
t.Fatalf("unexpected bh unmarshaled; got\n%+v; want\n%+v", &bh1, bh)
|
|
}
|
|
|
|
prefix := []byte("foo")
|
|
dstNew := bh.Marshal(prefix)
|
|
if string(dstNew[:len(prefix)]) != string(prefix) {
|
|
t.Fatalf("unexpected prefix after marshaling bh=%+v; got\n%x; want\n%x", bh, dstNew[:len(prefix)], prefix)
|
|
}
|
|
if string(dstNew[len(prefix):]) != string(dst) {
|
|
t.Fatalf("unexpected prefixed dst for bh=%+v; got\n%x; want\n%x", bh, dstNew[len(prefix):], dst)
|
|
}
|
|
|
|
suffix := []byte("bar")
|
|
dst = append(dst, suffix...)
|
|
var bh2 blockHeader
|
|
tail, err = bh2.Unmarshal(dst)
|
|
if err != nil {
|
|
t.Fatalf("cannot unmarshal bh=%+v from suffixed dst=%x: %s", bh, dst, err)
|
|
}
|
|
if string(tail) != string(suffix) {
|
|
t.Fatalf("unexpected tail after unmarshaling bh=%+v; got\n%x; want\n%x", bh, tail, suffix)
|
|
}
|
|
if !reflect.DeepEqual(bh, &bh2) {
|
|
t.Fatalf("unexpected bh unmarshaled after adding suffix; got\n%+v; want\n%+v", &bh2, bh)
|
|
}
|
|
}
|