add: util for formatting bytes format

internal/util.go

@@ -0,0 +1,19 @@
+package internal
+
+import (
+	"fmt"
+)
+
+// ref: https://yourbasic.org/golang/formatting-byte-size-to-human-readable-format/
+func formatBytes(b int64) string {
+	const unit = 1000
+	if b < unit {
+		return fmt.Sprintf("%d", b)
+	}
+	div, exp := int64(unit), 0
+	for n := b / unit; n >= unit; n /= unit {
+		div *= unit
+		exp++
+	}
+	return fmt.Sprintf("%.1f%c", float64(b)/float64(div), "KMGTPE"[exp])
+}

internal/util_test.go

@@ -0,0 +1,19 @@
+package internal
+
+import "testing"
+
+func TestFormatBytes(t *testing.T) {
+	tests := map[int64]string{
+		999:       "999",
+		1024:      "1.0K",
+		1536:      "1.5K",
+		987654321: "987.7M",
+	}
+	for key, value := range tests {
+		got := formatBytes(key)
+		want := value
+		if got != want {
+			t.Fatalf("got: %v, want: %v\n", got, want)
+		}
+	}
+}