fix some linting / test

Author: alanprot

pkg/cortexpb/slicesPool.go

@@ -9,7 +9,7 @@ type byteSlicePools struct {
 	pools []sync.Pool
 }
 
-func NewSlicePool(pools int) *byteSlicePools {
+func newSlicePool(pools int) *byteSlicePools {
 	sp := byteSlicePools{}
 	sp.init(pools)
 	return &sp
@@ -39,7 +39,7 @@ func (sp *byteSlicePools) getSlice(size int) []byte {
 }
 
 func (sp *byteSlicePools) reuseSlice(s []byte) {
-	index := int(math.Ceil(math.Log2(float64(cap(s)))))
+	index := int(math.Floor(math.Log2(float64(cap(s)))))
 
 	if index < 0 || index >= len(sp.pools) {
 		return

pkg/cortexpb/slicesPool_test.go

@@ -7,12 +7,10 @@ import (
 )
 
 func TestByteSlicePools(t *testing.T) {
-	sut := NewSlicePool(20)
-
+	sut := newSlicePool(20)
 	for i := 0; i < 1024*1024; i = i + 128 {
 		s := sut.getSlice(i)
 		assert.Equal(t, len(s), i)
 		sut.reuseSlice(s)
 	}
-
 }

pkg/cortexpb/timeseries.go

@@ -45,7 +45,7 @@ var (
 			}
 		},
 	}
-	bytePool = NewSlicePool(20)
+	bytePool = newSlicePool(20)
 )
 
 // PreallocConfig configures how structures will be preallocated to optimise
@@ -82,11 +82,11 @@ func (p *PreallocTimeseries) Unmarshal(dAtA []byte) error {
 	return p.TimeSeries.Unmarshal(dAtA)
 }
 
-func (m *PreallocWriteRequest) Marshal() (dAtA []byte, err error) {
-	size := m.Size()
+func (p *PreallocWriteRequest) Marshal() (dAtA []byte, err error) {
+	size := p.Size()
 	dAtA = bytePool.getSlice(size)
-	m.data = dAtA
-	n, err := m.MarshalToSizedBuffer(dAtA[:size])
+	p.data = dAtA
+	n, err := p.MarshalToSizedBuffer(dAtA[:size])
 	if err != nil {
 		return nil, err
 	}

pkg/distributor/distributor_test.go

@@ -2688,6 +2688,10 @@ func (i *mockIngester) Close() error {
 	return nil
 }
 
+func (i *mockIngester) PushPreAlloc(ctx context.Context, in *cortexpb.PreallocWriteRequest, opts ...grpc.CallOption) (*cortexpb.WriteResponse, error) {
+	return i.Push(ctx, &in.WriteRequest, opts...)
+}
+
 func (i *mockIngester) Push(ctx context.Context, req *cortexpb.WriteRequest, opts ...grpc.CallOption) (*cortexpb.WriteResponse, error) {
 	i.Lock()
 	defer i.Unlock()