only show decimal points for smaller file size numbers

This commit is contained in:
Jeffrey Morgan 2023-11-20 09:13:53 -05:00
parent be61a81758
commit 93a108214c

View file

@ -1,6 +1,9 @@
package format package format
import "fmt" import (
"fmt"
"math"
)
const ( const (
Byte = 1 Byte = 1
@ -11,16 +14,32 @@ const (
) )
func HumanBytes(b int64) string { func HumanBytes(b int64) string {
var value float64
var unit string
switch { switch {
case b > TeraByte: case b >= TeraByte:
return fmt.Sprintf("%.1f TB", float64(b)/TeraByte) value = float64(b) / TeraByte
case b > GigaByte: unit = "TB"
return fmt.Sprintf("%.1f GB", float64(b)/GigaByte) case b >= GigaByte:
case b > MegaByte: value = float64(b) / GigaByte
return fmt.Sprintf("%.1f MB", float64(b)/MegaByte) unit = "GB"
case b > KiloByte: case b >= MegaByte:
return fmt.Sprintf("%.1f KB", float64(b)/KiloByte) value = float64(b) / MegaByte
unit = "MB"
case b >= KiloByte:
value = float64(b) / KiloByte
unit = "KB"
default: default:
return fmt.Sprintf("%d B", b) return fmt.Sprintf("%d B", b)
} }
switch {
case value >= 100:
return fmt.Sprintf("%d %s", int(value), unit)
case value != math.Trunc(value):
return fmt.Sprintf("%.1f %s", value, unit)
default:
return fmt.Sprintf("%d %s", int(value), unit)
}
} }