Python Queue模块详解

Linux评论36阅读模式

Python中,队列是线程间最常用的交换数据的形式。Queue模块是提供队列操作的模块,虽然简单易用,但是不小心的话,还是会出现一些意外。

创建一个“队列”对象
import Queue
q = Queue.Queue(maxsize = 10)
Queue.Queue类即是一个队列的同步实现。队列长度可为无限或者有限。可通过Queue的构造函数的可选参数maxsize来设定队列长度。如果maxsize小于1就表示队列长度无限。

将一个值放入队列中
q.put(10)
调用队列对象的put()方法在队尾插入一个项目。put()有两个参数,第一个item为必需的,为插入项目的值;第二个block为可选参数,默认为
1。如果队列当前为空且block为1,put()方法就使调用线程暂停,直到空出一个数据单元。如果block为0,put方法将引发Full异常。

将一个值从队列中取出
q.get()
调用队列对象的get()方法从队头删除并返回一个项目。可选参数为block,默认为True。如果队列为空且block为True,get()就使调用线程暂停,直至有项目可用。如果队列为空且block为False,队列将引发Empty异常。

Python Queue模块有三种队列及构造函数:
1、Python Queue模块的FIFO队列先进先出。     class Queue.Queue(maxsize)
2、LIFO类似于堆,即先进后出。                         class Queue.LifoQueue(maxsize)
3、还有一种是优先级队列级别越低越先出来。    class Queue.PriorityQueue(maxsize)

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

范例:
实现一个线程不断生成一个随机数到一个队列中(考虑使用Queue这个模块)
实现一个线程从上面的队列里面不断的取出奇数
实现另外一个线程从上面的队列里面不断取出偶数

  1. #!/usr/bin/env python
  2. #coding:utf8
  3. import random,threading,time
  4. from Queue import Queue
  5. #Producer thread
  6. class Producer(threading.Thread):
  7. def __init__(self, t_name, queue):
  8. threading.Thread.__init__(self,name=t_name)
  9. self.data=queue
  10. def run(self):
  11. for i in range(10): #随机产生10个数字 ,可以修改为任意大小
  12. randomnum=random.randint(1,99)
  13. print "%s: %s is producing %d to the queue!" % (time.ctime(), self.getName(), randomnum)
  14. self.data.put(randomnum) #将数据依次存入队列
  15. time.sleep(1)
  16. print "%s: %s finished!" %(time.ctime(), self.getName())
  17. #Consumer thread
  18. class Consumer_even(threading.Thread):
  19. def __init__(self,t_name,queue):
  20. threading.Thread.__init__(self,name=t_name)
  21. self.data=queue
  22. def run(self):
  23. while 1:
  24. try:
  25. val_even = self.data.get(1,5) #get(self, block=True, timeout=None) ,1就是阻塞等待,5是超时5秒
  26. if val_even%2==0:
  27. print "%s: %s is consuming. %d in the queue is consumed!" % (time.ctime(),self.getName(),val_even)
  28. time.sleep(2)
  29. else:
  30. self.data.put(val_even)
  31. time.sleep(2)
  32. except: #等待输入,超过5秒 就报异常
  33. print "%s: %s finished!" %(time.ctime(),self.getName())
  34. break
  35. class Consumer_odd(threading.Thread):
  36. def __init__(self,t_name,queue):
  37. threading.Thread.__init__(self, name=t_name)
  38. self.data=queue
  39. def run(self):
  40. while 1:
  41. try:
  42. val_odd = self.data.get(1,5)
  43. if val_odd%2!=0:
  44. print "%s: %s is consuming. %d in the queue is consumed!" % (time.ctime(), self.getName(), val_odd)
  45. time.sleep(2)
  46. else:
  47. self.data.put(val_odd)
  48. time.sleep(2)
  49. except:
  50. print "%s: %s finished!" % (time.ctime(), self.getName())
  51. break
  52. #Main thread
  53. def main():
  54. queue = Queue()
  55. producer = Producer('Pro.', queue)
  56. consumer_even = Consumer_even('Con_even.', queue)
  57. consumer_odd = Consumer_odd('Con_odd.',queue)
  58. producer.start()
  59. consumer_even.start()
  60. consumer_odd.start()
  61. producer.join()
  62. consumer_even.join()
  63. consumer_odd.join()
  64. print 'All threads terminate!'
  65. if __name__ == '__main__':
  66. main()

Mon Jun 17 15:43:39 CST 2013

 
  • 本文由 yeho 发表于 2013-06-17
  • 转载请务必保留本文链接:https://linuxeye.com/334.html
脚本

腾讯云COS上传、批量删除工具(Python)

腾讯云对象存储COS是类似于阿里云OSS,相比OSS,COS提供每月免费额度:存储空间50G、外网访问流量10G(内网免费)、免费读请求100万次、写请求10万次。对网站备份来说不错,但是,腾讯云提供...
Python多线程抓取代理服务器 Linux

Python多线程抓取代理服务器

Python作为一门功能强大的脚本语言来说,经常被用来写爬虫程序,下面是Python爬虫多线程抓取代理服务器。 年前是用 //linuxeye.com/340.html 来抓取代理服务器的,谁知道过完...
匿名

发表评论

匿名网友
确定

拖动滑块以完成验证
加载中...