Add CPU affinity management and garbage collection to optimize performance

2026-01-01 22:05:17 +06:00
parent a8a4fb2a0e
commit 5e3505207b
2 changed files with 36 additions and 1 deletions
@@ -0,0 +1,31 @@
+import os
+import psutil
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+def set_cpu_affinity():
+    try:
+        process = psutil.Process()
+        cpu_count = os.cpu_count()
+        logger.info(f"System has {cpu_count} CPU cores")
+        if cpu_count >= 4:
+            affinity_cores = [cpu_count - 2, cpu_count - 1]
+        elif cpu_count >= 2:
+            affinity_cores = [cpu_count - 1]
+        else:
+            affinity_cores = [0]
+
+        process.cpu_affinity(affinity_cores)
+        logger.info(f"CPU affinity set to cores: {affinity_cores}")
+
+        if os.name == "nt":  # Windows
+            process.nice(psutil.HIGH_PRIORITY_CLASS)
+            logger.info("Process priority set to HIGH")
+        else:  # Linux/Unix
+            process.nice(-10)  # Higher priority (lower nice value)
+            logger.info("Process nice value set to -10 (high priority)")
+
+    except Exception as e:
+        logger.warning(f"Failed to set CPU affinity: {e}")