mirror of https://github.com/grafana/loki
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1293 lines
32 KiB
1293 lines
32 KiB
package storage
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
_ "net/http/pprof"
|
|
"path"
|
|
"runtime"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/cespare/xxhash/v2"
|
|
"github.com/prometheus/common/model"
|
|
"github.com/prometheus/prometheus/model/labels"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/weaveworks/common/user"
|
|
|
|
"github.com/grafana/dskit/flagext"
|
|
|
|
"github.com/grafana/loki/pkg/iter"
|
|
"github.com/grafana/loki/pkg/logproto"
|
|
"github.com/grafana/loki/pkg/logql"
|
|
"github.com/grafana/loki/pkg/querier/astmapper"
|
|
"github.com/grafana/loki/pkg/storage/chunk"
|
|
chunk_local "github.com/grafana/loki/pkg/storage/chunk/local"
|
|
"github.com/grafana/loki/pkg/storage/chunk/storage"
|
|
"github.com/grafana/loki/pkg/storage/stores/shipper"
|
|
util_log "github.com/grafana/loki/pkg/util/log"
|
|
"github.com/grafana/loki/pkg/util/marshal"
|
|
"github.com/grafana/loki/pkg/validation"
|
|
)
|
|
|
|
var (
|
|
start = model.Time(1523750400000)
|
|
m runtime.MemStats
|
|
ctx = user.InjectOrgID(context.Background(), "fake")
|
|
cm = storage.NewClientMetrics()
|
|
chunkStore = getLocalStore(cm)
|
|
)
|
|
|
|
// go test -bench=. -benchmem -memprofile memprofile.out -cpuprofile profile.out
|
|
func Benchmark_store_SelectLogsRegexBackward(b *testing.B) {
|
|
benchmarkStoreQuery(b, &logproto.QueryRequest{
|
|
Selector: `{foo="bar"} |~ "fuzz"`,
|
|
Limit: 1000,
|
|
Start: time.Unix(0, start.UnixNano()),
|
|
End: time.Unix(0, (24*time.Hour.Nanoseconds())+start.UnixNano()),
|
|
Direction: logproto.BACKWARD,
|
|
})
|
|
}
|
|
|
|
func Benchmark_store_SelectLogsLogQLBackward(b *testing.B) {
|
|
benchmarkStoreQuery(b, &logproto.QueryRequest{
|
|
Selector: `{foo="bar"} |= "test" != "toto" |= "fuzz"`,
|
|
Limit: 1000,
|
|
Start: time.Unix(0, start.UnixNano()),
|
|
End: time.Unix(0, (24*time.Hour.Nanoseconds())+start.UnixNano()),
|
|
Direction: logproto.BACKWARD,
|
|
})
|
|
}
|
|
|
|
func Benchmark_store_SelectLogsRegexForward(b *testing.B) {
|
|
benchmarkStoreQuery(b, &logproto.QueryRequest{
|
|
Selector: `{foo="bar"} |~ "fuzz"`,
|
|
Limit: 1000,
|
|
Start: time.Unix(0, start.UnixNano()),
|
|
End: time.Unix(0, (24*time.Hour.Nanoseconds())+start.UnixNano()),
|
|
Direction: logproto.FORWARD,
|
|
})
|
|
}
|
|
|
|
func Benchmark_store_SelectLogsForward(b *testing.B) {
|
|
benchmarkStoreQuery(b, &logproto.QueryRequest{
|
|
Selector: `{foo="bar"}`,
|
|
Limit: 1000,
|
|
Start: time.Unix(0, start.UnixNano()),
|
|
End: time.Unix(0, (24*time.Hour.Nanoseconds())+start.UnixNano()),
|
|
Direction: logproto.FORWARD,
|
|
})
|
|
}
|
|
|
|
func Benchmark_store_SelectLogsBackward(b *testing.B) {
|
|
benchmarkStoreQuery(b, &logproto.QueryRequest{
|
|
Selector: `{foo="bar"}`,
|
|
Limit: 1000,
|
|
Start: time.Unix(0, start.UnixNano()),
|
|
End: time.Unix(0, (24*time.Hour.Nanoseconds())+start.UnixNano()),
|
|
Direction: logproto.BACKWARD,
|
|
})
|
|
}
|
|
|
|
// rm -Rf /tmp/benchmark/chunks/ /tmp/benchmark/index
|
|
// go run -mod=vendor ./pkg/storage/hack/main.go
|
|
// go test -benchmem -run=^$ -mod=vendor ./pkg/storage -bench=Benchmark_store_SelectSample -memprofile memprofile.out -cpuprofile cpuprofile.out
|
|
func Benchmark_store_SelectSample(b *testing.B) {
|
|
var sampleRes []logproto.Sample
|
|
for _, test := range []string{
|
|
`count_over_time({foo="bar"}[5m])`,
|
|
`rate({foo="bar"}[5m])`,
|
|
`bytes_rate({foo="bar"}[5m])`,
|
|
`bytes_over_time({foo="bar"}[5m])`,
|
|
} {
|
|
b.Run(test, func(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
iter, err := chunkStore.SelectSamples(ctx, logql.SelectSampleParams{
|
|
SampleQueryRequest: newSampleQuery(test, time.Unix(0, start.UnixNano()), time.Unix(0, (24*time.Hour.Nanoseconds())+start.UnixNano())),
|
|
})
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
|
|
for iter.Next() {
|
|
sampleRes = append(sampleRes, iter.Sample())
|
|
}
|
|
iter.Close()
|
|
}
|
|
})
|
|
}
|
|
log.Print("sample processed ", len(sampleRes))
|
|
}
|
|
|
|
func benchmarkStoreQuery(b *testing.B, query *logproto.QueryRequest) {
|
|
b.ReportAllocs()
|
|
// force to run gc 10x more often this can be useful to detect fast allocation vs leak.
|
|
// debug.SetGCPercent(10)
|
|
stop := make(chan struct{})
|
|
go func() {
|
|
_ = http.ListenAndServe(":6060", http.DefaultServeMux)
|
|
}()
|
|
go func() {
|
|
ticker := time.NewTicker(time.Millisecond)
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
// print and capture the max in use heap size
|
|
printHeap(b, false)
|
|
case <-stop:
|
|
ticker.Stop()
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
for i := 0; i < b.N; i++ {
|
|
iter, err := chunkStore.SelectLogs(ctx, logql.SelectLogParams{QueryRequest: query})
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
res := []logproto.Entry{}
|
|
printHeap(b, true)
|
|
j := uint32(0)
|
|
for iter.Next() {
|
|
j++
|
|
printHeap(b, false)
|
|
res = append(res, iter.Entry())
|
|
// limit result like the querier would do.
|
|
if j == query.Limit {
|
|
break
|
|
}
|
|
}
|
|
iter.Close()
|
|
printHeap(b, true)
|
|
log.Println("line fetched", len(res))
|
|
}
|
|
close(stop)
|
|
}
|
|
|
|
var maxHeapInuse uint64
|
|
|
|
func printHeap(b *testing.B, show bool) {
|
|
runtime.ReadMemStats(&m)
|
|
if m.HeapInuse > maxHeapInuse {
|
|
maxHeapInuse = m.HeapInuse
|
|
}
|
|
if show {
|
|
log.Printf("Benchmark %d maxHeapInuse: %d Mbytes\n", b.N, maxHeapInuse/1024/1024)
|
|
log.Printf("Benchmark %d currentHeapInuse: %d Mbytes\n", b.N, m.HeapInuse/1024/1024)
|
|
}
|
|
}
|
|
|
|
func getLocalStore(cm storage.ClientMetrics) Store {
|
|
limits, err := validation.NewOverrides(validation.Limits{
|
|
MaxQueryLength: model.Duration(6000 * time.Hour),
|
|
}, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
storeConfig := Config{
|
|
Config: storage.Config{
|
|
BoltDBConfig: chunk_local.BoltDBConfig{Directory: "/tmp/benchmark/index"},
|
|
FSConfig: chunk_local.FSConfig{Directory: "/tmp/benchmark/chunks"},
|
|
},
|
|
MaxChunkBatchSize: 10,
|
|
}
|
|
|
|
schemaConfig := SchemaConfig{
|
|
chunk.SchemaConfig{
|
|
Configs: []chunk.PeriodConfig{
|
|
{
|
|
From: chunk.DayTime{Time: start},
|
|
IndexType: "boltdb",
|
|
ObjectType: "filesystem",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Prefix: "index_",
|
|
Period: time.Hour * 168,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
chunkStore, err := storage.NewStore(
|
|
storeConfig.Config,
|
|
chunk.StoreConfig{},
|
|
schemaConfig.SchemaConfig, limits, cm, nil, nil, util_log.Logger)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
store, err := NewStore(storeConfig, schemaConfig, chunkStore, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return store
|
|
}
|
|
|
|
func Test_store_SelectLogs(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
req *logproto.QueryRequest
|
|
expected []logproto.Stream
|
|
}{
|
|
{
|
|
"all",
|
|
newQuery("{foo=~\"ba.*\"}", from, from.Add(6*time.Millisecond), nil),
|
|
[]logproto.Stream{
|
|
{
|
|
Labels: "{foo=\"bar\"}",
|
|
Entries: []logproto.Entry{
|
|
{
|
|
Timestamp: from,
|
|
Line: "1",
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(time.Millisecond),
|
|
Line: "2",
|
|
},
|
|
{
|
|
Timestamp: from.Add(2 * time.Millisecond),
|
|
Line: "3",
|
|
},
|
|
{
|
|
Timestamp: from.Add(3 * time.Millisecond),
|
|
Line: "4",
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(4 * time.Millisecond),
|
|
Line: "5",
|
|
},
|
|
{
|
|
Timestamp: from.Add(5 * time.Millisecond),
|
|
Line: "6",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Labels: "{foo=\"bazz\"}",
|
|
Entries: []logproto.Entry{
|
|
{
|
|
Timestamp: from,
|
|
Line: "1",
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(time.Millisecond),
|
|
Line: "2",
|
|
},
|
|
{
|
|
Timestamp: from.Add(2 * time.Millisecond),
|
|
Line: "3",
|
|
},
|
|
{
|
|
Timestamp: from.Add(3 * time.Millisecond),
|
|
Line: "4",
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(4 * time.Millisecond),
|
|
Line: "5",
|
|
},
|
|
{
|
|
Timestamp: from.Add(5 * time.Millisecond),
|
|
Line: "6",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"filter regex",
|
|
newQuery("{foo=~\"ba.*\"} |~ \"1|2|3\" !~ \"2|3\"", from, from.Add(6*time.Millisecond), nil),
|
|
[]logproto.Stream{
|
|
{
|
|
Labels: "{foo=\"bar\"}",
|
|
Entries: []logproto.Entry{
|
|
{
|
|
Timestamp: from,
|
|
Line: "1",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Labels: "{foo=\"bazz\"}",
|
|
Entries: []logproto.Entry{
|
|
{
|
|
Timestamp: from,
|
|
Line: "1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"filter matcher",
|
|
newQuery("{foo=\"bar\"}", from, from.Add(6*time.Millisecond), nil),
|
|
[]logproto.Stream{
|
|
{
|
|
Labels: "{foo=\"bar\"}",
|
|
Entries: []logproto.Entry{
|
|
{
|
|
Timestamp: from,
|
|
Line: "1",
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(time.Millisecond),
|
|
Line: "2",
|
|
},
|
|
{
|
|
Timestamp: from.Add(2 * time.Millisecond),
|
|
Line: "3",
|
|
},
|
|
{
|
|
Timestamp: from.Add(3 * time.Millisecond),
|
|
Line: "4",
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(4 * time.Millisecond),
|
|
Line: "5",
|
|
},
|
|
{
|
|
Timestamp: from.Add(5 * time.Millisecond),
|
|
Line: "6",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"filter time",
|
|
newQuery("{foo=~\"ba.*\"}", from, from.Add(time.Millisecond), nil),
|
|
[]logproto.Stream{
|
|
{
|
|
Labels: "{foo=\"bar\"}",
|
|
Entries: []logproto.Entry{
|
|
{
|
|
Timestamp: from,
|
|
Line: "1",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Labels: "{foo=\"bazz\"}",
|
|
Entries: []logproto.Entry{
|
|
{
|
|
Timestamp: from,
|
|
Line: "1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s := &store{
|
|
Store: storeFixture,
|
|
cfg: Config{
|
|
MaxChunkBatchSize: 10,
|
|
},
|
|
chunkMetrics: NilMetrics,
|
|
}
|
|
|
|
ctx = user.InjectOrgID(context.Background(), "test-user")
|
|
it, err := s.SelectLogs(ctx, logql.SelectLogParams{QueryRequest: tt.req})
|
|
if err != nil {
|
|
t.Errorf("store.LazyQuery() error = %v", err)
|
|
return
|
|
}
|
|
|
|
streams, _, err := iter.ReadBatch(it, tt.req.Limit)
|
|
_ = it.Close()
|
|
if err != nil {
|
|
t.Fatalf("error reading batch %s", err)
|
|
}
|
|
assertStream(t, tt.expected, streams.Streams)
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_store_SelectSample(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
req *logproto.SampleQueryRequest
|
|
expected []logproto.Series
|
|
}{
|
|
{
|
|
"all",
|
|
newSampleQuery("count_over_time({foo=~\"ba.*\"}[5m])", from, from.Add(6*time.Millisecond)),
|
|
[]logproto.Series{
|
|
{
|
|
Labels: "{foo=\"bar\"}",
|
|
Samples: []logproto.Sample{
|
|
{
|
|
Timestamp: from.UnixNano(),
|
|
Hash: xxhash.Sum64String("1"),
|
|
Value: 1.,
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("2"),
|
|
Value: 1.,
|
|
},
|
|
{
|
|
Timestamp: from.Add(2 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("3"),
|
|
Value: 1.,
|
|
},
|
|
{
|
|
Timestamp: from.Add(3 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("4"),
|
|
Value: 1.,
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(4 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("5"),
|
|
Value: 1.,
|
|
},
|
|
{
|
|
Timestamp: from.Add(5 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("6"),
|
|
Value: 1.,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Labels: "{foo=\"bazz\"}",
|
|
Samples: []logproto.Sample{
|
|
{
|
|
Timestamp: from.UnixNano(),
|
|
Hash: xxhash.Sum64String("1"),
|
|
Value: 1.,
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("2"),
|
|
Value: 1.,
|
|
},
|
|
{
|
|
Timestamp: from.Add(2 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("3"),
|
|
Value: 1.,
|
|
},
|
|
{
|
|
Timestamp: from.Add(3 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("4"),
|
|
Value: 1.,
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(4 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("5"),
|
|
Value: 1.,
|
|
},
|
|
{
|
|
Timestamp: from.Add(5 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("6"),
|
|
Value: 1.,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"filter regex",
|
|
newSampleQuery("rate({foo=~\"ba.*\"} |~ \"1|2|3\" !~ \"2|3\"[1m])", from, from.Add(6*time.Millisecond)),
|
|
[]logproto.Series{
|
|
{
|
|
Labels: "{foo=\"bar\"}",
|
|
Samples: []logproto.Sample{
|
|
{
|
|
Timestamp: from.UnixNano(),
|
|
Hash: xxhash.Sum64String("1"),
|
|
Value: 1.,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Labels: "{foo=\"bazz\"}",
|
|
Samples: []logproto.Sample{
|
|
{
|
|
Timestamp: from.UnixNano(),
|
|
Hash: xxhash.Sum64String("1"),
|
|
Value: 1.,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"filter matcher",
|
|
newSampleQuery("count_over_time({foo=\"bar\"}[10m])", from, from.Add(6*time.Millisecond)),
|
|
[]logproto.Series{
|
|
{
|
|
Labels: "{foo=\"bar\"}",
|
|
Samples: []logproto.Sample{
|
|
{
|
|
Timestamp: from.UnixNano(),
|
|
Hash: xxhash.Sum64String("1"),
|
|
Value: 1.,
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("2"),
|
|
Value: 1.,
|
|
},
|
|
{
|
|
Timestamp: from.Add(2 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("3"),
|
|
Value: 1.,
|
|
},
|
|
{
|
|
Timestamp: from.Add(3 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("4"),
|
|
Value: 1.,
|
|
},
|
|
|
|
{
|
|
Timestamp: from.Add(4 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("5"),
|
|
Value: 1.,
|
|
},
|
|
{
|
|
Timestamp: from.Add(5 * time.Millisecond).UnixNano(),
|
|
Hash: xxhash.Sum64String("6"),
|
|
Value: 1.,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"filter time",
|
|
newSampleQuery("count_over_time({foo=~\"ba.*\"}[1s])", from, from.Add(time.Millisecond)),
|
|
[]logproto.Series{
|
|
{
|
|
Labels: "{foo=\"bar\"}",
|
|
Samples: []logproto.Sample{
|
|
{
|
|
Timestamp: from.UnixNano(),
|
|
Hash: xxhash.Sum64String("1"),
|
|
Value: 1.,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Labels: "{foo=\"bazz\"}",
|
|
Samples: []logproto.Sample{
|
|
{
|
|
Timestamp: from.UnixNano(),
|
|
Hash: xxhash.Sum64String("1"),
|
|
Value: 1.,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s := &store{
|
|
Store: storeFixture,
|
|
cfg: Config{
|
|
MaxChunkBatchSize: 10,
|
|
},
|
|
chunkMetrics: NilMetrics,
|
|
}
|
|
|
|
ctx = user.InjectOrgID(context.Background(), "test-user")
|
|
it, err := s.SelectSamples(ctx, logql.SelectSampleParams{SampleQueryRequest: tt.req})
|
|
if err != nil {
|
|
t.Errorf("store.LazyQuery() error = %v", err)
|
|
return
|
|
}
|
|
|
|
series, _, err := iter.ReadSampleBatch(it, uint32(100000))
|
|
_ = it.Close()
|
|
if err != nil {
|
|
t.Fatalf("error reading batch %s", err)
|
|
}
|
|
assertSeries(t, tt.expected, series.Series)
|
|
})
|
|
}
|
|
}
|
|
|
|
type fakeChunkFilterer struct{}
|
|
|
|
func (f fakeChunkFilterer) ForRequest(ctx context.Context) ChunkFilterer {
|
|
return f
|
|
}
|
|
|
|
func (f fakeChunkFilterer) ShouldFilter(metric labels.Labels) bool {
|
|
return metric.Get("foo") == "bazz"
|
|
}
|
|
|
|
func Test_ChunkFilterer(t *testing.T) {
|
|
s := &store{
|
|
Store: storeFixture,
|
|
cfg: Config{
|
|
MaxChunkBatchSize: 10,
|
|
},
|
|
chunkMetrics: NilMetrics,
|
|
}
|
|
s.SetChunkFilterer(&fakeChunkFilterer{})
|
|
ctx = user.InjectOrgID(context.Background(), "test-user")
|
|
it, err := s.SelectSamples(ctx, logql.SelectSampleParams{SampleQueryRequest: newSampleQuery("count_over_time({foo=~\"ba.*\"}[1s])", from, from.Add(1*time.Hour))})
|
|
if err != nil {
|
|
t.Errorf("store.SelectSamples() error = %v", err)
|
|
return
|
|
}
|
|
defer it.Close()
|
|
for it.Next() {
|
|
v := mustParseLabels(it.Labels())["foo"]
|
|
require.NotEqual(t, "bazz", v)
|
|
}
|
|
|
|
logit, err := s.SelectLogs(ctx, logql.SelectLogParams{QueryRequest: newQuery("{foo=~\"ba.*\"}", from, from.Add(1*time.Hour), nil)})
|
|
if err != nil {
|
|
t.Errorf("store.SelectLogs() error = %v", err)
|
|
return
|
|
}
|
|
defer logit.Close()
|
|
for logit.Next() {
|
|
v := mustParseLabels(it.Labels())["foo"]
|
|
require.NotEqual(t, "bazz", v)
|
|
}
|
|
ids, err := s.GetSeries(ctx, logql.SelectLogParams{QueryRequest: newQuery("{foo=~\"ba.*\"}", from, from.Add(1*time.Hour), nil)})
|
|
require.NoError(t, err)
|
|
for _, id := range ids {
|
|
v := id.Labels["foo"]
|
|
require.NotEqual(t, "bazz", v)
|
|
}
|
|
}
|
|
|
|
func Test_store_GetSeries(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
req *logproto.QueryRequest
|
|
expected []logproto.SeriesIdentifier
|
|
batchSize int
|
|
}{
|
|
{
|
|
"all",
|
|
newQuery("{foo=~\"ba.*\"}", from, from.Add(6*time.Millisecond), nil),
|
|
[]logproto.SeriesIdentifier{
|
|
{Labels: mustParseLabels("{foo=\"bar\"}")},
|
|
{Labels: mustParseLabels("{foo=\"bazz\"}")},
|
|
},
|
|
1,
|
|
},
|
|
{
|
|
"all-single-batch",
|
|
newQuery("{foo=~\"ba.*\"}", from, from.Add(6*time.Millisecond), nil),
|
|
[]logproto.SeriesIdentifier{
|
|
{Labels: mustParseLabels("{foo=\"bar\"}")},
|
|
{Labels: mustParseLabels("{foo=\"bazz\"}")},
|
|
},
|
|
5,
|
|
},
|
|
{
|
|
"regexp filter (post chunk fetching)",
|
|
newQuery("{foo=~\"bar.*\"}", from, from.Add(6*time.Millisecond), nil),
|
|
[]logproto.SeriesIdentifier{
|
|
{Labels: mustParseLabels("{foo=\"bar\"}")},
|
|
},
|
|
1,
|
|
},
|
|
{
|
|
"filter matcher",
|
|
newQuery("{foo=\"bar\"}", from, from.Add(6*time.Millisecond), nil),
|
|
[]logproto.SeriesIdentifier{
|
|
{Labels: mustParseLabels("{foo=\"bar\"}")},
|
|
},
|
|
1,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s := &store{
|
|
Store: storeFixture,
|
|
cfg: Config{
|
|
MaxChunkBatchSize: tt.batchSize,
|
|
},
|
|
chunkMetrics: NilMetrics,
|
|
}
|
|
ctx = user.InjectOrgID(context.Background(), "test-user")
|
|
out, err := s.GetSeries(ctx, logql.SelectLogParams{QueryRequest: tt.req})
|
|
if err != nil {
|
|
t.Errorf("store.GetSeries() error = %v", err)
|
|
return
|
|
}
|
|
require.Equal(t, tt.expected, out)
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_store_decodeReq_Matchers(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
req *logproto.QueryRequest
|
|
matchers []*labels.Matcher
|
|
}{
|
|
{
|
|
"unsharded",
|
|
newQuery("{foo=~\"ba.*\"}", from, from.Add(6*time.Millisecond), nil),
|
|
[]*labels.Matcher{
|
|
labels.MustNewMatcher(labels.MatchRegexp, "foo", "ba.*"),
|
|
labels.MustNewMatcher(labels.MatchEqual, labels.MetricName, "logs"),
|
|
},
|
|
},
|
|
{
|
|
"sharded",
|
|
newQuery(
|
|
"{foo=~\"ba.*\"}", from, from.Add(6*time.Millisecond),
|
|
[]astmapper.ShardAnnotation{
|
|
{Shard: 1, Of: 2},
|
|
},
|
|
),
|
|
[]*labels.Matcher{
|
|
labels.MustNewMatcher(labels.MatchRegexp, "foo", "ba.*"),
|
|
labels.MustNewMatcher(labels.MatchEqual, labels.MetricName, "logs"),
|
|
labels.MustNewMatcher(
|
|
labels.MatchEqual,
|
|
astmapper.ShardLabel,
|
|
astmapper.ShardAnnotation{Shard: 1, Of: 2}.String(),
|
|
),
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
ms, _, _, err := decodeReq(logql.SelectLogParams{QueryRequest: tt.req})
|
|
if err != nil {
|
|
t.Errorf("store.GetSeries() error = %v", err)
|
|
return
|
|
}
|
|
require.Equal(t, tt.matchers, ms)
|
|
})
|
|
}
|
|
}
|
|
|
|
type timeRange struct {
|
|
from, to time.Time
|
|
}
|
|
|
|
func TestStore_MultipleBoltDBShippersInConfig(t *testing.T) {
|
|
tempDir := t.TempDir()
|
|
|
|
limits, err := validation.NewOverrides(validation.Limits{}, nil)
|
|
require.NoError(t, err)
|
|
|
|
// config for BoltDB Shipper
|
|
boltdbShipperConfig := shipper.Config{}
|
|
flagext.DefaultValues(&boltdbShipperConfig)
|
|
boltdbShipperConfig.ActiveIndexDirectory = path.Join(tempDir, "index")
|
|
boltdbShipperConfig.SharedStoreType = "filesystem"
|
|
boltdbShipperConfig.CacheLocation = path.Join(tempDir, "boltdb-shipper-cache")
|
|
|
|
// dates for activation of boltdb shippers
|
|
firstStoreDate := parseDate("2019-01-01")
|
|
secondStoreDate := parseDate("2019-01-02")
|
|
|
|
config := Config{
|
|
Config: storage.Config{
|
|
FSConfig: chunk_local.FSConfig{Directory: path.Join(tempDir, "chunks")},
|
|
},
|
|
BoltDBShipperConfig: boltdbShipperConfig,
|
|
}
|
|
|
|
schemaConfig := SchemaConfig{
|
|
chunk.SchemaConfig{
|
|
Configs: []chunk.PeriodConfig{
|
|
{
|
|
From: chunk.DayTime{Time: timeToModelTime(firstStoreDate)},
|
|
IndexType: "boltdb-shipper",
|
|
ObjectType: "filesystem",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Prefix: "index_",
|
|
Period: time.Hour * 168,
|
|
},
|
|
},
|
|
{
|
|
From: chunk.DayTime{Time: timeToModelTime(secondStoreDate)},
|
|
IndexType: "boltdb-shipper",
|
|
ObjectType: "filesystem",
|
|
Schema: "v11",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Prefix: "index_",
|
|
Period: time.Hour * 168,
|
|
},
|
|
RowShards: 2,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
RegisterCustomIndexClients(&config, cm, nil)
|
|
|
|
chunkStore, err := storage.NewStore(
|
|
config.Config,
|
|
chunk.StoreConfig{},
|
|
schemaConfig.SchemaConfig,
|
|
limits,
|
|
cm,
|
|
nil,
|
|
nil,
|
|
util_log.Logger,
|
|
)
|
|
require.NoError(t, err)
|
|
store, err := NewStore(config, schemaConfig, chunkStore, nil)
|
|
require.NoError(t, err)
|
|
|
|
// time ranges adding a chunk for each store and a chunk which overlaps both the stores
|
|
chunksToBuildForTimeRanges := []timeRange{
|
|
{
|
|
// chunk just for first store
|
|
secondStoreDate.Add(-3 * time.Hour),
|
|
secondStoreDate.Add(-2 * time.Hour),
|
|
},
|
|
{
|
|
// chunk overlapping both the stores
|
|
secondStoreDate.Add(-time.Hour),
|
|
secondStoreDate.Add(time.Hour),
|
|
},
|
|
{
|
|
// chunk just for second store
|
|
secondStoreDate.Add(2 * time.Hour),
|
|
secondStoreDate.Add(3 * time.Hour),
|
|
},
|
|
}
|
|
|
|
// build and add chunks to the store
|
|
addedChunkIDs := map[string]struct{}{}
|
|
for _, tr := range chunksToBuildForTimeRanges {
|
|
chk := newChunk(buildTestStreams(fooLabelsWithName, tr))
|
|
|
|
err := store.PutOne(ctx, chk.From, chk.Through, chk)
|
|
require.NoError(t, err)
|
|
|
|
addedChunkIDs[schemaConfig.ExternalKey(chk)] = struct{}{}
|
|
}
|
|
|
|
// recreate the store because boltdb-shipper now runs queriers on snapshots which are created every 1 min and during startup.
|
|
store.Stop()
|
|
|
|
chunkStore, err = storage.NewStore(
|
|
config.Config,
|
|
chunk.StoreConfig{},
|
|
schemaConfig.SchemaConfig,
|
|
limits,
|
|
cm,
|
|
nil,
|
|
nil,
|
|
util_log.Logger,
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
store, err = NewStore(config, schemaConfig, chunkStore, nil)
|
|
require.NoError(t, err)
|
|
|
|
defer store.Stop()
|
|
|
|
// get all the chunks from both the stores
|
|
chunks, _, err := store.GetChunkRefs(ctx, "fake", timeToModelTime(firstStoreDate), timeToModelTime(secondStoreDate.Add(24*time.Hour)), newMatchers(fooLabelsWithName.String())...)
|
|
require.NoError(t, err)
|
|
var totalChunks int
|
|
for _, chks := range chunks {
|
|
totalChunks += len(chks)
|
|
}
|
|
// we get common chunk twice because it is indexed in both the stores
|
|
require.Equal(t, totalChunks, len(addedChunkIDs)+1)
|
|
|
|
// check whether we got back all the chunks which were added
|
|
for i := range chunks {
|
|
for _, c := range chunks[i] {
|
|
_, ok := addedChunkIDs[schemaConfig.ExternalKey(c)]
|
|
require.True(t, ok)
|
|
}
|
|
}
|
|
}
|
|
|
|
func mustParseLabels(s string) map[string]string {
|
|
l, err := marshal.NewLabelSet(s)
|
|
if err != nil {
|
|
log.Fatalf("Failed to parse %s", s)
|
|
}
|
|
|
|
return l
|
|
}
|
|
|
|
func parseDate(in string) time.Time {
|
|
t, err := time.Parse("2006-01-02", in)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return t
|
|
}
|
|
|
|
func buildTestStreams(labels labels.Labels, tr timeRange) logproto.Stream {
|
|
stream := logproto.Stream{
|
|
Labels: labels.String(),
|
|
Hash: labels.Hash(),
|
|
Entries: []logproto.Entry{},
|
|
}
|
|
|
|
for from := tr.from; from.Before(tr.to); from = from.Add(time.Second) {
|
|
stream.Entries = append(stream.Entries, logproto.Entry{
|
|
Timestamp: from,
|
|
Line: from.String(),
|
|
})
|
|
}
|
|
|
|
return stream
|
|
}
|
|
|
|
func timeToModelTime(t time.Time) model.Time {
|
|
return model.TimeFromUnixNano(t.UnixNano())
|
|
}
|
|
|
|
func TestActiveIndexType(t *testing.T) {
|
|
var cfg SchemaConfig
|
|
|
|
// just one PeriodConfig in the past
|
|
cfg.Configs = []chunk.PeriodConfig{{
|
|
From: chunk.DayTime{Time: model.Now().Add(-24 * time.Hour)},
|
|
IndexType: "first",
|
|
}}
|
|
|
|
assert.Equal(t, 0, ActivePeriodConfig(cfg.Configs))
|
|
|
|
// add a newer PeriodConfig in the past which should be considered
|
|
cfg.Configs = append(cfg.Configs, chunk.PeriodConfig{
|
|
From: chunk.DayTime{Time: model.Now().Add(-12 * time.Hour)},
|
|
IndexType: "second",
|
|
})
|
|
assert.Equal(t, 1, ActivePeriodConfig(cfg.Configs))
|
|
|
|
// add a newer PeriodConfig in the future which should not be considered
|
|
cfg.Configs = append(cfg.Configs, chunk.PeriodConfig{
|
|
From: chunk.DayTime{Time: model.Now().Add(time.Hour)},
|
|
IndexType: "third",
|
|
})
|
|
assert.Equal(t, 1, ActivePeriodConfig(cfg.Configs))
|
|
}
|
|
|
|
func TestUsingBoltdbShipper(t *testing.T) {
|
|
var cfg SchemaConfig
|
|
|
|
// just one PeriodConfig in the past using boltdb-shipper
|
|
cfg.Configs = []chunk.PeriodConfig{{
|
|
From: chunk.DayTime{Time: model.Now().Add(-24 * time.Hour)},
|
|
IndexType: "boltdb-shipper",
|
|
}}
|
|
assert.Equal(t, true, UsingBoltdbShipper(cfg.Configs))
|
|
|
|
// just one PeriodConfig in the past not using boltdb-shipper
|
|
cfg.Configs[0].IndexType = "boltdb"
|
|
assert.Equal(t, false, UsingBoltdbShipper(cfg.Configs))
|
|
|
|
// add a newer PeriodConfig in the future using boltdb-shipper
|
|
cfg.Configs = append(cfg.Configs, chunk.PeriodConfig{
|
|
From: chunk.DayTime{Time: model.Now().Add(time.Hour)},
|
|
IndexType: "boltdb-shipper",
|
|
})
|
|
assert.Equal(t, true, UsingBoltdbShipper(cfg.Configs))
|
|
}
|
|
|
|
func TestSchemaConfig_Validate(t *testing.T) {
|
|
for _, tc := range []struct {
|
|
name string
|
|
configs []chunk.PeriodConfig
|
|
err error
|
|
}{
|
|
{
|
|
name: "empty",
|
|
configs: []chunk.PeriodConfig{},
|
|
err: errZeroLengthConfig,
|
|
},
|
|
{
|
|
name: "NOT using boltdb-shipper",
|
|
configs: []chunk.PeriodConfig{{
|
|
From: chunk.DayTime{Time: model.Now().Add(-24 * time.Hour)},
|
|
IndexType: "boltdb",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Period: 7 * 24 * time.Hour,
|
|
},
|
|
}},
|
|
},
|
|
{
|
|
name: "current config boltdb-shipper with 7 days periodic config, without future index type changes",
|
|
configs: []chunk.PeriodConfig{{
|
|
From: chunk.DayTime{Time: model.Now().Add(-24 * time.Hour)},
|
|
IndexType: "boltdb-shipper",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Period: 7 * 24 * time.Hour,
|
|
},
|
|
}},
|
|
err: errCurrentBoltdbShipperNon24Hours,
|
|
},
|
|
{
|
|
name: "current config boltdb-shipper with 1 day periodic config, without future index type changes",
|
|
configs: []chunk.PeriodConfig{{
|
|
From: chunk.DayTime{Time: model.Now().Add(-24 * time.Hour)},
|
|
IndexType: "boltdb-shipper",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Period: 24 * time.Hour,
|
|
},
|
|
}},
|
|
},
|
|
{
|
|
name: "current config boltdb-shipper with 7 days periodic config, upcoming config NOT boltdb-shipper",
|
|
configs: []chunk.PeriodConfig{{
|
|
From: chunk.DayTime{Time: model.Now().Add(-24 * time.Hour)},
|
|
IndexType: "boltdb-shipper",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Period: 24 * time.Hour,
|
|
},
|
|
}, {
|
|
From: chunk.DayTime{Time: model.Now().Add(time.Hour)},
|
|
IndexType: "boltdb",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Period: 7 * 24 * time.Hour,
|
|
},
|
|
}},
|
|
},
|
|
{
|
|
name: "current and upcoming config boltdb-shipper with 7 days periodic config",
|
|
configs: []chunk.PeriodConfig{{
|
|
From: chunk.DayTime{Time: model.Now().Add(-24 * time.Hour)},
|
|
IndexType: "boltdb-shipper",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Period: 24 * time.Hour,
|
|
},
|
|
}, {
|
|
From: chunk.DayTime{Time: model.Now().Add(time.Hour)},
|
|
IndexType: "boltdb-shipper",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Period: 7 * 24 * time.Hour,
|
|
},
|
|
}},
|
|
err: errUpcomingBoltdbShipperNon24Hours,
|
|
},
|
|
{
|
|
name: "current config NOT boltdb-shipper, upcoming config boltdb-shipper with 7 days periodic config",
|
|
configs: []chunk.PeriodConfig{{
|
|
From: chunk.DayTime{Time: model.Now().Add(-24 * time.Hour)},
|
|
IndexType: "boltdb",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Period: 24 * time.Hour,
|
|
},
|
|
}, {
|
|
From: chunk.DayTime{Time: model.Now().Add(time.Hour)},
|
|
IndexType: "boltdb-shipper",
|
|
Schema: "v9",
|
|
IndexTables: chunk.PeriodicTableConfig{
|
|
Period: 7 * 24 * time.Hour,
|
|
},
|
|
}},
|
|
err: errUpcomingBoltdbShipperNon24Hours,
|
|
},
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
cfg := SchemaConfig{chunk.SchemaConfig{Configs: tc.configs}}
|
|
err := cfg.Validate()
|
|
if tc.err == nil {
|
|
require.NoError(t, err)
|
|
} else {
|
|
require.EqualError(t, err, tc.err.Error())
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_OverlappingChunks(t *testing.T) {
|
|
chunks := []chunk.Chunk{
|
|
newChunk(logproto.Stream{
|
|
Labels: `{foo="bar"}`,
|
|
Entries: []logproto.Entry{
|
|
{Timestamp: time.Unix(0, 1), Line: "1"},
|
|
{Timestamp: time.Unix(0, 4), Line: "4"},
|
|
},
|
|
}),
|
|
newChunk(logproto.Stream{
|
|
Labels: `{foo="bar"}`,
|
|
Entries: []logproto.Entry{
|
|
{Timestamp: time.Unix(0, 2), Line: "2"},
|
|
{Timestamp: time.Unix(0, 3), Line: "3"},
|
|
},
|
|
}),
|
|
}
|
|
s := &store{
|
|
Store: &mockChunkStore{chunks: chunks, client: &mockChunkStoreClient{chunks: chunks}},
|
|
cfg: Config{
|
|
MaxChunkBatchSize: 10,
|
|
},
|
|
chunkMetrics: NilMetrics,
|
|
}
|
|
|
|
ctx = user.InjectOrgID(context.Background(), "test-user")
|
|
it, err := s.SelectLogs(ctx, logql.SelectLogParams{QueryRequest: &logproto.QueryRequest{
|
|
Selector: `{foo="bar"}`,
|
|
Limit: 1000,
|
|
Direction: logproto.BACKWARD,
|
|
Start: time.Unix(0, 0),
|
|
End: time.Unix(0, 10),
|
|
}})
|
|
if err != nil {
|
|
t.Errorf("store.SelectLogs() error = %v", err)
|
|
return
|
|
}
|
|
defer it.Close()
|
|
require.True(t, it.Next())
|
|
require.Equal(t, "4", it.Entry().Line)
|
|
require.True(t, it.Next())
|
|
require.Equal(t, "3", it.Entry().Line)
|
|
require.True(t, it.Next())
|
|
require.Equal(t, "2", it.Entry().Line)
|
|
require.True(t, it.Next())
|
|
require.Equal(t, "1", it.Entry().Line)
|
|
require.False(t, it.Next())
|
|
}
|
|
|
|
func Test_GetSeries(t *testing.T) {
|
|
var (
|
|
store = &store{
|
|
Store: newMockChunkStore([]*logproto.Stream{
|
|
{
|
|
Labels: `{foo="bar",buzz="boo"}`,
|
|
Entries: []logproto.Entry{
|
|
{Timestamp: time.Unix(0, 1), Line: "1"},
|
|
},
|
|
},
|
|
{
|
|
Labels: `{foo="buzz"}`,
|
|
Entries: []logproto.Entry{
|
|
{Timestamp: time.Unix(0, 1), Line: "1"},
|
|
},
|
|
},
|
|
{
|
|
Labels: `{bar="foo"}`,
|
|
Entries: []logproto.Entry{
|
|
{Timestamp: time.Unix(0, 1), Line: "1"},
|
|
},
|
|
},
|
|
}),
|
|
cfg: Config{
|
|
MaxChunkBatchSize: 10,
|
|
},
|
|
chunkMetrics: NilMetrics,
|
|
}
|
|
ctx = user.InjectOrgID(context.Background(), "test-user")
|
|
expectedSeries = []logproto.SeriesIdentifier{
|
|
{
|
|
Labels: map[string]string{"bar": "foo"},
|
|
},
|
|
{
|
|
Labels: map[string]string{"foo": "bar", "buzz": "boo"},
|
|
},
|
|
{
|
|
Labels: map[string]string{"foo": "buzz"},
|
|
},
|
|
}
|
|
)
|
|
|
|
for _, tt := range []struct {
|
|
name string
|
|
req logql.SelectLogParams
|
|
expectedSeries []logproto.SeriesIdentifier
|
|
}{
|
|
{
|
|
"all series",
|
|
logql.SelectLogParams{
|
|
QueryRequest: &logproto.QueryRequest{
|
|
Selector: ``,
|
|
Start: time.Unix(0, 0),
|
|
End: time.Unix(0, 10),
|
|
},
|
|
},
|
|
expectedSeries,
|
|
},
|
|
{
|
|
"all series with sharding",
|
|
logql.SelectLogParams{
|
|
QueryRequest: &logproto.QueryRequest{
|
|
Selector: ``,
|
|
Start: time.Unix(0, 0),
|
|
End: time.Unix(0, 10),
|
|
Shards: []string{astmapper.ShardAnnotation{Shard: 1, Of: 16}.String()},
|
|
},
|
|
},
|
|
expectedSeries,
|
|
},
|
|
{
|
|
"selected series",
|
|
logql.SelectLogParams{
|
|
QueryRequest: &logproto.QueryRequest{
|
|
Selector: `{buzz=~".oo"}`,
|
|
Start: time.Unix(0, 0),
|
|
End: time.Unix(0, 10),
|
|
},
|
|
},
|
|
[]logproto.SeriesIdentifier{
|
|
{
|
|
Labels: map[string]string{"foo": "bar", "buzz": "boo"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"selected series with sharding",
|
|
logql.SelectLogParams{
|
|
QueryRequest: &logproto.QueryRequest{
|
|
Selector: `{buzz=~".oo"}`,
|
|
Start: time.Unix(0, 0),
|
|
End: time.Unix(0, 10),
|
|
Shards: []string{astmapper.ShardAnnotation{Shard: 1, Of: 16}.String()},
|
|
},
|
|
},
|
|
[]logproto.SeriesIdentifier{
|
|
{
|
|
Labels: map[string]string{"foo": "bar", "buzz": "boo"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"no match",
|
|
logql.SelectLogParams{
|
|
QueryRequest: &logproto.QueryRequest{
|
|
Selector: `{buzz=~"foo"}`,
|
|
Start: time.Unix(0, 0),
|
|
End: time.Unix(0, 10),
|
|
},
|
|
},
|
|
[]logproto.SeriesIdentifier{},
|
|
},
|
|
} {
|
|
tt := tt
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
series, err := store.GetSeries(ctx, tt.req)
|
|
require.NoError(t, err)
|
|
require.Equal(t, tt.expectedSeries, series)
|
|
})
|
|
}
|
|
}
|
|
|