diff --git a/helpers.go b/helpers.go index faef03e..8af3f60 100644 --- a/helpers.go +++ b/helpers.go @@ -115,3 +115,27 @@ func (c *Cluster) AddDroplet(uuid string, hostname string, cpus int, mem int) *D c.Droplets = append(c.Droplets, d) return d } + +func FormatBytes(b int64) string { + if b < 2000 { + return fmt.Sprintf("%d B", b) + } + + kb := int(b / 1024) + if kb < 2000 { + return fmt.Sprintf("%d KB", kb) + } + + mb := int(b / (1024 * 1024)) + if mb < 2000 { + return fmt.Sprintf("%d MB", mb) + } + + gb := int(b / (1024 * 1024 * 1024)) + if gb < 2000 { + return fmt.Sprintf("%d GB", gb) + } + + tb := int(b / (1024 * 1024 * 1024 * 1024)) + return fmt.Sprintf("%d TB", tb) +}