|
|
@ -3,6 +3,7 @@ from __future__ import division, print_function, unicode_literals
|
|
|
|
import threading
|
|
|
|
import threading
|
|
|
|
import abc
|
|
|
|
import abc
|
|
|
|
import uuid
|
|
|
|
import uuid
|
|
|
|
|
|
|
|
import time
|
|
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
import queue
|
|
|
|
import queue
|
|
|
@ -59,7 +60,7 @@ class WorkerThread(threading.Thread):
|
|
|
|
threading.Thread.__init__(self)
|
|
|
|
threading.Thread.__init__(self)
|
|
|
|
|
|
|
|
|
|
|
|
self.dequeued = list()
|
|
|
|
self.dequeued = list()
|
|
|
|
self.daemon = True
|
|
|
|
|
|
|
|
self.doLock = threading.Lock()
|
|
|
|
self.doLock = threading.Lock()
|
|
|
|
self.queue = ImprovedQueue()
|
|
|
|
self.queue = ImprovedQueue()
|
|
|
|
self.num = 0
|
|
|
|
self.num = 0
|
|
|
@ -101,8 +102,18 @@ class WorkerThread(threading.Thread):
|
|
|
|
|
|
|
|
|
|
|
|
# Main thread loop starting the different tasks
|
|
|
|
# Main thread loop starting the different tasks
|
|
|
|
def run(self):
|
|
|
|
def run(self):
|
|
|
|
# this blocks until something is available
|
|
|
|
main_thread = _get_main_thread()
|
|
|
|
item = self.queue.get()
|
|
|
|
while main_thread.is_alive():
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
|
|
|
# this blocks until something is available. This can cause issues when the main thread dies - this
|
|
|
|
|
|
|
|
# thread will remain alive. We implement a timeout to unblock every second which allows us to check if
|
|
|
|
|
|
|
|
# the main thread is still alive.
|
|
|
|
|
|
|
|
# We don't use a daemon here because we don't want the tasks to just be abruptly halted, leading to
|
|
|
|
|
|
|
|
# possible file / database corruption
|
|
|
|
|
|
|
|
item = self.queue.get(timeout=1)
|
|
|
|
|
|
|
|
except queue.Empty as ex:
|
|
|
|
|
|
|
|
time.sleep(1)
|
|
|
|
|
|
|
|
continue
|
|
|
|
|
|
|
|
|
|
|
|
with self.doLock:
|
|
|
|
with self.doLock:
|
|
|
|
# add to list so that in-progress tasks show up
|
|
|
|
# add to list so that in-progress tasks show up
|
|
|
|