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 b0a8cbd6a5f2b507002b72fd4fe9d939a91617b8..8bd128c3159671ba49dd7996461ef6e8eec4dafe 100644 (file)
--- 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 821cde25e58dd2a151269feb6f856ddb740225e0..a32c4bed6e36aff20cc3d889eca1797ad4c8f49f 100644 (file)
--- 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)