Move ThreadPoolExecutor() creation into own method

Move ThreadPoolExecutor() creation into it's own method so it is easier
to override when subclassing.
This commit is contained in:
Andrew Widdersheim 2019-02-20 15:55:31 -05:00 committed by Randall Leeds
parent 5680320e5c
commit a2a8bc1ae6

View File

@ -85,11 +85,15 @@ class ThreadWorker(base.Worker):
"Check the number of worker connections and threads.")
def init_process(self):
self.tpool = futures.ThreadPoolExecutor(max_workers=self.cfg.threads)
self.tpool = self.get_thread_pool()
self.poller = selectors.DefaultSelector()
self._lock = RLock()
super(ThreadWorker, self).init_process()
def get_thread_pool(self):
"""Override this method to customize how the thread pool is created"""
return futures.ThreadPoolExecutor(max_workers=self.cfg.threads)
def handle_quit(self, sig, frame):
self.alive = False
# worker_int callback