python多线程 (四) 线程优先级队列

python多线程 (四) 线程优先级队列

Python的Queue模块FIFO、LIFO、LifoQueue、PriorityQueue这些队列实现了锁原语可以在多线程中直接使用。

Queue模块中的常用方法:

• Queue.qsize()返回队列的大小
• Queue.empty()如果队列为空，返回True,反之False
• Queue.full()如果队列满了，返回True,反之False
• Queue.full与maxsize大小对应
• Queue.get([block[,timeout]])获取队列，timeout等待时间
• Queue.get_nowait()相当Queue.get(False)
• Queue.put(item)写入队列，timeout等待时间
• Queue.put_nowait(item)相当Queue.put(item,False)
• Queue.task_done()在完成一项工作之后，Queue.task_done()函数向任务已经完成的队列发送一个信号
• Queue.join()实际上意味着等到队列为空，再执行别的操作

上代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2019/8/5 19:38
# @Author  : Running_Tiger

import queue
import threading
import time

exitFlag = 0

class myThread (threading.Thread):
    def __init__(self, threadID, name, q):
        threading.Thread.__init__(self)
        self.threadID = threadID
        self.name = name
        self.q = q
    def run(self):
        print ("开启线程：" + self.name)
        process_data(self.name, self.q)
        print ("退出线程：" + self.name)

def process_data(threadName, q):
    while not exitFlag:
        queueLock.acquire()
        if not workQueue.empty():
            data = q.get()
            queueLock.release()
            print ("%s processing %s" % (threadName, data))
        else:
            queueLock.release()
        time.sleep(1)

threadList = ["Thread-1", "Thread-2", "Thread-3"]
nameList = ["One", "Two", "Three", "Four", "Five"]
queueLock = threading.Lock()
workQueue = queue.Queue(10)
threads = []
threadID = 1

# 创建新线程
for tName in threadList:
    thread = myThread(threadID, tName, workQueue)
    thread.start()
    threads.append(thread)
    threadID += 1

# 填充队列
queueLock.acquire()
for word in nameList:
    workQueue.put(word)
queueLock.release()

# 等待队列清空
while not workQueue.empty():
    pass

# 通知线程是时候退出
exitFlag = 1

# 等待所有线程完成
for t in threads:
    t.join()
print ("退出主线程")

执行结果：

喜欢就点赞评论+关注吧

感谢阅读，希望能帮助到大家，谢谢大家的支持！