From: Michael Tremer <michael.tremer@ipfire.org>
Date: Fri, 27 May 2022 10:09:58 +0000 (+0000)
Subject: daemon: Set job ID in process title
X-Git-Tag: 0.9.28~728
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=cb7def8aa6ce19c266460b644bca9d61defb49ff;p=pakfire.git

daemon: Set job ID in process title

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>
---

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)