From cb7def8aa6ce19c266460b644bca9d61defb49ff Mon Sep 17 00:00:00 2001 From: Michael Tremer Date: Fri, 27 May 2022 10:09:58 +0000 Subject: [PATCH] daemon: Set job ID in process title Signed-off-by: Michael Tremer --- configure.ac | 1 + src/pakfire/daemon.py | 9 +++++++++ 2 files changed, 10 insertions(+) diff --git a/configure.ac b/configure.ac index b0a8cbd6a..8bd128c31 100644 --- a/configure.ac +++ b/configure.ac @@ -234,6 +234,7 @@ AM_PATH_PYTHON([3.6]) AX_PYTHON_MODULE([cpuinfo], [fatal]) AX_PYTHON_MODULE([psutil], [fatal]) +AX_PYTHON_MODULE([setproctitle], [fatal]) AX_PYTHON_MODULE([systemd], [fatal]) AX_PYTHON_MODULE([tornado], [fatal]) diff --git a/src/pakfire/daemon.py b/src/pakfire/daemon.py index 821cde25e..a32c4bed6 100644 --- a/src/pakfire/daemon.py +++ b/src/pakfire/daemon.py @@ -6,6 +6,7 @@ import json import logging import multiprocessing import os +import setproctitle import signal import socket import sys @@ -225,6 +226,14 @@ class Worker(multiprocessing.Process): # Register signal handlers self.register_signal_handlers() + # Extract the job id + job_id = self.data.get("id") + if not job_id: + raise ValueError("Did not receive a job ID") + + # Set the process title + setproctitle.setproctitle("pakfire-worker job %s" % job_id) + # XXX Do something for now import time time.sleep(10) -- 2.47.3