Update app.py

addition of get_cgroupv2_cpu_limit() & get_cgroupv2_memory_limit()

app.py

@@ -39,10 +39,36 @@ def get_container_memory_limit():
         return int(psutil.virtual_memory().total // (1024 * 1024))
     except Exception:
         return None
+
+# = for cgroup v2
+def get_cgroupv2_cpu_limit():
+    try:
+        with open("/sys/fs/cgroup/cpu.max") as f:
+            data = f.read().strip()
+        quota, period = data.split()
+        if quota == "max":
+            print("no CPU limit")
+            return os.cpu_count()  # no limit
+        return int(int(quota) / int(period))
+    except Exception as ex:
+        print(repr(ex))
+        return os.cpu_count()
+
+def get_cgroupv2_memory_limit():
+    try:
+        with open("/sys/fs/cgroup/memory.max") as f:
+            mem_bytes = f.read().strip()
+        if mem_bytes == "max":
+            print("no MEMORY limit")
+            return int(psutil.virtual_memory().total // (1024*1024)) # no limit
+        return int(mem_bytes) // (1024*1024)  # in MB
+    except Exception as ex:
+        print(repr(ex))
+        return int(psutil.virtual_memory().total // (1024*1024))
         
 @app.get("/")
 def greet_json():
-    return {"Hello": "World!", "cpu_count": get_container_cpu_count(), "total_ram_in_MB": get_container_memory_limit()}
+    return {"Hello": "World!", "cpu_count": get_cgroupv2_cpu_limit(), "total_ram_in_MB": get_cgroupv2_memory_limit()}
 
 @app.get("/health")
 def health():