From afe548b61a9d6c6e211948edb0fbb4b7d4ad2eec Mon Sep 17 00:00:00 2001
From: pra-moh <49077256+pra-moh@users.noreply.github.com>
Date: Fri, 26 Mar 2021 10:24:02 -0700
Subject: [PATCH] [201911][procdockerstatsd] Add missing unit conversion
 (#7157)

Fixing same issue in 201911 as mention here #7151
---
 .../procdockerstatsd/procdockerstatsd            | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/files/image_config/procdockerstatsd/procdockerstatsd b/files/image_config/procdockerstatsd/procdockerstatsd
index abb909cb4a..8901cf3e30 100755
--- a/files/image_config/procdockerstatsd/procdockerstatsd
+++ b/files/image_config/procdockerstatsd/procdockerstatsd
@@ -69,8 +69,11 @@ class ProcDockerStats(daemon_base.DaemonBase):
         UNITS_B = 'B'
         UNITS_KB = 'KB'
         UNITS_MB = 'MB'
+        UNITS_GB = 'GB'
+        UNITS_GB = 'TB'
         UNITS_MiB = 'MiB'
         UNITS_GiB = 'GiB'
+        UNITS_TiB = 'TiB'
 
         res = re.match(r'(\d+\.?\d*)([a-zA-Z]+)', value)
         value = float(res.groups()[0])
@@ -79,11 +82,17 @@ class ProcDockerStats(daemon_base.DaemonBase):
         if units.lower() == UNITS_KB.lower():
             value *= 1000
         elif units.lower() == UNITS_MB.lower():
-            value *= (1000 * 1000)
+            value *= (1000 ** 2)
+        elif units.lower() == UNITS_GB.lower():
+            value *= (1000 ** 3)
+        elif units.lower() == UNITS_TB.lower():
+            value *= (1000 ** 4)
         elif units.lower() == UNITS_MiB.lower():
-            value *= (1024 * 1024)
+            value *= (1024 ** 2)
         elif units.lower() == UNITS_GiB.lower():
-            value *= (1024 * 1024 * 1024)
+            value *= (1024 ** 3)
+        elif units.lower() == UNITS_TiB.lower():
+            value *= (1024 ** 4)
 
         return int(round(value))
 
@@ -204,4 +213,3 @@ def main():
 
 if __name__ == '__main__':
     main()
-