1
0
mirror of https://github.com/duke-git/lancet.git synced 2026-03-01 00:35:28 +08:00

refactoring: change unit variable to package internal

This commit is contained in:
dudaodong
2024-08-26 16:25:57 +08:00
parent 3e8c3bd396
commit 38148978cf
2 changed files with 47 additions and 47 deletions

View File

@@ -18,63 +18,63 @@ import (
// http://en.wikipedia.org/wiki/Binary_prefix // http://en.wikipedia.org/wiki/Binary_prefix
const ( const (
// Decimal // Decimal
UnitB = 1 unitB = 1
UnitKB = 1000 unitKB = 1000
UnitMB = 1000 * UnitKB unitMB = 1000 * unitKB
UnitGB = 1000 * UnitMB unitGB = 1000 * unitMB
UnitTB = 1000 * UnitGB unitTB = 1000 * unitGB
UnitPB = 1000 * UnitTB unitPB = 1000 * unitTB
UnitEB = 1000 * UnitPB unitEB = 1000 * unitPB
// Binary // Binary
UnitBiB = 1 unitBiB = 1
UnitKiB = 1024 unitKiB = 1024
UnitMiB = 1024 * UnitKiB unitMiB = 1024 * unitKiB
UnitGiB = 1024 * UnitMiB unitGiB = 1024 * unitMiB
UnitTiB = 1024 * UnitGiB unitTiB = 1024 * unitGiB
UnitPiB = 1024 * UnitTiB unitPiB = 1024 * unitTiB
UnitEiB = 1024 * UnitPiB unitEiB = 1024 * unitPiB
) )
// type byteUnitMap map[byte]int64 // type byteUnitMap map[byte]int64
var ( var (
decimalByteMap = map[string]uint64{ decimalByteMap = map[string]uint64{
"b": UnitB, "b": unitB,
"kb": UnitKB, "kb": unitKB,
"mb": UnitMB, "mb": unitMB,
"gb": UnitGB, "gb": unitGB,
"tb": UnitTB, "tb": unitTB,
"pb": UnitPB, "pb": unitPB,
"eb": UnitEB, "eb": unitEB,
// Without suffix // Without suffix
"": UnitB, "": unitB,
"k": UnitKB, "k": unitKB,
"m": UnitMB, "m": unitMB,
"g": UnitGB, "g": unitGB,
"t": UnitTB, "t": unitTB,
"p": UnitPB, "p": unitPB,
"e": UnitEB, "e": unitEB,
} }
binaryByteMap = map[string]uint64{ binaryByteMap = map[string]uint64{
"bi": UnitBiB, "bi": unitBiB,
"kib": UnitKiB, "kib": unitKiB,
"mib": UnitMiB, "mib": unitMiB,
"gib": UnitGiB, "gib": unitGiB,
"tib": UnitTiB, "tib": unitTiB,
"pib": UnitPiB, "pib": unitPiB,
"eib": UnitEiB, "eib": unitEiB,
// Without suffix // Without suffix
"": UnitBiB, "": unitBiB,
"ki": UnitKiB, "ki": unitKiB,
"mi": UnitMiB, "mi": unitMiB,
"gi": UnitGiB, "gi": unitGiB,
"ti": UnitTiB, "ti": unitTiB,
"pi": UnitPiB, "pi": unitPiB,
"ei": UnitEiB, "ei": unitEiB,
} }
) )

View File

@@ -15,11 +15,11 @@ func TestDecimalBytes(t *testing.T) {
assert.Equal("1.024KB", DecimalBytes(1024)) assert.Equal("1.024KB", DecimalBytes(1024))
assert.Equal("1.2346MB", DecimalBytes(1234567)) assert.Equal("1.2346MB", DecimalBytes(1234567))
assert.Equal("1.235MB", DecimalBytes(1234567, 3)) assert.Equal("1.235MB", DecimalBytes(1234567, 3))
assert.Equal("1.123GB", DecimalBytes(float64(1.123*UnitGB))) assert.Equal("1.123GB", DecimalBytes(float64(1.123*unitGB)))
assert.Equal("2.123TB", DecimalBytes(float64(2.123*UnitTB))) assert.Equal("2.123TB", DecimalBytes(float64(2.123*unitTB)))
assert.Equal("3.123PB", DecimalBytes(float64(3.123*UnitPB))) assert.Equal("3.123PB", DecimalBytes(float64(3.123*unitPB)))
assert.Equal("4.123EB", DecimalBytes(float64(4.123*UnitEB))) assert.Equal("4.123EB", DecimalBytes(float64(4.123*unitEB)))
assert.Equal("1EB", DecimalBytes(float64(1000*UnitPB))) assert.Equal("1EB", DecimalBytes(float64(1000*unitPB)))
assert.Equal("62MB", DecimalBytes(61812496, 0)) assert.Equal("62MB", DecimalBytes(61812496, 0))
assert.Equal("61.8MB", DecimalBytes(61812496, 1)) assert.Equal("61.8MB", DecimalBytes(61812496, 1))
assert.Equal("401MB", DecimalBytes(401000000)) assert.Equal("401MB", DecimalBytes(401000000))