122 lines
4.1 KiB
Go
122 lines
4.1 KiB
Go
package tsdb_test
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/influxdata/influxdb/internal"
|
|
"github.com/influxdata/influxdb/models"
|
|
"github.com/influxdata/influxdb/pkg/slices"
|
|
"github.com/influxdata/influxdb/tsdb"
|
|
"github.com/influxdata/influxdb/tsdb/index/inmem"
|
|
"github.com/influxdata/influxql"
|
|
)
|
|
|
|
func TestIndex_MeasurementNamesByExpr(t *testing.T) {
|
|
// Setup indexes
|
|
indexes := map[string]*Index{}
|
|
for _, name := range tsdb.RegisteredIndexes() {
|
|
idx := NewIndex(name)
|
|
idx.AddSeries("cpu", map[string]string{"region": "east"})
|
|
idx.AddSeries("cpu", map[string]string{"region": "west", "secret": "foo"})
|
|
idx.AddSeries("disk", map[string]string{"secret": "foo"})
|
|
idx.AddSeries("mem", map[string]string{"region": "west"})
|
|
idx.AddSeries("gpu", map[string]string{"region": "east"})
|
|
idx.AddSeries("pci", map[string]string{"region": "east", "secret": "foo"})
|
|
indexes[name] = idx
|
|
}
|
|
|
|
authorizer := &internal.AuthorizerMock{
|
|
AuthorizeSeriesReadFn: func(database string, measurement []byte, tags models.Tags) bool {
|
|
if tags.GetString("secret") != "" {
|
|
t.Logf("Rejecting series db=%s, m=%s, tags=%v", database, measurement, tags)
|
|
return false
|
|
}
|
|
return true
|
|
},
|
|
}
|
|
|
|
type example struct {
|
|
name string
|
|
expr influxql.Expr
|
|
expected [][]byte
|
|
}
|
|
|
|
// These examples should be run without any auth.
|
|
examples := []example{
|
|
{name: "all", expected: slices.StringsToBytes("cpu", "disk", "gpu", "mem", "pci")},
|
|
{name: "EQ", expr: influxql.MustParseExpr(`region = 'west'`), expected: slices.StringsToBytes("cpu", "mem")},
|
|
{name: "NEQ", expr: influxql.MustParseExpr(`region != 'west'`), expected: slices.StringsToBytes("gpu", "pci")},
|
|
{name: "EQREGEX", expr: influxql.MustParseExpr(`region =~ /.*st/`), expected: slices.StringsToBytes("cpu", "gpu", "mem", "pci")},
|
|
{name: "NEQREGEX", expr: influxql.MustParseExpr(`region !~ /.*est/`), expected: slices.StringsToBytes("gpu", "pci")},
|
|
}
|
|
|
|
// These examples should be run with the authorizer.
|
|
authExamples := []example{
|
|
{name: "all", expected: slices.StringsToBytes("cpu", "gpu", "mem")},
|
|
{name: "EQ", expr: influxql.MustParseExpr(`region = 'west'`), expected: slices.StringsToBytes("mem")},
|
|
{name: "NEQ", expr: influxql.MustParseExpr(`region != 'west'`), expected: slices.StringsToBytes("gpu")},
|
|
{name: "EQREGEX", expr: influxql.MustParseExpr(`region =~ /.*st/`), expected: slices.StringsToBytes("cpu", "gpu", "mem")},
|
|
{name: "NEQREGEX", expr: influxql.MustParseExpr(`region !~ /.*est/`), expected: slices.StringsToBytes("gpu")},
|
|
}
|
|
|
|
for _, idx := range tsdb.RegisteredIndexes() {
|
|
t.Run(idx, func(t *testing.T) {
|
|
t.Run("no authorization", func(t *testing.T) {
|
|
for _, example := range examples {
|
|
t.Run(example.name, func(t *testing.T) {
|
|
names, err := indexes[idx].MeasurementNamesByExpr(nil, example.expr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
} else if !reflect.DeepEqual(names, example.expected) {
|
|
t.Fatalf("got names: %v, expected %v", slices.BytesToStrings(names), slices.BytesToStrings(example.expected))
|
|
}
|
|
})
|
|
}
|
|
})
|
|
|
|
t.Run("with authorization", func(t *testing.T) {
|
|
for _, example := range authExamples {
|
|
t.Run(example.name, func(t *testing.T) {
|
|
names, err := indexes[idx].MeasurementNamesByExpr(authorizer, example.expr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
} else if !reflect.DeepEqual(names, example.expected) {
|
|
t.Fatalf("got names: %v, expected %v", slices.BytesToStrings(names), slices.BytesToStrings(example.expected))
|
|
}
|
|
})
|
|
}
|
|
})
|
|
})
|
|
}
|
|
}
|
|
|
|
type Index struct {
|
|
tsdb.Index
|
|
}
|
|
|
|
func NewIndex(index string) *Index {
|
|
opts := tsdb.NewEngineOptions()
|
|
opts.IndexVersion = index
|
|
|
|
if index == inmem.IndexName {
|
|
opts.InmemIndex = inmem.NewIndex("db0")
|
|
}
|
|
|
|
path, err := ioutil.TempDir("", "influxdb-tsdb")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
idx := &Index{Index: tsdb.MustOpenIndex(0, "db0", filepath.Join(path, "index"), opts)}
|
|
return idx
|
|
}
|
|
|
|
func (idx *Index) AddSeries(name string, tags map[string]string) error {
|
|
t := models.NewTags(tags)
|
|
key := fmt.Sprintf("%s,%s", name, t.HashKey())
|
|
return idx.CreateSeriesIfNotExists([]byte(key), []byte(name), t)
|
|
}
|