func TestIndirectIndex_MaxBlocks(t *testing.T) {
index := tsm1.NewIndexWriter()
for i := 0; i < 1<<16; i++ {
index.Add("cpu", tsm1.BlockFloat64, 0, 1, 10, 20)
}
if _, err := index.MarshalBinary(); err == nil {
t.Fatalf("expected max block count error. got nil")
} else {
println(err.Error())
}
}
func BenchmarkIndirectIndex_UnmarshalBinary(b *testing.B) {
index := tsm1.NewIndexWriter()
for i := 0; i < 100000; i++ {
index.Add(fmt.Sprintf("cpu-%d", i), tsm1.BlockFloat64, int64(i*2), int64(i*2+1), 10, 100)
}
bytes, err := index.MarshalBinary()
if err != nil {
b.Fatalf("unexpected error marshaling index: %v", err)
}
indirect := tsm1.NewIndirectIndex()
b.ResetTimer()
for i := 0; i < b.N; i++ {
if err := indirect.UnmarshalBinary(bytes); err != nil {
b.Fatalf("unexpected error unmarshaling index: %v", err)
}
}
}
func TestIndirectIndex_Type(t *testing.T) {
index := tsm1.NewIndexWriter()
index.Add("cpu", tsm1.BlockInteger, 0, 1, 10, 20)
b, err := index.MarshalBinary()
ind := tsm1.NewIndirectIndex()
if err := ind.UnmarshalBinary(b); err != nil {
fatal(t, "unmarshal binary", err)
}
typ, err := ind.Type("cpu")
if err != nil {
fatal(t, "reading type", err)
}
if got, exp := typ, tsm1.BlockInteger; got != exp {
t.Fatalf("type mismatch: got %v, exp %v", got, exp)
}
}
func TestIndirectIndex_Entries(t *testing.T) {
index := tsm1.NewIndexWriter()
index.Add("cpu", tsm1.BlockFloat64, 0, 1, 10, 100)
index.Add("cpu", tsm1.BlockFloat64, 2, 3, 20, 200)
index.Add("mem", tsm1.BlockFloat64, 0, 1, 10, 100)
b, err := index.MarshalBinary()
if err != nil {
t.Fatalf("unexpected error marshaling index: %v", err)
}
indirect := tsm1.NewIndirectIndex()
if err := indirect.UnmarshalBinary(b); err != nil {
t.Fatalf("unexpected error unmarshaling index: %v", err)
}
exp := index.Entries("cpu")
entries := indirect.Entries("cpu")
if got, exp := len(entries), len(exp); got != exp {
t.Fatalf("entries length mismatch: got %v, exp %v", got, exp)
}
for i, exp := range exp {
got := entries[i]
if exp.MinTime != got.MinTime {
t.Fatalf("minTime mismatch: got %v, exp %v", got.MinTime, exp.MinTime)
}
if exp.MaxTime != got.MaxTime {
t.Fatalf("minTime mismatch: got %v, exp %v", got.MaxTime, exp.MaxTime)
}
if exp.Size != got.Size {
t.Fatalf("size mismatch: got %v, exp %v", got.Size, exp.Size)
}
if exp.Offset != got.Offset {
t.Fatalf("size mismatch: got %v, exp %v", got.Offset, exp.Offset)
}
}
}
func TestIndirectIndex_Keys(t *testing.T) {
index := tsm1.NewIndexWriter()
index.Add("cpu", tsm1.BlockFloat64, 0, 1, 10, 20)
index.Add("mem", tsm1.BlockFloat64, 0, 1, 10, 20)
index.Add("cpu", tsm1.BlockFloat64, 1, 2, 20, 30)
keys := index.Keys()
// 2 distinct keys
if got, exp := len(keys), 2; got != exp {
t.Fatalf("length mismatch: got %v, exp %v", got, exp)
}
// Keys should be sorted
if got, exp := keys[0], "cpu"; got != exp {
t.Fatalf("key mismatch: got %v, exp %v", got, exp)
}
if got, exp := keys[1], "mem"; got != exp {
t.Fatalf("key mismatch: got %v, exp %v", got, exp)
}
}
func TestIndirectIndex_Entries_NonExistent(t *testing.T) {
index := tsm1.NewIndexWriter()
index.Add("cpu", tsm1.BlockFloat64, 0, 1, 10, 100)
index.Add("cpu", tsm1.BlockFloat64, 2, 3, 20, 200)
b, err := index.MarshalBinary()
if err != nil {
t.Fatalf("unexpected error marshaling index: %v", err)
}
indirect := tsm1.NewIndirectIndex()
if err := indirect.UnmarshalBinary(b); err != nil {
t.Fatalf("unexpected error unmarshaling index: %v", err)
}
// mem has not been added to the index so we should get no entries back
// for both
exp := index.Entries("mem")
entries := indirect.Entries("mem")
if got, exp := len(entries), len(exp); got != exp && exp != 0 {
t.Fatalf("entries length mismatch: got %v, exp %v", got, exp)
}
}