mirror of
https://github.com/Dispatcharr/Dispatcharr.git
synced 2026-07-17 16:50:53 +00:00
c1ff5e35aa
- Introduced a new module to determine the process role based on command-line arguments, enhancing PostgreSQL connection labeling for better monitoring. - Updated the database connection parameters to include the application name derived from the process role. - Added unit tests to validate the process role detection logic for different scenarios, including Celery and uWSGI.
34 lines
1,016 B
Python
34 lines
1,016 B
Python
"""Label Postgres connections by Dispatcharr process role (for pg_stat_activity)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
|
|
|
|
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 os.environ.get("GEVENT_SUPPORT"):
|
|
return "uwsgi"
|
|
return "django"
|
|
|
|
|
|
def db_application_name() -> str:
|
|
return f"Dispatcharr-{get_process_role()}-{os.getpid()}"
|