python queue模块详解

队列是我们经常使用的一种数据结构,在python中,也有队列queue这一模块,并且python中提供的queue是线程安全的,可以在多线程处理数据时安全交换数据。

一、queue 模块定义的不同类型的队列

queue 模块定义了以下四种不同类型的队列,它们之间的区别在于数据入队列之后出队列的顺序不同。

1.1 Queue(maxsize=0)

queue是先进先出(FIFO)队列,最早进入队列的数据拥有出队列的优先权。其入参 maxsize 是一个整数,用于设置队列的最大长度。一旦队列达到上限,插入数据将会被阻塞,直到有数据出队列之后才可以继续插入。如果 maxsize 设置为小于或等于零,则队列的长度没有限制。
使用举例:

import queue

q = queue.Queue()  # 创建 Queue 队列,大小不限制
for i in range(5):
    q.put(i)  # 入队
for i in range(5):
    print(q.get())  # 出队打印

1.2 LifoQueue(maxsize=0)

LifoQueue是后进先出(LIFO)队列,最后进入队列的数据拥有出队列的优先权,就像栈一样。其入参 maxsize 与先进先出队列的定义一样。
使用举例:

import queue

q = queue.LifoQueue()  # 创建 LifoQueue 队列,大小不限制
for i in range(5):
    q.put(i)  # 入队
for i in range(5):
    print(q.get())  # 出队打印

1.3 PriorityQueue(maxsize=0)

PriorityQueue是优先级队列,比较队列中每个数据的大小,值最小的数据拥有出队列的优先权。数据一般以元组的形式插入,典型形式为(priority_number, data)。如果队列中的数据没有可比性,那么数据将被包装在一个类中,忽略数据值,仅仅比较优先级数字。入参 maxsize 与先进先出队列的定义一样。
使用举例:

import queue

q = queue.PriorityQueue()  # 创建 PriorityQueue 队列
data1 = (1, 'python')
data2 = (2, 'test')
data3 = (3, 'queue')
style = (data2, data3, data1)
for i in style:
    q.put(i)  # 入队
for i in range(3):
    print(q.get())  # 出队打印

1.4 SimpleQueue(maxsize=0)

SimpleQueue是先进先出类型的简单队列,没有大小限制。由于它是简单队列,相比于 Queue 队列会缺少一些高级功能。
使用举例:

import queue

q = queue.SimpleQueue()  # 创建 SimpleQueue 队列
for i in range(5):
    q.put(i)  # 入队
for i in range(5):
    print(q.get())  # 出队打印

二、出队入队引发的两种异常

2.1 Empty 异常

当队列中没有数据元素时,取出队列中的数据会引发 queue.Empty 异常,主要是不正当使用 get() 和 get_nowait() 引起的。
使用举例:

import queue

try:
    q = queue.Queue(3)  # 设置队列上限为3
    q.put('python')
    q.put('test')
    q.put('queue')
    for i in range(5):  # 出队,取5次,引发 queue.Empty 异常
        print(q.get(block=False))
except queue.Empty:
    print('queue.Empty')

2.2 Full 异常

当队列数据元素容量达到上限时,继续往队列中放入数据会引发 queue.Empty 异常,主要是不正当使用 put() 和 put_nowait() 引起的。
使用举例:

import queue

try:
    q = queue.Queue(3)  # 设置队列上限为3
    q.put('python')
    q.put('test')
    q.put('queue')
    q.put('full error', block=False)  # 队列已满,继续入队会引发 queue.Full 异常
except queue.Full:
    print('queue.Full')

三、队列基本函数方法的使用

注意:以下函数方法为各种队列的通用方法,此处以Queue队列举例。

3.1 qsize()

返回队列中数据元素的个数。
使用举例:

import queue

q = queue.Queue()
q.put('python queue')
print(q.qsize())

3.2 empty()

如果队列为空,返回 True,否则返回 False。
使用举例:

import queue

q = queue.Queue()
print(q.empty())  # 对列为空,返回 True
q.put('python queue')
print(q.empty())  # 对列不为空,返回 False

3.3 full()

如果队列中元素个数达到上限,返回 True,否则返回 False。
使用举例:

import queue

q = queue.Queue(2)  # 定义一个长度为2的队列
print(q.full())  # 元素个数未达到上限,返回 False
q.put('python')
q.put('queue')
print(q.full())  # 元素个数达到上限,返回 True

3.4 put(item, block=True, timeout=None)

参数说明:

  • item,要入队的数据元素。
  • block,当队列中元素个数达到上限继续往里放数据时:如果 block=False,直接引发 queue.Full 异常;如果 block=True,且 timeout=None,则一直等待直到有数据出队列后可以放入数据;如果 block=True,且 timeout=N,N 为某一正整数时,则等待 N 秒,如果队列中还没有位置放入数据就引发 queue.Full 异常。
  • timeout,设置超时时间。

使用举例:

import queue

try:
    q = queue.Queue(2)  # 设置队列上限为2
    q.put('python')
    q.put('test')
    q.put('queue', block = True, timeout = 6) # 队列已满,继续入队,等待6秒后会引发 queue.Full 异常
except queue.Full:
    print('queue.Full')

3.5 put_nowait(item)

put_nowait(item)相当于 put(item, block=False),当队列中元素个数达到上限继续往里放数据时直接引发 queue.Full 异常。
使用举例:

import queue

try:
    q = queue.Queue(2)  # 设置队列上限为2
    q.put_nowait('python')
    q.put_nowait('test')
    q.put_nowait('queue') # 队列已满,继续入队,直接引发 queue.Full 异常
except queue.Full:
    print('queue.Full')

3.6 get(block=True, timeout=None)

从队列中取出数据并返回该数据内容。
参数说明:

  • block,当队列中没有数据元素继续取数据时:如果 block=False,直接引发 queue.Empty 异常;如果 block=True,且 timeout=None,则一直等待直到有数据入队列后可以取出数据;如果 block=True,且 timeout=N,N 为某一正整数时,则等待 N 秒,如果队列中还没有数据放入的话就引发 queue.Empty 异常。
  • timeout,设置超时时间。

使用举例:

import queue

try:
    q = queue.Queue()
    q.get(block = True, timeout = 5) # 队列为空,出队时等待5秒后会引发 queue.Empty 异常
except queue.Empty:
    print('queue.Empty')

3.7 get_nowait()

get_nowait() 相当于 Queue.get(block=False)block,当队列中没有数据元素继续取数据时直接引发 queue.Empty 异常。
使用举例:

import queue
try:
    q = queue.Queue()
    q.get_nowait() # 队列为空,取数据出队时直接引发 queue.Empty 异常
except queue.Empty:
    print('queue.Empty')

四、Queue、LifoQueue 和 PriorityQueue 对象的高级使用方法

SimpleQueue 是 Python 3.7 版本中新加入的特性,与 Queue、LifoQueue 和 PriorityQueue 三种队列相比缺少了 task_done 和 join 的高级使用方法,所以才会取名叫 Simple 了,下面介绍一下 task_done 和 join 的使用方法。

  • task_done,表示队列内的数据元素已经被取出,即每个 get 用于获取一个数据元素, 后续调用 task_done 告诉队列,该数据的处理已经完成。如果被调用的次数多于放入队列中的元素个数,将引发 ValueError 异常。
  • join,一直阻塞直到队列中的所有数据元素都被取出和执行,只要有元素添加到 queue 中就会增加。当未完成任务的计数等于0,join 就不会阻塞。

使用举例:

import queue

q = queue.Queue()
q.put('python')
q.put('-')
q.put('100')
for i in range(3):
    print(q.get())
    q.task_done()  # 如果不执行 task_done,join 会一直处于阻塞状态,等待 task_done 告知它数据的处理已经完成
q.join()

下面是一个经典示例,生产者和消费者线程分别生产数据和消费数据,先生产后消费。采用 task_done 和 join 确保处理信息在多个线程间安全交换,生产者生产的数据能够全部被消费者消费掉。
使用举例:

from queue import Queue
import random
import threading
import time

#生产者线程
class Producer(threading.Thread):
    def __init__(self, t_name, queue):
        threading.Thread.__init__(self, name=t_name)
        self.data=queue
    def run(self):
        for i in range(5):
            print ("%s: %s is producing %d to the queue!" %(time.ctime(), self.getName(), i))
            self.data.put(i)  # 将生产的数据放入队列
            time.sleep(random.randrange(10)/5)
        print ("%s: %s finished!" %(time.ctime(), self.getName()))

#消费者线程
class Consumer(threading.Thread):
    def __init__(self, t_name, queue):
        threading.Thread.__init__(self, name=t_name)
        self.data=queue
    def run(self):
        for i in range(5):
            val = self.data.get()  # 拿出已经生产好的数据
            print ("%s: %s is consuming. %d in the queue is consumed!" %(time.ctime(), self.getName(), val))
            time.sleep(random.randrange(5))
            self.data.task_done() # 告诉队列有关这个数据的任务已经处理完成
        print ("%s: %s finished!" %(time.ctime(), self.getName()))

#主线程
def main():
    queue = Queue()
    producer = Producer('Producer.', queue)
    consumer = Consumer('Consumer.', queue)
    producer.start()
    consumer.start()
    queue.join()  # 阻塞,直到生产者生产的数据全都被消费掉
    producer.join() # 等待生产者线程结束
    consumer.join() # 等待消费者线程结束
    print ('All threads terminate!')
 
if __name__ == '__main__':
    main()
  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值