Add ParseBytesInt function (#15)

unit/unit.go

@@ -16,6 +16,15 @@ func ParseBytes(s string) (bytes uint64, iec bool, err error) {
 	return bytes, iec, nil
 }
 
+// ParseBytesInt parse the string to bytes
+func ParseBytesInt(s string) (int, error) {
+	bytes, err := humanize.ParseBytes(s)
+	if err != nil {
+		return 0, err
+	}
+	return int(bytes), nil
+}
+
 // Bytes produces a human readable representation of an SI size
 func Bytes(b uint64) string {
 	return humanize.Bytes(b)

unit/unit_test.go

@@ -57,7 +57,7 @@ func TestParseBytes(t *testing.T) {
 		t.Run(tc.s, func(t *testing.T) {
 			bytes, iec, err := ParseBytes(tc.s)
 			if err != nil {
-				t.Errorf("parse bytes error => %v", err)
+				t.Errorf("ParseBytes error => %v", err)
 				return
 			}
 			if bytes != tc.v {
@@ -68,6 +68,16 @@ func TestParseBytes(t *testing.T) {
 				t.Errorf("IEC expect %v, but get %v", tc.iec, iec)
 				return
 			}
+
+			bytesInt, err := ParseBytesInt(tc.s)
+			if err != nil {
+				t.Errorf("ParseBytesInt error => %v", err)
+				return
+			}
+			if bytesInt != int(tc.v) {
+				t.Errorf("expect %d, but get %d", tc.v, bytesInt)
+				return
+			}
 		})
 	}
 }
@@ -84,7 +94,12 @@ func TestParseBytesReturnError(t *testing.T) {
 		t.Run(tc.s, func(t *testing.T) {
 			_, _, err := ParseBytes(tc.s)
 			if err == nil {
-				t.Errorf("expect get an error, but get nil")
+				t.Errorf("ParseBytes expect get an error, but get nil")
+			}
+
+			_, err = ParseBytesInt(tc.s)
+			if err == nil {
+				t.Errorf("ParseBytesInt expect get an error, but get nil")
 			}
 		})
 	}