init: 1.0.0

pkg/traffic/convert.go

@@ -0,0 +1,66 @@
+package traffic
+
+import "fmt"
+
+const (
+	bitsInTiB = 1024.0 * 1024.0 * 1024.0 * 1024.0
+	bitsInTb  = 1000.0 * 1000.0 * 1000.0 * 1000.0
+	bitsInGiB = 1024.0 * 1024.0 * 1024.0
+	bitsInGb  = 1000.0 * 1000.0 * 1000.0
+	bitsInMiB = 1024.0 * 1024.0
+	bitsInMb  = 1000.0 * 1000.0
+
+	Mb  = "Mb"
+	MiB = "MiB"
+	Gb  = "Gb"
+	GiB = "GiB"
+	Tb  = "Tb"
+	TiB = "TiB"
+)
+
+// Convert converts the given traffic in bits to the specified unit ("MiB" or "GiB")
+func Convert(bits int64, unit string) float64 {
+	switch unit {
+	case Mb:
+		return float64(bits) / bitsInMb
+	case MiB:
+		return float64(bits) / bitsInMiB
+	case GiB:
+		return float64(bits) / bitsInGiB
+	case Gb:
+		return float64(bits) / bitsInGb
+	case TiB:
+		return float64(bits) / bitsInTiB
+	case Tb:
+		return float64(bits) / bitsInTb
+	default:
+		return 0 // Return 0 for unsupported units
+	}
+}
+
+// AutoConvert converts the given traffic in bits to the largest possible unit and returns a formatted string
+func AutoConvert(bits int64, useBinary bool) string {
+	if useBinary {
+		switch {
+		case float64(bits) >= bitsInTiB:
+			return fmt.Sprintf("%.2f TiB", float64(bits)/bitsInTiB)
+		case float64(bits) >= bitsInGiB:
+			return fmt.Sprintf("%.2f GiB", float64(bits)/bitsInGiB)
+		case float64(bits) >= bitsInMiB:
+			return fmt.Sprintf("%.2f MiB", float64(bits)/bitsInMiB)
+		default:
+			return fmt.Sprintf("%d bits", bits)
+		}
+	} else {
+		switch {
+		case float64(bits) >= bitsInTb:
+			return fmt.Sprintf("%.2f Tb", float64(bits)/bitsInTb)
+		case float64(bits) >= bitsInGb:
+			return fmt.Sprintf("%.2f Gb", float64(bits)/bitsInGb)
+		case float64(bits) >= bitsInMb:
+			return fmt.Sprintf("%.2f Mb", float64(bits)/bitsInMb)
+		default:
+			return fmt.Sprintf("%d bits", bits)
+		}
+	}
+}