@ -15,6 +15,7 @@ import (
"github.com/weaveworks/common/user"
"github.com/grafana/loki/pkg/chunkenc"
"github.com/grafana/loki/pkg/distributor/writefailures"
"github.com/grafana/loki/pkg/ingester/client"
"github.com/grafana/loki/pkg/logproto"
"github.com/grafana/loki/pkg/logql/log"
@ -67,7 +68,7 @@ func TestIngesterWAL(t *testing.T) {
}
}
i , err := New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err := New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
@ -110,7 +111,7 @@ func TestIngesterWAL(t *testing.T) {
expectCheckpoint ( t , walDir , false , time . Second )
// restart the ingester
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )
@ -124,7 +125,7 @@ func TestIngesterWAL(t *testing.T) {
require . Nil ( t , services . StopAndAwaitTerminated ( context . Background ( ) , i ) )
// restart the ingester
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )
@ -147,7 +148,7 @@ func TestIngesterWALIgnoresStreamLimits(t *testing.T) {
}
}
i , err := New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err := New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
@ -193,7 +194,7 @@ func TestIngesterWALIgnoresStreamLimits(t *testing.T) {
require . NoError ( t , err )
// restart the ingester
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )
@ -250,7 +251,7 @@ func TestIngesterWALBackpressureSegments(t *testing.T) {
}
}
i , err := New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err := New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
@ -271,7 +272,7 @@ func TestIngesterWALBackpressureSegments(t *testing.T) {
expectCheckpoint ( t , walDir , false , time . Second )
// restart the ingester, ensuring we replayed from WAL.
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )
@ -292,7 +293,7 @@ func TestIngesterWALBackpressureCheckpoint(t *testing.T) {
}
}
i , err := New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err := New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
@ -313,7 +314,7 @@ func TestIngesterWALBackpressureCheckpoint(t *testing.T) {
require . Nil ( t , services . StopAndAwaitTerminated ( context . Background ( ) , i ) )
// restart the ingester, ensuring we can replay from the checkpoint as well.
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )
@ -449,7 +450,7 @@ func Test_SeriesIterator(t *testing.T) {
limiter := NewLimiter ( limits , NilMetrics , & ringCountMock { count : 1 } , 1 )
for i := 0 ; i < 3 ; i ++ {
inst , err := newInstance ( defaultConfig ( ) , defaultPeriodConfigs , fmt . Sprintf ( "%d" , i ) , limiter , runtime . DefaultTenantConfigs ( ) , noopWAL { } , NilMetrics , nil , nil , NewStreamRateCalculator ( ) )
inst , err := newInstance ( defaultConfig ( ) , defaultPeriodConfigs , fmt . Sprintf ( "%d" , i ) , limiter , runtime . DefaultTenantConfigs ( ) , noopWAL { } , NilMetrics , nil , nil , NewStreamRateCalculator ( ) , nil )
require . Nil ( t , err )
require . NoError ( t , inst . Push ( context . Background ( ) , & logproto . PushRequest { Streams : [ ] logproto . Stream { stream1 } } ) )
require . NoError ( t , inst . Push ( context . Background ( ) , & logproto . PushRequest { Streams : [ ] logproto . Stream { stream2 } } ) )
@ -496,7 +497,7 @@ func Benchmark_SeriesIterator(b *testing.B) {
limiter := NewLimiter ( limits , NilMetrics , & ringCountMock { count : 1 } , 1 )
for i := range instances {
inst , _ := newInstance ( defaultConfig ( ) , defaultPeriodConfigs , fmt . Sprintf ( "instance %d" , i ) , limiter , runtime . DefaultTenantConfigs ( ) , noopWAL { } , NilMetrics , nil , nil , NewStreamRateCalculator ( ) )
inst , _ := newInstance ( defaultConfig ( ) , defaultPeriodConfigs , fmt . Sprintf ( "instance %d" , i ) , limiter , runtime . DefaultTenantConfigs ( ) , noopWAL { } , NilMetrics , nil , nil , NewStreamRateCalculator ( ) , nil )
require . NoError ( b ,
inst . Push ( context . Background ( ) , & logproto . PushRequest {
@ -588,7 +589,7 @@ func TestIngesterWALReplaysUnorderedToOrdered(t *testing.T) {
}
}
i , err := New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err := New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
@ -660,7 +661,7 @@ func TestIngesterWALReplaysUnorderedToOrdered(t *testing.T) {
require . NoError ( t , err )
// restart the ingester
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil )
i , err = New ( ingesterConfig , client . Config { } , newStore ( ) , limits , runtime . DefaultTenantConfigs ( ) , nil , writefailures . Cfg { } )
require . NoError ( t , err )
defer services . StopAndAwaitTerminated ( context . Background ( ) , i ) //nolint:errcheck
require . Nil ( t , services . StartAndAwaitRunning ( context . Background ( ) , i ) )