mirror of
https://github.com/Dispatcharr/Dispatcharr.git
synced 2026-07-17 16:50:53 +00:00
a7b4db83fa
- Added a new function `_is_uwsgi_worker` to accurately identify when the application is running within a uWSGI worker. - Updated `get_process_role` to label processes as "uwsgi" based on the new function or command-line arguments, improving process role accuracy. - Expanded unit tests to cover various scenarios for uWSGI role detection, ensuring robust validation of the new logic.
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
"""Label Postgres connections by Dispatcharr process role (for pg_stat_activity)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
|
|
|
|
def _is_uwsgi_worker() -> bool:
|
|
"""True when running inside a uWSGI worker (not master or a non-uWSGI process)."""
|
|
try:
|
|
import uwsgi
|
|
except ImportError:
|
|
return False
|
|
return uwsgi.worker_id() > 0
|
|
|
|
|
|
def get_process_role(argv: list[str] | None = None) -> str:
|
|
argv = argv if argv is not None else sys.argv
|
|
argv0 = os.path.basename(argv[0]) if argv else ""
|
|
cmdline = " ".join(argv)
|
|
|
|
if "celery" in argv0 or any("celery" in arg for arg in argv):
|
|
if "beat" in cmdline:
|
|
return "celery-beat"
|
|
if "-Q" in argv:
|
|
try:
|
|
if "dvr" in argv[argv.index("-Q") + 1]:
|
|
return "celery-dvr"
|
|
except (IndexError, ValueError):
|
|
pass
|
|
return "celery-worker"
|
|
if "daphne" in argv0:
|
|
return "daphne"
|
|
if argv0 == "manage.py" and len(argv) > 1:
|
|
return f"manage-{argv[1]}"
|
|
if _is_uwsgi_worker() or argv0 == "uwsgi":
|
|
return "uwsgi"
|
|
return "django"
|
|
|
|
|
|
def db_application_name() -> str:
|
|
return f"Dispatcharr-{get_process_role()}-{os.getpid()}"
|