Golang工具方法 — Byte与KB、MB、GB、TB之间的转换

1 Byte转成KB、MB、GB、TB

// 以1000作为基数
func ByteCountSI(b int64) string {
	const unit = 1000
	if b < unit {
		return fmt.Sprintf("%d B", b)
	}
	div, exp := int64(unit), 0
	for n := b / unit; n >= unit; n /= unit {
		div *= unit
		exp++
	}
	return fmt.Sprintf("%.1f %cB",
		float64(b)/float64(div), "kMGTPE"[exp])
}

// 以1024作为基数
func ByteCountIEC(b int64) string {
	const unit = 1024
	if b < unit {
		return fmt.Sprintf("%d B", b)
	}
	div, exp := int64(unit), 0
	for n := b / unit; n >= unit; n /= unit {
		div *= unit
		exp++
	}
	return fmt.Sprintf("%.1f %cB",
		float64(b)/float64(div), "KMGTPE"[exp])
}

2 KB、MB、GB、TB转换为Byte

// 将KB、MB、GB和TB转换为字节的
const (
	_        = iota
	KB int64 = 1 << (10 * iota)
	MB
	GB
	TB
)

// size是数值，uint是单位
func ToBytes(size int64, unit string) int64 {
	switch strings.ToLower(unit) {
	case "kb":
		return size * KB
	case "mb":
		return size * MB
	case "gb":
		return size * GB
	case "tb":
		return size * TB
	default:
		return size
	}
}