参照菜鸟做了一个线程池
from threading import Thread, Lock
from queue import Queue
import time
import random
thread_active = True
thread_lock = Lock()
class MyThread(Thread):
"""自制线程"""
def __init__(self, thread_name: str, thread_id: int, q: Queue):
"""
初始化
:param thread_name:线程名
:param thread_id: 线程id
:param q: 队列
"""
super(MyThread, self).__init__()
self.thread_name = thread_name
self.thread_id = thread_id
self.q = q
def run(self):
print(f'开启线程:{self.thread_name} - {self.thread_id}')
process_date(self.thread_name, self.q)
print(f'线程结束:{self.thread_name} - {self.thread_id}')
def process_date(thread_name: str, q: Queue):
while thread_active:
thread_lock.acquire()
if q.empty():
thread_lock.release()
# pass
else:
data = q.get()
thread_lock.release()
print(f"线程{thread_name} --- 处理 {data},--{time.ctime(time.time())}")
time.sleep(random.randint(0, 5))
work_queue = Queue()
thread_names = ['one', 'tow', 'three', 'four', 'five']
# 填充线程列表
thread_list = []
for index, name in enumerate(thread_names):
t = MyThread(name, index, work_queue)
thread_list.append(t)
# 启动线程
[t.start() for t in thread_list]
# 填装队列
[work_queue.put(i) for i in range(100)]
# 等待队列被清空
while not work_queue.empty():
pass
thread_active = False
[t.join() for t in thread_list]