Merge pull request #33 from calyptia/fix-valid-size

fix valid size
chronosphereio · Sep 21, 2022 · d8960a4 · d8960a4
2 parents 6f381e8 + 0239158
commit d8960a4
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/validator.go b/validator.go
@@ -409,6 +409,12 @@ func validByteSize(prop Field) bool {
 		return true
 	}
 
+	// try with number as string
+	if len(prop.Values) == 1 && prop.Values[0].String != nil {
+		_, err := strconv.ParseInt(*prop.Values[0].String, 10, 64)
+		return err == nil
+	}
+
 	// try with bytes parser.
 	_, err := bytefmt.ToBytes(prop.Values.ToString())
 	if err == nil {

diff --git a/validator_test.go b/validator_test.go
@@ -359,6 +359,14 @@ func TestConfigSection_Validate(t *testing.T) {
 			kind: NoneSection,
 			want: `unsupported plugin kind "NONE"`,
 		},
+		{
+			name: "input_forward_buffer_max_size_size",
+			kind: InputSection,
+			props: map[string]string{
+				"Name":            "forward",
+				"buffer_max_size": "61440000",
+			},
+		},
 	}
 	for _, tc := range tt {
 		t.Run(tc.name, func(t *testing.T) {