From 2ce99643b5d7322546a6293a11ede4a06656570c Mon Sep 17 00:00:00 2001
From: dingguodong <dingguodong@easycorp.ltd>
Date: Mon, 12 Jun 2023 19:06:16 +0800
Subject: [PATCH] * Refactor the helper::formatKB to resolve the noncompliant
 code.

---
 framework/helper.class.php | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/framework/helper.class.php b/framework/helper.class.php
index 360f1c0b19..55fbe178ef 100644
--- a/framework/helper.class.php
+++ b/framework/helper.class.php
@@ -243,13 +243,10 @@ class helper extends baseHelper
      */
     public static function formatKB($traffic, $precision = 2)
     {
-        $gigaByte = round($traffic / (1024 * 1024), $precision);
-        if($gigaByte >= 1) return $gigaByte . 'GB';
+        $base     = log($traffic, 1024);
+        $suffixes = array('', 'KB', 'MB', 'GB', 'TB');
 
-        $megaByte = round($traffic / 1024, $precision);
-        if($megaByte >= 1) return $megaByte . 'MB';
-
-        return round($traffic, $precision) . 'KB';
+        return round(pow(1024, $base - floor($base)), $precision) . $suffixes[floor($base)];
     }
 
 	/**