feat: add worker mem metrics

monitoring_prometheus/controllers/prometheus_metrics.py

@@ -5,8 +5,11 @@
 
 from odoo.http import Controller, route
 
+from ..models.psutils_helpers import get_process_info
+
 
 class PrometheusController(Controller):
     @route("/metrics", auth="public")
     def metrics(self):
+        get_process_info()
         return generate_latest()

monitoring_prometheus/models/__init__.py

@@ -1 +1,2 @@
 from . import ir_http
+from . import psutils_helpers

monitoring_prometheus/models/ir_http.py

@@ -9,6 +9,7 @@
 REQUEST_TIME = Summary(
     "request_latency_sec", "Request response time in sec", ["query_type"]
 )
+
 LONGPOLLING_COUNT = Counter("longpolling", "Longpolling request count")
 
 

monitoring_prometheus/models/psutils_helpers.py

@@ -0,0 +1,33 @@
+import psutil
+from prometheus_client import Gauge
+
+MEMORY_USAGE_VMS = Gauge("memory_user_vms_mb", "Memory usage in MB", ["process", "pid"])
+
+MEMORY_USAGE_RSS = Gauge("memory_user_rss_mb", "Memory usage in MB", ["process", "pid"])
+
+
+def get_process_info():
+    for process in psutil.process_iter(
+        ["pid", "name", "memory_full_info", "cmdline", "nice"]
+    ):
+        try:
+            if process.info["memory_full_info"]:
+                if process.info["nice"] == 10:
+                    ProcessLabel = "workercron"
+                elif process.info["pid"] == 1:
+                    ProcessLabel = "dispatcher"
+                elif any(["gevent" in x for x in process.info["cmdline"]]):
+                    ProcessLabel = "gevent"
+                elif any(["odoo" in x for x in process.info["cmdline"]]):
+                    ProcessLabel = "workerhttp"
+                else:
+                    ProcessLabel = "other"
+                MEMORY_USAGE_VMS.labels(ProcessLabel, process.info["pid"]).set(
+                    process.info["memory_full_info"].rss / 1024 / 1024
+                )
+                MEMORY_USAGE_RSS.labels(ProcessLabel, process.info["pid"]).set(
+                    process.info["memory_full_info"].vms / 1024 / 1024
+                )
+
+        except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
+            continue