本文主要是介绍WithThread多线程识别某一线程受阻,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
从源代码修改
from threading import Thread
from queue import Queueclass WithThread:def __init__(self, thread_num):self.thread_num = thread_numself.task_queue = Queue()self.threads = []self.is_running = Falseself.thread_status = [True] * thread_num # 新增一个列表记录每个线程的状态def put(self, func, args=()):if not self.is_running:self.start()self.task_queue.put((func, args))def start(self):self.is_running = Truefor i in range(self.thread_num):t = Thread(target=self._run, args=(i,)) # 将线程ID作为参数传入方法中t.start()self.threads.append(t)def stop(self):self.is_running = Falsefor t in self.threads:t.join()def _run(self, i):while self.is_running:try:func, args = self.task_queue.get(timeout=1)except Exception as e:# 捕获Empty异常continuetry:func(*args)except Exception as e:print("任务执行失败: ", e)finally:self.task_queue.task_done()self.thread_status[i] = True # 在任务完成后将状态标记为Truedef get_thread_status(self):return self.thread_status # 返回线程状态列表
新增了一个thread_status
列表来记录每个线程的状态,并在每个线程的_run
方法中更新状态。另外,我们还新增了一个get_thread_status
方法,用于获取所有线程的状态。
from threading import Thread
from queue import Queueclass WithThread:def __init__(self, thread_num, max_queue_size=100):self.thread_num = thread_numself.task_queue = Queue()self.max_queue_size = max_queue_sizeself.threads = []self.is_running = Falsedef put(self, func, args=()):if not self.is_running:self.start()self.task_queue.put((func, args))def start(self):self.is_running = Truefor i in range(self.thread_num):t = Thread(target=self._run)t.start()self.threads.append(t)def stop(self):self.is_running = Falsefor t in self.threads:t.join()def _run(self):while self.is_running:try:func, args = self.task_queue.get(timeout=1)except Exception as e:# 捕获Empty异常continuetry:func(*args)except Exception as e:print("任务执行失败: ", e)finally:self.task_queue.task_done()# 判断队列长度是否超出阈值if self.task_queue.qsize() > self.max_queue_size:print("当前线程受阻,队列长度:", self.task_queue.qsize())
新增了一个max_queue_size
参数,并在线程的_run
方法中判断队列长度是否超过该阈值。如果队列长度超过阈值,就认为该线程受阻。
这篇关于WithThread多线程识别某一线程受阻的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!