✨ feat: mathutil - add new func: FormatBytes Format the byte size to be a readable string.

Author: inhere

mathutil/format.go

@@ -22,6 +22,21 @@ func DataSize(size uint64) string {
 	}
 }
 
+// FormatBytes Format the byte size to be a readable string. eg: 1024 => 1 KB
+func FormatBytes(bytes int) string {
+	const unit = 1024
+	if bytes < unit {
+		return fmt.Sprintf("%d B", bytes)
+	}
+
+	div, exp := int64(unit), 0
+	for n := bytes / unit; n >= unit; n /= unit {
+		div *= unit
+		exp++
+	}
+	return fmt.Sprintf("%.2f %cB", float64(bytes)/float64(div), "KMGTPE"[exp])
+}
+
 var timeFormats = [][]int{
 	{0},
 	{1},

mathutil/format_test.go

@@ -24,6 +24,25 @@ func TestDataSize(t *testing.T) {
 	}
 }
 
+func TestFormatBytes(t *testing.T) {
+	tests := []struct {
+		input    int
+		expected string
+	}{
+		{0, "0 B"},
+		{1024, "1.00 KB"},
+		{1536, "1.50 KB"},
+		{1048576, "1.00 MB"},
+		{1073741824, "1.00 GB"},
+		{1200346778, "1.12 GB"},
+	}
+
+	for _, tt := range tests {
+		result := mathutil.FormatBytes(tt.input)
+		assert.Eq(t, tt.expected, result)
+	}
+}
+
 func TestHowLongAgo(t *testing.T) {
 	tests := []struct {
 		args int64