写一个线程池,线程池中每个线程都有自己的阻塞队列
时间: 2023-10-26 14:07:51 浏览: 96
一个线程池的设计
好的,下面是一个简单的线程池实现,每个线程都有自己的阻塞队列:
```python
import threading
import queue
class ThreadPool:
def __init__(self, num_threads):
self.num_threads = num_threads
self.tasks = queue.Queue()
self.threads = []
self.stop_event = threading.Event()
for _ in range(num_threads):
t = threading.Thread(target=self.worker)
self.threads.append(t)
t.start()
def worker(self):
while not self.stop_event.is_set():
try:
task = self.tasks.get(timeout=1)
except queue.Empty:
continue
# Execute the task
result = task()
def submit(self, task):
self.tasks.put(task)
def shutdown(self):
self.stop_event.set()
for t in self.threads:
t.join()
```
在这个线程池中,我们使用 Python 内置的 `queue` 模块来实现阻塞队列。在初始化时,我们创建了一定数量的线程,并且每个线程都会执行 `worker` 方法。这个方法会不断地从阻塞队列中取出任务并执行。如果队列为空,线程会阻塞等待新的任务。
我们还提供了 `submit` 方法,用来向线程池中提交新的任务。每个任务都应该是一个可调用对象,例如一个函数或一个方法。
最后,我们还提供了一个 `shutdown` 方法,用来停止线程池中所有的线程。这个方法会向每个线程发送一个停止信号,并等待所有线程结束。
阅读全文