diff --git a/tool/helper.go b/tool/helper.go index 2b5935d..2015e6a 100644 --- a/tool/helper.go +++ b/tool/helper.go @@ -5,6 +5,7 @@ import ( "bytes" "errors" "fmt" + "github.com/spf13/cast" "io" "math" "math/rand" @@ -43,7 +44,7 @@ func ByteFmt(size int64) string { if frac > 0 { return fmt.Sprintf("%.1f%s", math.Floor(val*10)/10, unitArr[p]) } else { - return fmt.Sprintf("%d%s", int(val), unitArr[p]) + return fmt.Sprintf("%d%s", cast.ToInt(val), unitArr[p]) } }