mirror of
https://github.com/gofiber/fiber.git
synced 2025-02-23 20:03:43 +00:00
86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
// ⚡️ Fiber is an Express inspired web framework written in Go with ☕️
|
|
// 🤖 Github Repository: https://github.com/gofiber/fiber
|
|
// 📌 API Documentation: https://docs.gofiber.io
|
|
|
|
package utils
|
|
|
|
import (
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
"unsafe"
|
|
)
|
|
|
|
// #nosec G103
|
|
// GetString returns a string pointer without allocation
|
|
func UnsafeString(b []byte) string {
|
|
return *(*string)(unsafe.Pointer(&b))
|
|
}
|
|
|
|
// #nosec G103
|
|
// GetBytes returns a byte pointer without allocation
|
|
func UnsafeBytes(s string) (bs []byte) {
|
|
sh := (*reflect.StringHeader)(unsafe.Pointer(&s))
|
|
bh := (*reflect.SliceHeader)(unsafe.Pointer(&bs))
|
|
bh.Data = sh.Data
|
|
bh.Len = sh.Len
|
|
bh.Cap = sh.Len
|
|
return
|
|
}
|
|
|
|
// CopyString copies a string to make it immutable
|
|
func CopyString(s string) string {
|
|
return string(UnsafeBytes(s))
|
|
}
|
|
|
|
// CopyBytes copies a slice to make it immutable
|
|
func CopyBytes(b []byte) []byte {
|
|
tmp := make([]byte, len(b))
|
|
copy(tmp, b)
|
|
return tmp
|
|
}
|
|
|
|
const (
|
|
uByte = 1 << (10 * iota)
|
|
uKilobyte
|
|
uMegabyte
|
|
uGigabyte
|
|
uTerabyte
|
|
uPetabyte
|
|
uExabyte
|
|
)
|
|
|
|
// ByteSize returns a human-readable byte string of the form 10M, 12.5K, and so forth.
|
|
// The unit that results in the smallest number greater than or equal to 1 is always chosen.
|
|
func ByteSize(bytes uint64) string {
|
|
unit := ""
|
|
value := float64(bytes)
|
|
switch {
|
|
case bytes >= uExabyte:
|
|
unit = "EB"
|
|
value = value / uExabyte
|
|
case bytes >= uPetabyte:
|
|
unit = "PB"
|
|
value = value / uPetabyte
|
|
case bytes >= uTerabyte:
|
|
unit = "TB"
|
|
value = value / uTerabyte
|
|
case bytes >= uGigabyte:
|
|
unit = "GB"
|
|
value = value / uGigabyte
|
|
case bytes >= uMegabyte:
|
|
unit = "MB"
|
|
value = value / uMegabyte
|
|
case bytes >= uKilobyte:
|
|
unit = "KB"
|
|
value = value / uKilobyte
|
|
case bytes >= uByte:
|
|
unit = "B"
|
|
default:
|
|
return "0B"
|
|
}
|
|
result := strconv.FormatFloat(value, 'f', 1, 64)
|
|
result = strings.TrimSuffix(result, ".0")
|
|
return result + unit
|
|
}
|