max line limits

pkg/distributor/limits.go

@@ -0,0 +1,24 @@
+package distributor
+
+// Limits controls the max line size
+type Limits interface {
+	MaxLineSize(userID string) int
+}
+
+type constLimits int
+
+func (c constLimits) MaxLineSize(userID string) int {
+	return int(c)
+}
+
+// PriorityLimits returns the first non-zero result from a set of []Limits
+type PriorityLimits []Limits
+
+func (ls PriorityLimits) MaxLineSize(userID string) (res int) {
+	for _, l := range ls {
+		if res = l.MaxLineSize(userID); res != 0 {
+			return res
+		}
+	}
+	return res
+}

pkg/distributor/limits_test.go

@@ -0,0 +1,18 @@
+package distributor
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestLimits(t *testing.T) {
+	require.Equal(t, 0, constLimits(0).MaxLineSize("a"))
+	require.Equal(t, 2, constLimits(2).MaxLineSize("a"))
+	require.Equal(t, 1,
+		PriorityLimits([]Limits{
+			constLimits(0),
+			constLimits(1),
+		}).MaxLineSize("a"),
+	)
+}

pkg/util/flagext/bytesize.go

@@ -27,5 +27,5 @@ func (bs *ByteSize) Set(s string) error {
 }
 
 func (bs ByteSize) Get() interface{} {
-	return uint64(bs)
+	return int(bs)
 }

pkg/util/flagext/bytesize_test.go

@@ -10,7 +10,7 @@ func Test_ByteSize(t *testing.T) {
 	for _, tc := range []struct {
 		in  string
 		err bool
-		out uint64
+		out int
 	}{
 		{
 			in:  "abc",
@@ -62,7 +62,7 @@ func Test_ByteSize(t *testing.T) {
 				require.NotNil(t, err)
 			} else {
 				require.Nil(t, err)
-				require.Equal(t, tc.out, bs.Get().(uint64))
+				require.Equal(t, tc.out, bs.Get().(int))
 			}
 
 		})