From: Michael Tremer <michael.tremer@ipfire.org>
Date: Fri, 27 May 2022 08:57:45 +0000 (+0000)
Subject: daemon: Close queue connection immediately when shutting down
X-Git-Tag: 0.9.28~736
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=0c94fe09078b2839febc0dfbb878abcb7102cf8c;p=pakfire.git

daemon: Close queue connection immediately when shutting down

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

diff --git a/src/pakfire/daemon.py b/src/pakfire/daemon.py
index 35d7c9c15..e0acaa48f 100644
--- a/src/pakfire/daemon.py
+++ b/src/pakfire/daemon.py
@@ -49,6 +49,7 @@ class Daemon(object):
 
 		# Connect to the Pakfire Hub
 		self.hub = self.connect_to_hub()
+		self.queue = None
 
 		# Indicates if this daemon is in running mode.
 		self.__running = True
@@ -80,7 +81,7 @@ class Daemon(object):
 		await self.hub.send_builder_info()
 
 		# Join the job queue
-		await self.hub.queue(self.job_received)
+		self.queue = await self.hub.queue(self.job_received)
 
 		# Run main loop
 		while self.__running:
@@ -103,6 +104,10 @@ class Daemon(object):
 		self.log.info(_("Shutting down..."))
 		self.__running = False
 
+		# Close queue connection so we won't receive any new jobs
+		if self.queue:
+			self.queue.close()
+
 	def spawn_worker(self, *args, **kwargs):
 		"""
 			Spawns a new worker process.