change shutdown handling to be able to join()

This commit is contained in:
Jakob Ketterl 2021-01-22 17:34:35 +01:00
parent ae1287b8a2
commit f29f7b20e3
1 changed files with 6 additions and 4 deletions

View File

@ -46,8 +46,6 @@ class QueueWorker(threading.Thread):
job = self.queue.get()
if job is PoisonPill:
self.doRun = False
# put the poison pill back on the queue for the next worker
self.queue.put(PoisonPill)
else:
try:
job.run()
@ -102,10 +100,14 @@ class DecoderQueue(Queue):
while not self.empty():
job = self.get()
job.unlink()
self.task_done()
except Empty:
pass
# put() PoisonPill to tell workers to shut down
self.put(PoisonPill)
# put() a PoisonPill for all active workers to shut them down
for w in self.workers:
if w.is_alive():
self.put(PoisonPill)
self.join()
def put(self, item, **kwars):
self.inCounter.inc()