2019-05-22 21:16:55 +00:00
package storage
import (
"bytes"
"fmt"
"io"
"math/rand"
"os"
"reflect"
2020-05-28 08:58:47 +00:00
"regexp"
2020-11-04 22:15:43 +00:00
"sort"
2019-05-22 21:16:55 +00:00
"testing"
"time"
"github.com/VictoriaMetrics/VictoriaMetrics/lib/bytesutil"
2019-10-08 13:25:24 +00:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/encoding"
2021-06-11 09:42:26 +00:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/fs"
2021-02-21 20:06:45 +00:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/mergeset"
2020-11-02 17:11:48 +00:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/uint64set"
2019-08-13 18:35:19 +00:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/workingsetcache"
2019-05-22 21:16:55 +00:00
)
2020-05-27 18:35:58 +00:00
func TestReverseBytes ( t * testing . T ) {
f := func ( s , resultExpected string ) {
t . Helper ( )
result := reverseBytes ( nil , [ ] byte ( s ) )
if string ( result ) != resultExpected {
t . Fatalf ( "unexpected result for reverseBytes(%q); got %q; want %q" , s , result , resultExpected )
}
}
f ( "" , "" )
f ( "a" , "a" )
f ( "av" , "va" )
f ( "foo.bar" , "rab.oof" )
}
2019-10-08 13:25:24 +00:00
func TestMergeTagToMetricIDsRows ( t * testing . T ) {
f := func ( items [ ] string , expectedItems [ ] string ) {
t . Helper ( )
var data [ ] byte
2021-02-21 20:06:45 +00:00
var itemsB [ ] mergeset . Item
2019-10-08 13:25:24 +00:00
for _ , item := range items {
data = append ( data , item ... )
2021-02-21 20:06:45 +00:00
itemsB = append ( itemsB , mergeset . Item {
Start : uint32 ( len ( data ) - len ( item ) ) ,
End : uint32 ( len ( data ) ) ,
} )
2019-10-08 13:25:24 +00:00
}
2021-02-21 20:06:45 +00:00
if ! checkItemsSorted ( data , itemsB ) {
2019-11-06 12:24:48 +00:00
t . Fatalf ( "source items aren't sorted; items:\n%q" , itemsB )
2019-10-08 13:25:24 +00:00
}
resultData , resultItemsB := mergeTagToMetricIDsRows ( data , itemsB )
if len ( resultItemsB ) != len ( expectedItems ) {
t . Fatalf ( "unexpected len(resultItemsB); got %d; want %d" , len ( resultItemsB ) , len ( expectedItems ) )
}
2021-02-21 20:06:45 +00:00
if ! checkItemsSorted ( resultData , resultItemsB ) {
2019-11-06 12:24:48 +00:00
t . Fatalf ( "result items aren't sorted; items:\n%q" , resultItemsB )
2019-10-08 13:25:24 +00:00
}
2021-02-21 20:06:45 +00:00
buf := resultData
for i , it := range resultItemsB {
item := it . Bytes ( resultData )
if ! bytes . HasPrefix ( buf , item ) {
t . Fatalf ( "unexpected prefix for resultData #%d;\ngot\n%X\nwant\n%X" , i , buf , item )
2019-10-08 13:25:24 +00:00
}
2021-02-21 20:06:45 +00:00
buf = buf [ len ( item ) : ]
2019-10-08 13:25:24 +00:00
}
2021-02-21 20:06:45 +00:00
if len ( buf ) != 0 {
t . Fatalf ( "unexpected tail left in resultData: %X" , buf )
2019-10-08 13:25:24 +00:00
}
var resultItems [ ] string
2021-02-21 20:06:45 +00:00
for _ , it := range resultItemsB {
resultItems = append ( resultItems , string ( it . Bytes ( resultData ) ) )
2019-10-08 13:25:24 +00:00
}
if ! reflect . DeepEqual ( expectedItems , resultItems ) {
t . Fatalf ( "unexpected items;\ngot\n%X\nwant\n%X" , resultItems , expectedItems )
}
}
2019-11-09 21:17:42 +00:00
xy := func ( nsPrefix byte , key , value string , metricIDs [ ] uint64 ) string {
dst := marshalCommonPrefix ( nil , nsPrefix )
if nsPrefix == nsPrefixDateTagToMetricIDs {
dst = encoding . MarshalUint64 ( dst , 1234567901233 )
}
2019-10-08 13:25:24 +00:00
t := & Tag {
Key : [ ] byte ( key ) ,
Value : [ ] byte ( value ) ,
}
dst = t . Marshal ( dst )
for _ , metricID := range metricIDs {
dst = encoding . MarshalUint64 ( dst , metricID )
}
return string ( dst )
}
2019-11-09 21:17:42 +00:00
x := func ( key , value string , metricIDs [ ] uint64 ) string {
return xy ( nsPrefixTagToMetricIDs , key , value , metricIDs )
}
y := func ( key , value string , metricIDs [ ] uint64 ) string {
return xy ( nsPrefixDateTagToMetricIDs , key , value , metricIDs )
}
2019-10-08 13:25:24 +00:00
f ( nil , nil )
f ( [ ] string { } , nil )
f ( [ ] string { "foo" } , [ ] string { "foo" } )
f ( [ ] string { "a" , "b" , "c" , "def" } , [ ] string { "a" , "b" , "c" , "def" } )
f ( [ ] string { "\x00" , "\x00b" , "\x00c" , "\x00def" } , [ ] string { "\x00" , "\x00b" , "\x00c" , "\x00def" } )
f ( [ ] string {
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
} , [ ] string {
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
} )
2019-11-09 21:17:42 +00:00
f ( [ ] string {
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
} , [ ] string {
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
} )
2019-10-08 13:25:24 +00:00
f ( [ ] string {
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
"xyz" ,
} , [ ] string {
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
"xyz" ,
} )
f ( [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
} , [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
} )
f ( [ ] string {
"\x00asdf" ,
2019-11-09 21:17:42 +00:00
y ( "" , "" , [ ] uint64 { 0 } ) ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
} , [ ] string {
"\x00asdf" ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
} )
f ( [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
2019-10-08 13:25:24 +00:00
x ( "" , "" , [ ] uint64 { 0 } ) ,
2019-11-09 21:17:42 +00:00
"xyz" ,
} , [ ] string {
"\x00asdf" ,
2019-10-08 13:25:24 +00:00
x ( "" , "" , [ ] uint64 { 0 } ) ,
2019-11-09 21:17:42 +00:00
"xyz" ,
} )
f ( [ ] string {
"\x00asdf" ,
2019-10-08 13:25:24 +00:00
x ( "" , "" , [ ] uint64 { 0 } ) ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
2019-11-09 21:17:42 +00:00
y ( "" , "" , [ ] uint64 { 0 } ) ,
y ( "" , "" , [ ] uint64 { 0 } ) ,
2019-10-08 13:25:24 +00:00
"xyz" ,
} , [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 0 } ) ,
2019-11-09 21:17:42 +00:00
y ( "" , "" , [ ] uint64 { 0 } ) ,
2019-10-08 13:25:24 +00:00
"xyz" ,
} )
f ( [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 1 } ) ,
x ( "" , "" , [ ] uint64 { 2 } ) ,
x ( "" , "" , [ ] uint64 { 3 } ) ,
x ( "" , "" , [ ] uint64 { 4 } ) ,
"xyz" ,
} , [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 1 , 2 , 3 , 4 } ) ,
"xyz" ,
} )
f ( [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 1 } ) ,
x ( "" , "" , [ ] uint64 { 2 } ) ,
x ( "" , "" , [ ] uint64 { 3 } ) ,
x ( "" , "" , [ ] uint64 { 4 } ) ,
} , [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 1 , 2 , 3 } ) ,
x ( "" , "" , [ ] uint64 { 4 } ) ,
} )
f ( [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 1 } ) ,
x ( "" , "" , [ ] uint64 { 2 , 3 , 4 } ) ,
x ( "" , "" , [ ] uint64 { 2 , 3 , 4 , 5 } ) ,
x ( "" , "" , [ ] uint64 { 3 , 5 } ) ,
"foo" ,
} , [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 1 , 2 , 3 , 4 , 5 } ) ,
"foo" ,
} )
f ( [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 1 } ) ,
x ( "" , "a" , [ ] uint64 { 2 , 3 , 4 } ) ,
x ( "" , "a" , [ ] uint64 { 2 , 3 , 4 , 5 } ) ,
x ( "" , "b" , [ ] uint64 { 3 , 5 } ) ,
"foo" ,
} , [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 1 } ) ,
x ( "" , "a" , [ ] uint64 { 2 , 3 , 4 , 5 } ) ,
x ( "" , "b" , [ ] uint64 { 3 , 5 } ) ,
"foo" ,
} )
f ( [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 1 } ) ,
x ( "x" , "a" , [ ] uint64 { 2 , 3 , 4 } ) ,
x ( "y" , "" , [ ] uint64 { 2 , 3 , 4 , 5 } ) ,
x ( "y" , "x" , [ ] uint64 { 3 , 5 } ) ,
"foo" ,
} , [ ] string {
"\x00asdf" ,
x ( "" , "" , [ ] uint64 { 1 } ) ,
x ( "x" , "a" , [ ] uint64 { 2 , 3 , 4 } ) ,
x ( "y" , "" , [ ] uint64 { 2 , 3 , 4 , 5 } ) ,
x ( "y" , "x" , [ ] uint64 { 3 , 5 } ) ,
"foo" ,
} )
f ( [ ] string {
"\x00asdf" ,
x ( "sdf" , "aa" , [ ] uint64 { 1 , 1 , 3 } ) ,
x ( "sdf" , "aa" , [ ] uint64 { 1 , 2 } ) ,
"foo" ,
} , [ ] string {
"\x00asdf" ,
x ( "sdf" , "aa" , [ ] uint64 { 1 , 2 , 3 } ) ,
"foo" ,
} )
f ( [ ] string {
"\x00asdf" ,
x ( "sdf" , "aa" , [ ] uint64 { 1 , 2 , 2 , 4 } ) ,
x ( "sdf" , "aa" , [ ] uint64 { 1 , 2 , 3 } ) ,
"foo" ,
} , [ ] string {
"\x00asdf" ,
x ( "sdf" , "aa" , [ ] uint64 { 1 , 2 , 3 , 4 } ) ,
"foo" ,
} )
// Construct big source chunks
var metricIDs [ ] uint64
metricIDs = metricIDs [ : 0 ]
for i := 0 ; i < maxMetricIDsPerRow - 1 ; i ++ {
metricIDs = append ( metricIDs , uint64 ( i ) )
}
f ( [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , metricIDs ) ,
2019-11-09 21:17:42 +00:00
y ( "foo" , "bar" , metricIDs ) ,
y ( "foo" , "bar" , metricIDs ) ,
2019-10-08 13:25:24 +00:00
"x" ,
} , [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
2019-11-09 21:17:42 +00:00
y ( "foo" , "bar" , metricIDs ) ,
2019-10-08 13:25:24 +00:00
"x" ,
} )
metricIDs = metricIDs [ : 0 ]
for i := 0 ; i < maxMetricIDsPerRow ; i ++ {
metricIDs = append ( metricIDs , uint64 ( i ) )
}
f ( [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , metricIDs ) ,
"x" ,
} , [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , metricIDs ) ,
"x" ,
} )
metricIDs = metricIDs [ : 0 ]
for i := 0 ; i < 3 * maxMetricIDsPerRow ; i ++ {
metricIDs = append ( metricIDs , uint64 ( i ) )
}
f ( [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , metricIDs ) ,
"x" ,
} , [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , metricIDs ) ,
"x" ,
} )
f ( [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , [ ] uint64 { 0 , 0 , 1 , 2 , 3 } ) ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , metricIDs ) ,
"x" ,
} , [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , [ ] uint64 { 0 , 1 , 2 , 3 } ) ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , metricIDs ) ,
"x" ,
} )
// Check for duplicate metricIDs removal
metricIDs = metricIDs [ : 0 ]
for i := 0 ; i < maxMetricIDsPerRow - 1 ; i ++ {
metricIDs = append ( metricIDs , 123 )
}
f ( [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , metricIDs ) ,
2019-11-09 21:17:42 +00:00
y ( "foo" , "bar" , metricIDs ) ,
2019-10-08 13:25:24 +00:00
"x" ,
} , [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , [ ] uint64 { 123 } ) ,
2019-11-09 21:17:42 +00:00
y ( "foo" , "bar" , [ ] uint64 { 123 } ) ,
2019-10-08 13:25:24 +00:00
"x" ,
} )
// Check fallback to the original items after merging, which result in incorrect ordering.
metricIDs = metricIDs [ : 0 ]
for i := 0 ; i < maxMetricIDsPerRow - 3 ; i ++ {
metricIDs = append ( metricIDs , uint64 ( 123 ) )
}
f ( [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 123 , 125 } ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 124 } ) ,
"x" ,
} , [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 123 , 125 } ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 124 } ) ,
"x" ,
} )
f ( [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 123 , 125 } ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 124 } ) ,
2019-11-09 21:17:42 +00:00
y ( "foo" , "bar" , [ ] uint64 { 123 , 124 } ) ,
2019-10-08 13:25:24 +00:00
} , [ ] string {
"\x00aa" ,
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 123 , 125 } ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 124 } ) ,
2019-11-09 21:17:42 +00:00
y ( "foo" , "bar" , [ ] uint64 { 123 , 124 } ) ,
2019-10-08 13:25:24 +00:00
} )
f ( [ ] string {
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 123 , 125 } ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 124 } ) ,
} , [ ] string {
x ( "foo" , "bar" , metricIDs ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 123 , 125 } ) ,
x ( "foo" , "bar" , [ ] uint64 { 123 , 124 } ) ,
} )
}
2019-09-25 14:55:13 +00:00
func TestRemoveDuplicateMetricIDs ( t * testing . T ) {
f := func ( metricIDs , expectedMetricIDs [ ] uint64 ) {
2019-10-08 13:25:24 +00:00
t . Helper ( )
2019-09-25 14:55:13 +00:00
a := removeDuplicateMetricIDs ( metricIDs )
if ! reflect . DeepEqual ( a , expectedMetricIDs ) {
t . Fatalf ( "unexpected result from removeDuplicateMetricIDs:\ngot\n%d\nwant\n%d" , a , expectedMetricIDs )
}
}
f ( nil , nil )
f ( [ ] uint64 { 123 } , [ ] uint64 { 123 } )
f ( [ ] uint64 { 123 , 123 } , [ ] uint64 { 123 } )
f ( [ ] uint64 { 123 , 123 , 123 } , [ ] uint64 { 123 } )
2019-09-25 15:23:13 +00:00
f ( [ ] uint64 { 123 , 1234 , 1235 } , [ ] uint64 { 123 , 1234 , 1235 } )
2019-09-25 14:55:13 +00:00
f ( [ ] uint64 { 0 , 1 , 1 , 2 } , [ ] uint64 { 0 , 1 , 2 } )
f ( [ ] uint64 { 0 , 0 , 0 , 1 , 1 , 2 } , [ ] uint64 { 0 , 1 , 2 } )
f ( [ ] uint64 { 0 , 1 , 1 , 2 , 2 } , [ ] uint64 { 0 , 1 , 2 } )
f ( [ ] uint64 { 0 , 1 , 2 , 2 } , [ ] uint64 { 0 , 1 , 2 } )
}
2019-05-22 21:16:55 +00:00
func TestMarshalUnmarshalTSIDs ( t * testing . T ) {
f := func ( tsids [ ] TSID ) {
t . Helper ( )
value := marshalTSIDs ( nil , tsids )
tsidsGot , err := unmarshalTSIDs ( nil , value )
if err != nil {
t . Fatalf ( "cannot unmarshal tsids: %s" , err )
}
if len ( tsids ) == 0 && len ( tsidsGot ) != 0 || len ( tsids ) > 0 && ! reflect . DeepEqual ( tsids , tsidsGot ) {
t . Fatalf ( "unexpected tsids unmarshaled\ngot\n%+v\nwant\n%+v" , tsidsGot , tsids )
}
// Try marshlaing with prefix
prefix := [ ] byte ( "prefix" )
valueExt := marshalTSIDs ( prefix , tsids )
if ! bytes . Equal ( valueExt [ : len ( prefix ) ] , prefix ) {
t . Fatalf ( "unexpected prefix after marshaling;\ngot\n%X\nwant\n%X" , valueExt [ : len ( prefix ) ] , prefix )
}
if ! bytes . Equal ( valueExt [ len ( prefix ) : ] , value ) {
t . Fatalf ( "unexpected prefixed marshaled value;\ngot\n%X\nwant\n%X" , valueExt [ len ( prefix ) : ] , value )
}
// Try unmarshaling with prefix
tsidPrefix := [ ] TSID { { MetricID : 123 } , { JobID : 456 } }
tsidsGot , err = unmarshalTSIDs ( tsidPrefix , value )
if err != nil {
t . Fatalf ( "cannot unmarshal prefixed tsids: %s" , err )
}
if ! reflect . DeepEqual ( tsidsGot [ : len ( tsidPrefix ) ] , tsidPrefix ) {
t . Fatalf ( "unexpected tsid prefix\ngot\n%+v\nwant\n%+v" , tsidsGot [ : len ( tsidPrefix ) ] , tsidPrefix )
}
if len ( tsids ) == 0 && len ( tsidsGot ) != len ( tsidPrefix ) || len ( tsids ) > 0 && ! reflect . DeepEqual ( tsidsGot [ len ( tsidPrefix ) : ] , tsids ) {
t . Fatalf ( "unexpected prefixed tsids unmarshaled\ngot\n%+v\nwant\n%+v" , tsidsGot [ len ( tsidPrefix ) : ] , tsids )
}
}
f ( nil )
f ( [ ] TSID { { MetricID : 123 } } )
f ( [ ] TSID { { JobID : 34 } , { MetricID : 2343 } , { InstanceID : 243321 } } )
}
func TestIndexDBOpenClose ( t * testing . T ) {
2021-06-11 09:42:26 +00:00
s := newTestStorage ( )
defer stopTestStorage ( s )
2019-06-25 17:09:57 +00:00
2019-05-22 21:16:55 +00:00
for i := 0 ; i < 5 ; i ++ {
2021-06-11 09:42:26 +00:00
db , err := openIndexDB ( "test-index-db" , s )
2019-05-22 21:16:55 +00:00
if err != nil {
t . Fatalf ( "cannot open indexDB: %s" , err )
}
db . MustClose ( )
}
if err := os . RemoveAll ( "test-index-db" ) ; err != nil {
t . Fatalf ( "cannot remove indexDB: %s" , err )
}
}
func TestIndexDB ( t * testing . T ) {
const metricGroups = 10
t . Run ( "serial" , func ( t * testing . T ) {
2021-06-11 09:42:26 +00:00
s := newTestStorage ( )
defer stopTestStorage ( s )
2019-06-25 17:09:57 +00:00
2019-05-22 21:16:55 +00:00
dbName := "test-index-db-serial"
2021-06-11 09:42:26 +00:00
db , err := openIndexDB ( dbName , s )
2019-05-22 21:16:55 +00:00
if err != nil {
t . Fatalf ( "cannot open indexDB: %s" , err )
}
defer func ( ) {
db . MustClose ( )
if err := os . RemoveAll ( dbName ) ; err != nil {
t . Fatalf ( "cannot remove indexDB: %s" , err )
}
} ( )
if err := testIndexDBBigMetricName ( db ) ; err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
2019-11-21 21:18:22 +00:00
mns , tsids , err := testIndexDBGetOrCreateTSIDByName ( db , metricGroups )
2019-05-22 21:16:55 +00:00
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if err := testIndexDBBigMetricName ( db ) ; err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if err := testIndexDBCheckTSIDByName ( db , mns , tsids , false ) ; err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if err := testIndexDBBigMetricName ( db ) ; err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
// Re-open the db and verify it works as expected.
db . MustClose ( )
2021-06-11 09:42:26 +00:00
db , err = openIndexDB ( dbName , s )
2019-05-22 21:16:55 +00:00
if err != nil {
t . Fatalf ( "cannot open indexDB: %s" , err )
}
if err := testIndexDBBigMetricName ( db ) ; err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if err := testIndexDBCheckTSIDByName ( db , mns , tsids , false ) ; err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if err := testIndexDBBigMetricName ( db ) ; err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
} )
t . Run ( "concurrent" , func ( t * testing . T ) {
2021-06-11 09:42:26 +00:00
s := newTestStorage ( )
defer stopTestStorage ( s )
2019-06-25 17:09:57 +00:00
2019-05-22 21:16:55 +00:00
dbName := "test-index-db-concurrent"
2021-06-11 09:42:26 +00:00
db , err := openIndexDB ( dbName , s )
2019-05-22 21:16:55 +00:00
if err != nil {
t . Fatalf ( "cannot open indexDB: %s" , err )
}
defer func ( ) {
db . MustClose ( )
if err := os . RemoveAll ( dbName ) ; err != nil {
t . Fatalf ( "cannot remove indexDB: %s" , err )
}
} ( )
ch := make ( chan error , 3 )
for i := 0 ; i < cap ( ch ) ; i ++ {
go func ( ) {
if err := testIndexDBBigMetricName ( db ) ; err != nil {
ch <- err
return
}
2019-11-21 21:18:22 +00:00
mns , tsid , err := testIndexDBGetOrCreateTSIDByName ( db , metricGroups )
2019-05-22 21:16:55 +00:00
if err != nil {
ch <- err
return
}
if err := testIndexDBBigMetricName ( db ) ; err != nil {
ch <- err
return
}
if err := testIndexDBCheckTSIDByName ( db , mns , tsid , true ) ; err != nil {
ch <- err
return
}
if err := testIndexDBBigMetricName ( db ) ; err != nil {
ch <- err
return
}
ch <- nil
} ( )
}
var errors [ ] error
for i := 0 ; i < cap ( ch ) ; i ++ {
select {
case err := <- ch :
if err != nil {
2020-06-30 19:58:18 +00:00
errors = append ( errors , fmt . Errorf ( "unexpected error: %w" , err ) )
2019-05-22 21:16:55 +00:00
}
case <- time . After ( 30 * time . Second ) :
t . Fatalf ( "timeout" )
}
}
if len ( errors ) > 0 {
t . Fatal ( errors [ 0 ] )
}
} )
}
func testIndexDBBigMetricName ( db * indexDB ) error {
var bigBytes [ ] byte
for i := 0 ; i < 128 * 1000 ; i ++ {
bigBytes = append ( bigBytes , byte ( i ) )
}
var mn MetricName
var tsid TSID
2020-07-23 17:42:57 +00:00
is := db . getIndexSearch ( noDeadline )
2019-05-22 21:16:55 +00:00
defer db . putIndexSearch ( is )
// Try creating too big metric group
mn . Reset ( )
mn . MetricGroup = append ( mn . MetricGroup [ : 0 ] , bigBytes ... )
mn . sortTags ( )
metricName := mn . Marshal ( nil )
if err := is . GetOrCreateTSIDByName ( & tsid , metricName ) ; err == nil {
return fmt . Errorf ( "expecting non-nil error on an attempt to insert metric with too big MetricGroup" )
}
// Try creating too big tag key
mn . Reset ( )
mn . MetricGroup = append ( mn . MetricGroup [ : 0 ] , "xxx" ... )
mn . Tags = [ ] Tag { {
Key : append ( [ ] byte ( nil ) , bigBytes ... ) ,
Value : [ ] byte ( "foobar" ) ,
} }
mn . sortTags ( )
metricName = mn . Marshal ( nil )
if err := is . GetOrCreateTSIDByName ( & tsid , metricName ) ; err == nil {
return fmt . Errorf ( "expecting non-nil error on an attempt to insert metric with too big tag key" )
}
// Try creating too big tag value
mn . Reset ( )
mn . MetricGroup = append ( mn . MetricGroup [ : 0 ] , "xxx" ... )
mn . Tags = [ ] Tag { {
Key : [ ] byte ( "foobar" ) ,
Value : append ( [ ] byte ( nil ) , bigBytes ... ) ,
} }
mn . sortTags ( )
metricName = mn . Marshal ( nil )
if err := is . GetOrCreateTSIDByName ( & tsid , metricName ) ; err == nil {
return fmt . Errorf ( "expecting non-nil error on an attempt to insert metric with too big tag value" )
}
// Try creating metric name with too many tags
mn . Reset ( )
mn . MetricGroup = append ( mn . MetricGroup [ : 0 ] , "xxx" ... )
for i := 0 ; i < 60000 ; i ++ {
mn . Tags = append ( mn . Tags , Tag {
Key : [ ] byte ( fmt . Sprintf ( "foobar %d" , i ) ) ,
Value : [ ] byte ( fmt . Sprintf ( "sdfjdslkfj %d" , i ) ) ,
} )
}
mn . sortTags ( )
metricName = mn . Marshal ( nil )
if err := is . GetOrCreateTSIDByName ( & tsid , metricName ) ; err == nil {
return fmt . Errorf ( "expecting non-nil error on an attempt to insert metric with too many tags" )
}
return nil
}
2019-11-21 21:18:22 +00:00
func testIndexDBGetOrCreateTSIDByName ( db * indexDB , metricGroups int ) ( [ ] MetricName , [ ] TSID , error ) {
2019-05-22 21:16:55 +00:00
// Create tsids.
var mns [ ] MetricName
var tsids [ ] TSID
2020-07-23 17:42:57 +00:00
is := db . getIndexSearch ( noDeadline )
2019-05-22 21:16:55 +00:00
defer db . putIndexSearch ( is )
2019-09-23 17:40:38 +00:00
var metricNameBuf [ ] byte
2019-05-22 21:16:55 +00:00
for i := 0 ; i < 4e2 + 1 ; i ++ {
var mn MetricName
// Init MetricGroup.
2020-05-27 18:35:58 +00:00
mn . MetricGroup = [ ] byte ( fmt . Sprintf ( "metricGroup.%d\x00\x01\x02" , i % metricGroups ) )
2019-05-22 21:16:55 +00:00
// Init other tags.
tagsCount := rand . Intn ( 10 ) + 1
for j := 0 ; j < tagsCount ; j ++ {
key := fmt . Sprintf ( "key\x01\x02\x00_%d_%d" , i , j )
value := fmt . Sprintf ( "val\x01_%d\x00_%d\x02" , i , j )
mn . AddTag ( key , value )
}
mn . sortTags ( )
2019-09-23 17:40:38 +00:00
metricNameBuf = mn . Marshal ( metricNameBuf [ : 0 ] )
2019-05-22 21:16:55 +00:00
// Create tsid for the metricName.
var tsid TSID
2019-09-23 17:40:38 +00:00
if err := is . GetOrCreateTSIDByName ( & tsid , metricNameBuf ) ; err != nil {
2020-06-30 19:58:18 +00:00
return nil , nil , fmt . Errorf ( "unexpected error when creating tsid for mn:\n%s: %w" , & mn , err )
2019-05-22 21:16:55 +00:00
}
mns = append ( mns , mn )
tsids = append ( tsids , tsid )
}
2019-06-25 09:55:27 +00:00
// fill Date -> MetricID cache
date := uint64 ( timestampFromTime ( time . Now ( ) ) ) / msecPerDay
for i := range tsids {
tsid := & tsids [ i ]
2021-05-23 13:39:55 +00:00
if err := is . storeDateMetricID ( date , tsid . MetricID , & mns [ i ] ) ; err != nil {
2020-06-30 19:58:18 +00:00
return nil , nil , fmt . Errorf ( "error in storeDateMetricID(%d, %d): %w" , date , tsid . MetricID , err )
2019-06-25 09:55:27 +00:00
}
}
2019-09-23 17:40:38 +00:00
// Flush index to disk, so it becomes visible for search
2019-06-25 09:55:27 +00:00
db . tb . DebugFlush ( )
2019-09-23 17:40:38 +00:00
return mns , tsids , nil
}
func testIndexDBCheckTSIDByName ( db * indexDB , mns [ ] MetricName , tsids [ ] TSID , isConcurrent bool ) error {
2019-05-22 21:16:55 +00:00
hasValue := func ( tvs [ ] string , v [ ] byte ) bool {
for _ , tv := range tvs {
if string ( v ) == tv {
return true
}
}
return false
}
timeseriesCounters := make ( map [ uint64 ] bool )
var tsidCopy TSID
var metricNameCopy [ ] byte
allKeys := make ( map [ string ] bool )
for i := range mns {
mn := & mns [ i ]
tsid := & tsids [ i ]
tc := timeseriesCounters
tc [ tsid . MetricID ] = true
mn . sortTags ( )
metricName := mn . Marshal ( nil )
if err := db . getTSIDByNameNoCreate ( & tsidCopy , metricName ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot obtain tsid #%d for mn %s: %w" , i , mn , err )
2019-05-22 21:16:55 +00:00
}
if isConcurrent {
// Copy tsid.MetricID, since multiple TSIDs may match
// the same mn in concurrent mode.
tsidCopy . MetricID = tsid . MetricID
}
if ! reflect . DeepEqual ( tsid , & tsidCopy ) {
return fmt . Errorf ( "unexpected tsid for mn:\n%s\ngot\n%+v\nwant\n%+v" , mn , & tsidCopy , tsid )
}
// Search for metric name for the given metricID.
var err error
2021-03-22 20:41:47 +00:00
metricNameCopy , err = db . searchMetricNameWithCache ( metricNameCopy [ : 0 ] , tsidCopy . MetricID )
2019-05-22 21:16:55 +00:00
if err != nil {
2021-03-22 20:41:47 +00:00
return fmt . Errorf ( "error in searchMetricNameWithCache for metricID=%d; i=%d: %w" , tsidCopy . MetricID , i , err )
2019-05-22 21:16:55 +00:00
}
if ! bytes . Equal ( metricName , metricNameCopy ) {
return fmt . Errorf ( "unexpected mn for metricID=%d;\ngot\n%q\nwant\n%q" , tsidCopy . MetricID , metricNameCopy , metricName )
}
// Try searching metric name for non-existent MetricID.
2021-03-22 20:41:47 +00:00
buf , err := db . searchMetricNameWithCache ( nil , 1 )
2019-05-22 21:16:55 +00:00
if err != io . EOF {
return fmt . Errorf ( "expecting io.EOF error when searching for non-existing metricID; got %v" , err )
}
if len ( buf ) > 0 {
return fmt . Errorf ( "expecting empty buf when searching for non-existent metricID; got %X" , buf )
}
// Test SearchTagValues
2020-07-23 17:42:57 +00:00
tvs , err := db . SearchTagValues ( nil , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "error in SearchTagValues for __name__: %w" , err )
2019-05-22 21:16:55 +00:00
}
if ! hasValue ( tvs , mn . MetricGroup ) {
return fmt . Errorf ( "SearchTagValues couldn't find %q; found %q" , mn . MetricGroup , tvs )
}
for i := range mn . Tags {
tag := & mn . Tags [ i ]
2020-07-23 17:42:57 +00:00
tvs , err := db . SearchTagValues ( tag . Key , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "error in SearchTagValues for __name__: %w" , err )
2019-05-22 21:16:55 +00:00
}
if ! hasValue ( tvs , tag . Value ) {
return fmt . Errorf ( "SearchTagValues couldn't find %q=%q; found %q" , tag . Key , tag . Value , tvs )
}
allKeys [ string ( tag . Key ) ] = true
}
}
// Test SearchTagKeys
2020-07-23 17:42:57 +00:00
tks , err := db . SearchTagKeys ( 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "error in SearchTagKeys: %w" , err )
2019-05-22 21:16:55 +00:00
}
if ! hasValue ( tks , nil ) {
return fmt . Errorf ( "cannot find __name__ in %q" , tks )
}
for key := range allKeys {
if ! hasValue ( tks , [ ] byte ( key ) ) {
return fmt . Errorf ( "cannot find %q in %q" , key , tks )
}
}
2019-06-10 09:36:42 +00:00
// Check timerseriesCounters only for serial test.
// Concurrent test may create duplicate timeseries, so GetSeriesCount
// would return more timeseries than needed.
if ! isConcurrent {
2020-07-23 17:42:57 +00:00
n , err := db . GetSeriesCount ( noDeadline )
2019-06-10 09:36:42 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "unexpected error in GetSeriesCount(): %w" , err )
2019-06-10 09:36:42 +00:00
}
if n != uint64 ( len ( timeseriesCounters ) ) {
return fmt . Errorf ( "unexpected GetSeriesCount(); got %d; want %d" , n , uint64 ( len ( timeseriesCounters ) ) )
}
}
2019-05-22 21:16:55 +00:00
// Try tag filters.
2020-10-01 16:03:34 +00:00
currentTime := timestampFromTime ( time . Now ( ) )
tr := TimeRange {
MinTimestamp : currentTime - msecPerDay ,
MaxTimestamp : currentTime + msecPerDay ,
}
2019-05-22 21:16:55 +00:00
for i := range mns {
mn := & mns [ i ]
tsid := & tsids [ i ]
// Search without regexps.
tfs := NewTagFilters ( )
if err := tfs . Add ( nil , mn . MetricGroup , false , false ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot create tag filter for MetricGroup: %w" , err )
2019-05-22 21:16:55 +00:00
}
for j := 0 ; j < len ( mn . Tags ) ; j ++ {
t := & mn . Tags [ j ]
if err := tfs . Add ( t . Key , t . Value , false , false ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot create tag filter for tag: %w" , err )
2019-05-22 21:16:55 +00:00
}
}
if err := tfs . Add ( nil , [ ] byte ( "foobar" ) , true , false ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot add negative filter: %w" , err )
2019-05-22 21:16:55 +00:00
}
if err := tfs . Add ( nil , nil , true , false ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot add no-op negative filter: %w" , err )
2019-05-22 21:16:55 +00:00
}
2020-10-01 16:03:34 +00:00
tsidsFound , err := db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search by exact tag filter: %w" , err )
2019-05-22 21:16:55 +00:00
}
if ! testHasTSID ( tsidsFound , tsid ) {
2019-09-23 17:40:38 +00:00
return fmt . Errorf ( "tsids is missing in exact tsidsFound\ntsid=%+v\ntsidsFound=%+v\ntfs=%s\nmn=%s\ni=%d" , tsid , tsidsFound , tfs , mn , i )
2019-05-22 21:16:55 +00:00
}
// Verify tag cache.
2020-10-01 16:03:34 +00:00
tsidsCached , err := db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search by exact tag filter: %w" , err )
2019-05-22 21:16:55 +00:00
}
if ! reflect . DeepEqual ( tsidsCached , tsidsFound ) {
return fmt . Errorf ( "unexpected tsids returned; got\n%+v; want\n%+v" , tsidsCached , tsidsFound )
}
// Add negative filter for zeroing search results.
if err := tfs . Add ( nil , mn . MetricGroup , true , false ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot add negative filter for zeroing search results: %w" , err )
2019-05-22 21:16:55 +00:00
}
2020-10-01 16:03:34 +00:00
tsidsFound , err = db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search by exact tag filter with full negative: %w" , err )
2019-05-22 21:16:55 +00:00
}
if testHasTSID ( tsidsFound , tsid ) {
return fmt . Errorf ( "unexpected tsid found for exact negative filter\ntsid=%+v\ntsidsFound=%+v\ntfs=%s\nmn=%s" , tsid , tsidsFound , tfs , mn )
}
2020-05-28 08:58:47 +00:00
// Search for Graphite wildcard
tfs . Reset ( )
n := bytes . IndexByte ( mn . MetricGroup , '.' )
if n < 0 {
return fmt . Errorf ( "cannot find dot in MetricGroup %q" , mn . MetricGroup )
}
re := "[^.]*" + regexp . QuoteMeta ( string ( mn . MetricGroup [ n : ] ) )
if err := tfs . Add ( nil , [ ] byte ( re ) , false , true ) ; err != nil {
return fmt . Errorf ( "cannot create regexp tag filter for Graphite wildcard" )
}
2020-10-01 16:03:34 +00:00
tsidsFound , err = db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
2020-05-28 08:58:47 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search by regexp tag filter for Graphite wildcard: %w" , err )
2020-05-28 08:58:47 +00:00
}
if ! testHasTSID ( tsidsFound , tsid ) {
return fmt . Errorf ( "tsids is missing in regexp for Graphite wildcard tsidsFound\ntsid=%+v\ntsidsFound=%+v\ntfs=%s\nmn=%s" , tsid , tsidsFound , tfs , mn )
}
2021-09-09 18:09:18 +00:00
// Search with a filter matching empty tag (a single filter)
// See https://github.com/VictoriaMetrics/VictoriaMetrics/issues/1601
tfs . Reset ( )
if err := tfs . Add ( nil , mn . MetricGroup , false , false ) ; err != nil {
return fmt . Errorf ( "cannot create tag filter for MetricGroup: %w" , err )
}
if err := tfs . Add ( [ ] byte ( "non-existent-tag" ) , [ ] byte ( "foo|" ) , false , true ) ; err != nil {
return fmt . Errorf ( "cannot create regexp tag filter for non-existing tag: %w" , err )
}
tsidsFound , err = db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
if err != nil {
return fmt . Errorf ( "cannot search with a filter matching empty tag: %w" , err )
}
2021-09-11 07:57:13 +00:00
if ! testHasTSID ( tsidsFound , tsid ) {
return fmt . Errorf ( "tsids is missing when matching a filter with empty tag tsidsFound\ntsid=%+v\ntsidsFound=%+v\ntfs=%s\nmn=%s" , tsid , tsidsFound , tfs , mn )
}
2021-09-09 18:09:18 +00:00
// Search with filters matching empty tags (multiple filters)
// See https://github.com/VictoriaMetrics/VictoriaMetrics/issues/1601
tfs . Reset ( )
if err := tfs . Add ( nil , mn . MetricGroup , false , false ) ; err != nil {
return fmt . Errorf ( "cannot create tag filter for MetricGroup: %w" , err )
}
if err := tfs . Add ( [ ] byte ( "non-existent-tag1" ) , [ ] byte ( "foo|" ) , false , true ) ; err != nil {
return fmt . Errorf ( "cannot create regexp tag filter for non-existing tag1: %w" , err )
}
if err := tfs . Add ( [ ] byte ( "non-existent-tag2" ) , [ ] byte ( "bar|" ) , false , true ) ; err != nil {
return fmt . Errorf ( "cannot create regexp tag filter for non-existing tag2: %w" , err )
}
tsidsFound , err = db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
if err != nil {
return fmt . Errorf ( "cannot search with multipel filters matching empty tags: %w" , err )
}
2021-09-11 07:57:13 +00:00
if ! testHasTSID ( tsidsFound , tsid ) {
return fmt . Errorf ( "tsids is missing when matching multiple filters with empty tags tsidsFound\ntsid=%+v\ntsidsFound=%+v\ntfs=%s\nmn=%s" , tsid , tsidsFound , tfs , mn )
}
2021-09-09 18:09:18 +00:00
2019-05-22 21:16:55 +00:00
// Search with regexps.
tfs . Reset ( )
if err := tfs . Add ( nil , mn . MetricGroup , false , true ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot create regexp tag filter for MetricGroup: %w" , err )
2019-05-22 21:16:55 +00:00
}
for j := 0 ; j < len ( mn . Tags ) ; j ++ {
t := & mn . Tags [ j ]
if err := tfs . Add ( t . Key , append ( t . Value , "|foo*." ... ) , false , true ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot create regexp tag filter for tag: %w" , err )
2019-05-22 21:16:55 +00:00
}
if err := tfs . Add ( t . Key , append ( t . Value , "|aaa|foo|bar" ... ) , false , true ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot create regexp tag filter for tag: %w" , err )
2019-05-22 21:16:55 +00:00
}
}
if err := tfs . Add ( nil , [ ] byte ( "^foobar$" ) , true , true ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot add negative filter with regexp: %w" , err )
2019-05-22 21:16:55 +00:00
}
if err := tfs . Add ( nil , nil , true , true ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot add no-op negative filter with regexp: %w" , err )
2019-05-22 21:16:55 +00:00
}
2020-10-01 16:03:34 +00:00
tsidsFound , err = db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search by regexp tag filter: %w" , err )
2019-05-22 21:16:55 +00:00
}
if ! testHasTSID ( tsidsFound , tsid ) {
return fmt . Errorf ( "tsids is missing in regexp tsidsFound\ntsid=%+v\ntsidsFound=%+v\ntfs=%s\nmn=%s" , tsid , tsidsFound , tfs , mn )
}
if err := tfs . Add ( nil , mn . MetricGroup , true , true ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot add negative filter for zeroing search results: %w" , err )
2019-05-22 21:16:55 +00:00
}
2020-10-01 16:03:34 +00:00
tsidsFound , err = db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search by regexp tag filter with full negative: %w" , err )
2019-05-22 21:16:55 +00:00
}
if testHasTSID ( tsidsFound , tsid ) {
return fmt . Errorf ( "unexpected tsid found for regexp negative filter\ntsid=%+v\ntsidsFound=%+v\ntfs=%s\nmn=%s" , tsid , tsidsFound , tfs , mn )
}
// Search with filter matching zero results.
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "non-existing-key" ) , [ ] byte ( "foobar" ) , false , false ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot add non-existing key: %w" , err )
2019-05-22 21:16:55 +00:00
}
if err := tfs . Add ( nil , mn . MetricGroup , false , true ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot create tag filter for MetricGroup matching zero results: %w" , err )
2019-05-22 21:16:55 +00:00
}
2020-10-01 16:03:34 +00:00
tsidsFound , err = db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search by non-existing tag filter: %w" , err )
2019-05-22 21:16:55 +00:00
}
if len ( tsidsFound ) > 0 {
return fmt . Errorf ( "non-zero tsidsFound for non-existing tag filter: %+v" , tsidsFound )
}
if isConcurrent {
// Skip empty filter search in concurrent mode, since it looks like
// it has a lag preventing from correct work.
continue
}
// Search with empty filter. It should match all the results.
tfs . Reset ( )
2020-10-01 16:03:34 +00:00
tsidsFound , err = db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search for common prefix: %w" , err )
2019-05-22 21:16:55 +00:00
}
if ! testHasTSID ( tsidsFound , tsid ) {
return fmt . Errorf ( "tsids is missing in common prefix\ntsid=%+v\ntsidsFound=%+v\ntfs=%s\nmn=%s" , tsid , tsidsFound , tfs , mn )
}
// Search with empty metricGroup. It should match zero results.
tfs . Reset ( )
if err := tfs . Add ( nil , nil , false , false ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot create tag filter for empty metricGroup: %w" , err )
2019-05-22 21:16:55 +00:00
}
2020-10-01 16:03:34 +00:00
tsidsFound , err = db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search for empty metricGroup: %w" , err )
2019-05-22 21:16:55 +00:00
}
if len ( tsidsFound ) != 0 {
return fmt . Errorf ( "unexpected non-empty tsids found for empty metricGroup: %v" , tsidsFound )
}
// Search with multiple tfss
tfs1 := NewTagFilters ( )
if err := tfs1 . Add ( nil , nil , false , false ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot create tag filter for empty metricGroup: %w" , err )
2019-05-22 21:16:55 +00:00
}
tfs2 := NewTagFilters ( )
if err := tfs2 . Add ( nil , mn . MetricGroup , false , false ) ; err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot create tag filter for MetricGroup: %w" , err )
2019-05-22 21:16:55 +00:00
}
2020-10-01 16:03:34 +00:00
tsidsFound , err = db . searchTSIDs ( [ ] * TagFilters { tfs1 , tfs2 } , tr , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search for empty metricGroup: %w" , err )
2019-05-22 21:16:55 +00:00
}
if ! testHasTSID ( tsidsFound , tsid ) {
return fmt . Errorf ( "tsids is missing when searching for multiple tfss \ntsid=%+v\ntsidsFound=%+v\ntfs=%s\nmn=%s" , tsid , tsidsFound , tfs , mn )
}
// Verify empty tfss
2020-10-01 16:03:34 +00:00
tsidsFound , err = db . searchTSIDs ( nil , tr , 1e5 , noDeadline )
2019-05-22 21:16:55 +00:00
if err != nil {
2020-06-30 19:58:18 +00:00
return fmt . Errorf ( "cannot search for nil tfss: %w" , err )
2019-05-22 21:16:55 +00:00
}
if len ( tsidsFound ) != 0 {
return fmt . Errorf ( "unexpected non-empty tsids fround for nil tfss; found %d tsids" , len ( tsidsFound ) )
}
}
return nil
}
func testHasTSID ( tsids [ ] TSID , tsid * TSID ) bool {
for i := range tsids {
if tsids [ i ] == * tsid {
return true
}
}
return false
}
func TestMatchTagFilters ( t * testing . T ) {
var mn MetricName
mn . MetricGroup = append ( mn . MetricGroup , "foobar_metric" ... )
for i := 0 ; i < 5 ; i ++ {
key := fmt . Sprintf ( "key %d" , i )
value := fmt . Sprintf ( "value %d" , i )
mn . AddTag ( key , value )
}
var bb bytesutil . ByteBuffer
var tfs TagFilters
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( "foobar_metric" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add filter: %s" , err )
}
ok , err := matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "should match" )
}
// Empty tag filters should match.
tfs . Reset ( )
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "empty tag filters should match" )
}
// Negative match by MetricGroup
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( "foobar" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, no negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( "obar.+" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, no negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( "foobar_metric" ) , true , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( "foob.+metric" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
2020-04-01 14:40:18 +00:00
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( ".+" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
2019-05-22 21:16:55 +00:00
// Positive match by MetricGroup
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( "foobar_metric" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, no negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( "foobar.+etric" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, no negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( "obar_metric" ) , true , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( "ob.+metric" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
2020-04-01 14:40:18 +00:00
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( ".+" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, positive filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
2019-05-22 21:16:55 +00:00
// Negative match by non-existing tag
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "non-existing-tag" ) , [ ] byte ( "foobar" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, no negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "non-existing-tag" ) , [ ] byte ( "obar.+" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, no negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "non-existing-tag" ) , [ ] byte ( "foobar_metric" ) , true , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
2019-07-30 12:14:09 +00:00
if ! ok {
t . Fatalf ( "Should match" )
2019-05-22 21:16:55 +00:00
}
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "non-existing-tag" ) , [ ] byte ( "foob.+metric" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
2019-07-30 12:14:09 +00:00
if ! ok {
t . Fatalf ( "Should match" )
}
2019-07-30 12:38:52 +00:00
tfs . Reset ( )
2019-07-30 12:14:09 +00:00
if err := tfs . Add ( [ ] byte ( "non-existing-tag" ) , [ ] byte ( ".+" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
2019-05-22 21:16:55 +00:00
}
2020-04-01 14:40:18 +00:00
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "non-existing-tag" ) , [ ] byte ( ".+" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, non-negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
2019-05-22 21:16:55 +00:00
// Negative match by existing tag
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 0" ) , [ ] byte ( "foobar" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, no negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 1" ) , [ ] byte ( "obar.+" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, no negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 2" ) , [ ] byte ( "value 2" ) , true , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 3" ) , [ ] byte ( "v.+lue 3" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
2020-04-01 14:40:18 +00:00
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 3" ) , [ ] byte ( ".+" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
2019-05-22 21:16:55 +00:00
2020-06-10 15:40:00 +00:00
// See https://github.com/VictoriaMetrics/VictoriaMetrics/issues/546
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 3" ) , [ ] byte ( "|value 3" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 3" ) , [ ] byte ( "|value 2" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
2019-05-22 21:16:55 +00:00
// Positive match by existing tag
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 0" ) , [ ] byte ( "value 0" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, no negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 1" ) , [ ] byte ( ".+lue 1" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, no negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 2" ) , [ ] byte ( "value 3" ) , true , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
tfs . Reset ( )
2020-04-01 14:40:18 +00:00
if err := tfs . Add ( [ ] byte ( "key 3" ) , [ ] byte ( "v.+lue 2|" ) , true , true ) ; err != nil {
2019-05-22 21:16:55 +00:00
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
2019-07-30 12:14:09 +00:00
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
2019-07-30 12:38:52 +00:00
tfs . Reset ( )
2019-07-30 12:14:09 +00:00
if err := tfs . Add ( [ ] byte ( "key 3" ) , [ ] byte ( "" ) , true , false ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
2019-05-22 21:16:55 +00:00
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
2020-04-01 14:40:18 +00:00
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 3" ) , [ ] byte ( ".+" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, non-negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
2019-05-22 21:16:55 +00:00
// Positive match by multiple tags and MetricGroup
tfs . Reset ( )
if err := tfs . Add ( [ ] byte ( "key 0" ) , [ ] byte ( "value 0" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, no negative filter: %s" , err )
}
if err := tfs . Add ( [ ] byte ( "key 2" ) , [ ] byte ( "value [0-9]" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, no negative filter: %s" , err )
}
if err := tfs . Add ( [ ] byte ( "key 3" ) , [ ] byte ( "value 23" ) , true , false ) ; err != nil {
t . Fatalf ( "cannt add no regexp, negative filter: %s" , err )
}
if err := tfs . Add ( [ ] byte ( "key 2" ) , [ ] byte ( "lue.+43" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
if err := tfs . Add ( nil , [ ] byte ( "foobar_metric" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, no negative filter: %s" , err )
}
if err := tfs . Add ( nil , [ ] byte ( "foo.+metric" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, no negative filter: %s" , err )
}
if err := tfs . Add ( nil , [ ] byte ( "sdfdsf" ) , true , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, negative filter: %s" , err )
}
if err := tfs . Add ( nil , [ ] byte ( "o.+metr" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! ok {
t . Fatalf ( "Should match" )
}
// Negative match by multiple tags and MetricGroup
tfs . Reset ( )
// Positive matches
if err := tfs . Add ( [ ] byte ( "key 0" ) , [ ] byte ( "value 0" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, no negative filter: %s" , err )
}
if err := tfs . Add ( [ ] byte ( "key 2" ) , [ ] byte ( "value [0-9]" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, no negative filter: %s" , err )
}
if err := tfs . Add ( [ ] byte ( "key 3" ) , [ ] byte ( "value 23" ) , true , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, negative filter: %s" , err )
}
// Negative matches
if err := tfs . Add ( [ ] byte ( "key 2" ) , [ ] byte ( "v.+2" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
if err := tfs . Add ( nil , [ ] byte ( "obar_metric" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, no negative filter: %s" , err )
}
if err := tfs . Add ( nil , [ ] byte ( "oo.+metric" ) , false , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, no negative filter: %s" , err )
}
// Positive matches
if err := tfs . Add ( nil , [ ] byte ( "sdfdsf" ) , true , false ) ; err != nil {
t . Fatalf ( "cannot add no regexp, negative filter: %s" , err )
}
if err := tfs . Add ( nil , [ ] byte ( "o.+metr" ) , true , true ) ; err != nil {
t . Fatalf ( "cannot add regexp, negative filter: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
2020-04-01 14:40:18 +00:00
// Negative match for multiple non-regexp positive filters
tfs . Reset ( )
if err := tfs . Add ( nil , [ ] byte ( "foobar_metric" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add non-regexp positive filter for MetricGroup: %s" , err )
}
if err := tfs . Add ( [ ] byte ( "non-existing-metric" ) , [ ] byte ( "foobar" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add non-regexp positive filter for non-existing tag: %s" , err )
}
ok , err = matchTagFilters ( & mn , toTFPointers ( tfs . tfs ) , & bb )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ok {
t . Fatalf ( "Shouldn't match" )
}
2019-05-22 21:16:55 +00:00
}
2019-11-12 13:09:33 +00:00
func TestSearchTSIDWithTimeRange ( t * testing . T ) {
2021-06-11 09:42:26 +00:00
s := newTestStorage ( )
defer stopTestStorage ( s )
2019-11-12 13:09:33 +00:00
dbName := "test-index-db-ts-range"
2021-06-11 09:42:26 +00:00
db , err := openIndexDB ( dbName , s )
2019-11-12 13:09:33 +00:00
if err != nil {
t . Fatalf ( "cannot open indexDB: %s" , err )
}
defer func ( ) {
db . MustClose ( )
if err := os . RemoveAll ( dbName ) ; err != nil {
t . Fatalf ( "cannot remove indexDB: %s" , err )
}
} ( )
2020-07-23 17:42:57 +00:00
is := db . getIndexSearch ( noDeadline )
2019-11-12 13:09:33 +00:00
defer db . putIndexSearch ( is )
// Create a bunch of per-day time series
const days = 5
const metricsPerDay = 1000
theDay := time . Date ( 2019 , time . October , 15 , 5 , 1 , 0 , 0 , time . UTC )
now := uint64 ( timestampFromTime ( theDay ) )
baseDate := now / msecPerDay
var metricNameBuf [ ] byte
2020-11-02 17:11:48 +00:00
perDayMetricIDs := make ( map [ uint64 ] * uint64set . Set )
var allMetricIDs uint64set . Set
2020-11-04 22:15:43 +00:00
tagKeys := [ ] string {
"" , "constant" , "day" , "uniqueid" ,
}
tagValues := [ ] string {
"testMetric" ,
}
sort . Strings ( tagKeys )
2019-11-12 13:09:33 +00:00
for day := 0 ; day < days ; day ++ {
var tsids [ ] TSID
2021-05-23 13:39:55 +00:00
var mns [ ] MetricName
2019-11-12 13:09:33 +00:00
for metric := 0 ; metric < metricsPerDay ; metric ++ {
var mn MetricName
mn . MetricGroup = [ ] byte ( "testMetric" )
mn . AddTag (
"constant" ,
"const" ,
)
mn . AddTag (
"day" ,
fmt . Sprintf ( "%v" , day ) ,
)
mn . AddTag (
"uniqueid" ,
fmt . Sprintf ( "%v" , metric ) ,
)
mn . sortTags ( )
metricNameBuf = mn . Marshal ( metricNameBuf [ : 0 ] )
var tsid TSID
if err := is . GetOrCreateTSIDByName ( & tsid , metricNameBuf ) ; err != nil {
t . Fatalf ( "unexpected error when creating tsid for mn:\n%s: %s" , & mn , err )
}
2021-05-23 13:39:55 +00:00
mns = append ( mns , mn )
2019-11-12 13:09:33 +00:00
tsids = append ( tsids , tsid )
}
// Add the metrics to the per-day stores
2020-10-01 16:03:34 +00:00
date := baseDate - uint64 ( day )
2020-11-02 17:11:48 +00:00
var metricIDs uint64set . Set
2019-11-12 13:09:33 +00:00
for i := range tsids {
tsid := & tsids [ i ]
2020-11-02 17:11:48 +00:00
metricIDs . Add ( tsid . MetricID )
2021-05-23 13:39:55 +00:00
if err := is . storeDateMetricID ( date , tsid . MetricID , & mns [ i ] ) ; err != nil {
2019-11-12 13:09:33 +00:00
t . Fatalf ( "error in storeDateMetricID(%d, %d): %s" , date , tsid . MetricID , err )
}
}
2020-11-02 17:11:48 +00:00
allMetricIDs . Union ( & metricIDs )
perDayMetricIDs [ date ] = & metricIDs
2019-11-12 13:09:33 +00:00
}
// Flush index to disk, so it becomes visible for search
db . tb . DebugFlush ( )
2020-11-02 17:11:48 +00:00
is2 := db . getIndexSearch ( noDeadline )
defer db . putIndexSearch ( is2 )
// Check that all the metrics are found for all the days.
for date := baseDate - days + 1 ; date <= baseDate ; date ++ {
metricIDs , err := is2 . getMetricIDsForDate ( date , metricsPerDay )
if err != nil {
t . Fatalf ( "unexpected error: %s" , err )
}
if ! perDayMetricIDs [ date ] . Equal ( metricIDs ) {
t . Fatalf ( "unexpected metricIDs found;\ngot\n%d\nwant\n%d" , metricIDs . AppendTo ( nil ) , perDayMetricIDs [ date ] . AppendTo ( nil ) )
}
}
2021-07-30 05:37:10 +00:00
// Check that all the metrics are found in global index
metricIDs , err := is2 . getMetricIDsForDate ( 0 , metricsPerDay * days )
if err != nil {
2020-11-02 17:11:48 +00:00
t . Fatalf ( "unexpected error: %s" , err )
}
2021-07-30 05:37:10 +00:00
if ! allMetricIDs . Equal ( metricIDs ) {
2020-11-02 17:11:48 +00:00
t . Fatalf ( "unexpected metricIDs found;\ngot\n%d\nwant\n%d" , metricIDs . AppendTo ( nil ) , allMetricIDs . AppendTo ( nil ) )
}
2020-11-04 22:15:43 +00:00
// Check SearchTagKeysOnTimeRange.
tks , err := db . SearchTagKeysOnTimeRange ( TimeRange {
MinTimestamp : int64 ( now ) - msecPerDay ,
MaxTimestamp : int64 ( now ) ,
} , 10000 , noDeadline )
if err != nil {
t . Fatalf ( "unexpected error in SearchTagKeysOnTimeRange: %s" , err )
}
sort . Strings ( tks )
if ! reflect . DeepEqual ( tks , tagKeys ) {
t . Fatalf ( "unexpected tagKeys; got\n%s\nwant\n%s" , tks , tagKeys )
}
// Check SearchTagValuesOnTimeRange.
tvs , err := db . SearchTagValuesOnTimeRange ( [ ] byte ( "" ) , TimeRange {
2020-11-04 23:36:13 +00:00
MinTimestamp : int64 ( now ) - msecPerDay ,
2020-11-04 22:15:43 +00:00
MaxTimestamp : int64 ( now ) ,
} , 10000 , noDeadline )
if err != nil {
t . Fatalf ( "unexpected error in SearchTagValuesOnTimeRange: %s" , err )
}
sort . Strings ( tvs )
if ! reflect . DeepEqual ( tvs , tagValues ) {
t . Fatalf ( "unexpected tagValues; got\n%s\nwant\n%s" , tvs , tagValues )
}
2019-11-12 13:09:33 +00:00
// Create a filter that will match series that occur across multiple days
tfs := NewTagFilters ( )
if err := tfs . Add ( [ ] byte ( "constant" ) , [ ] byte ( "const" ) , false , false ) ; err != nil {
t . Fatalf ( "cannot add filter: %s" , err )
}
2020-10-01 16:03:34 +00:00
// Perform a search within a day.
2019-11-12 13:09:33 +00:00
// This should return the metrics for the day
2020-10-01 16:03:34 +00:00
tr := TimeRange {
2019-11-12 13:09:33 +00:00
MinTimestamp : int64 ( now - 2 * msecPerHour - 1 ) ,
MaxTimestamp : int64 ( now ) ,
}
2020-10-01 16:03:34 +00:00
matchedTSIDs , err := db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 10000 , noDeadline )
2019-11-12 13:09:33 +00:00
if err != nil {
t . Fatalf ( "error searching tsids: %v" , err )
}
if len ( matchedTSIDs ) != metricsPerDay {
2020-10-01 16:03:34 +00:00
t . Fatalf ( "expected %d time series for current day, got %d time series" , metricsPerDay , len ( matchedTSIDs ) )
2019-11-12 13:09:33 +00:00
}
// Perform a search across all the days, should match all metrics
tr = TimeRange {
MinTimestamp : int64 ( now - msecPerDay * days ) ,
MaxTimestamp : int64 ( now ) ,
}
2020-07-23 17:42:57 +00:00
matchedTSIDs , err = db . searchTSIDs ( [ ] * TagFilters { tfs } , tr , 10000 , noDeadline )
2019-11-12 13:09:33 +00:00
if err != nil {
t . Fatalf ( "error searching tsids: %v" , err )
}
if len ( matchedTSIDs ) != metricsPerDay * days {
2020-10-01 16:03:34 +00:00
t . Fatalf ( "expected %d time series for all days, got %d time series" , metricsPerDay * days , len ( matchedTSIDs ) )
2019-11-12 13:09:33 +00:00
}
2020-04-22 16:57:36 +00:00
2021-05-13 05:56:02 +00:00
// Check GetTSDBStatusWithFiltersForDate with nil filters.
status , err := db . GetTSDBStatusWithFiltersForDate ( nil , baseDate , 5 , noDeadline )
2020-04-22 16:57:36 +00:00
if err != nil {
2021-05-13 05:56:02 +00:00
t . Fatalf ( "error in GetTSDBStatusWithFiltersForDate with nil filters: %s" , err )
2020-04-22 16:57:36 +00:00
}
if ! status . hasEntries ( ) {
t . Fatalf ( "expecting non-empty TSDB status" )
}
expectedSeriesCountByMetricName := [ ] TopHeapEntry {
{
Name : "testMetric" ,
Count : 1000 ,
} ,
}
if ! reflect . DeepEqual ( status . SeriesCountByMetricName , expectedSeriesCountByMetricName ) {
t . Fatalf ( "unexpected SeriesCountByMetricName;\ngot\n%v\nwant\n%v" , status . SeriesCountByMetricName , expectedSeriesCountByMetricName )
}
expectedLabelValueCountByLabelName := [ ] TopHeapEntry {
{
Name : "uniqueid" ,
Count : 1000 ,
} ,
{
Name : "__name__" ,
Count : 1 ,
} ,
{
Name : "constant" ,
Count : 1 ,
} ,
{
Name : "day" ,
Count : 1 ,
} ,
}
if ! reflect . DeepEqual ( status . LabelValueCountByLabelName , expectedLabelValueCountByLabelName ) {
t . Fatalf ( "unexpected LabelValueCountByLabelName;\ngot\n%v\nwant\n%v" , status . LabelValueCountByLabelName , expectedLabelValueCountByLabelName )
}
expectedSeriesCountByLabelValuePair := [ ] TopHeapEntry {
{
Name : "__name__=testMetric" ,
Count : 1000 ,
} ,
{
Name : "constant=const" ,
Count : 1000 ,
} ,
{
Name : "day=0" ,
Count : 1000 ,
} ,
{
Name : "uniqueid=0" ,
Count : 1 ,
} ,
{
Name : "uniqueid=1" ,
Count : 1 ,
} ,
}
if ! reflect . DeepEqual ( status . SeriesCountByLabelValuePair , expectedSeriesCountByLabelValuePair ) {
t . Fatalf ( "unexpected SeriesCountByLabelValuePair;\ngot\n%v\nwant\n%v" , status . SeriesCountByLabelValuePair , expectedSeriesCountByLabelValuePair )
}
2021-05-12 12:18:45 +00:00
2021-05-12 13:32:48 +00:00
// Check GetTSDBStatusWithFiltersForDate
2021-05-12 12:18:45 +00:00
tfs = NewTagFilters ( )
2021-05-12 13:32:48 +00:00
if err := tfs . Add ( [ ] byte ( "day" ) , [ ] byte ( "0" ) , false , false ) ; err != nil {
2021-05-12 12:18:45 +00:00
t . Fatalf ( "cannot add filter: %s" , err )
}
2021-05-12 13:32:48 +00:00
status , err = db . GetTSDBStatusWithFiltersForDate ( [ ] * TagFilters { tfs } , baseDate , 5 , noDeadline )
2021-05-12 12:18:45 +00:00
if err != nil {
2021-05-12 13:32:48 +00:00
t . Fatalf ( "error in GetTSDBStatusWithFiltersForDate: %s" , err )
2021-05-12 12:18:45 +00:00
}
if ! status . hasEntries ( ) {
t . Fatalf ( "expecting non-empty TSDB status" )
}
expectedSeriesCountByMetricName = [ ] TopHeapEntry {
{
Name : "testMetric" ,
Count : 1000 ,
} ,
}
if ! reflect . DeepEqual ( status . SeriesCountByMetricName , expectedSeriesCountByMetricName ) {
t . Fatalf ( "unexpected SeriesCountByMetricName;\ngot\n%v\nwant\n%v" , status . SeriesCountByMetricName , expectedSeriesCountByMetricName )
}
2019-11-12 13:09:33 +00:00
}
2019-05-22 21:16:55 +00:00
func toTFPointers ( tfs [ ] tagFilter ) [ ] * tagFilter {
tfps := make ( [ ] * tagFilter , len ( tfs ) )
for i := range tfs {
tfps [ i ] = & tfs [ i ]
}
return tfps
}
2021-06-11 09:42:26 +00:00
func newTestStorage ( ) * Storage {
2021-06-15 11:56:51 +00:00
s := & Storage {
2021-06-11 09:42:26 +00:00
cachePath : "test-storage-cache" ,
metricIDCache : workingsetcache . New ( 1234 , time . Hour ) ,
metricNameCache : workingsetcache . New ( 1234 , time . Hour ) ,
tsidCache : workingsetcache . New ( 1234 , time . Hour ) ,
}
2021-06-15 11:56:51 +00:00
s . setDeletedMetricIDs ( & uint64set . Set { } )
return s
2021-06-11 09:42:26 +00:00
}
func stopTestStorage ( s * Storage ) {
s . metricIDCache . Stop ( )
s . metricNameCache . Stop ( )
s . tsidCache . Stop ( )
fs . MustRemoveAll ( s . cachePath )
}