day9-队列

一、队列类型

import Queue

# 1.先进先出
q = Queue.Queue(maxsize=3)
# 2.后进先出
# q = Queue.LifoQueue(maxsize=3)

q.put("d1")
q.put("d2")
q.put("d3")
# 获取队列大小
print q.qsize()
print q.get()
print q.get()
print q.get()

# 此时q为空队列，如果用q.get()会卡死
# q.get_nowait()不等待，空队列直接抛出异常
# print q.get_nowait()
# 设置超时时间，超时空队列会抛出异常
# print q.get(timeout=2)

# 3.存储数据时可设置优先级
q = Queue.PriorityQueue(maxsize=3)
q.put((10, 'd1'))
q.put((-1, 'd2'))
q.put((5, 'd3'))
print q.get()
print q.get()
print q.get()

二、生产者消费者模型

import threading
import Queue
q = Queue.Queue()


def Producer(name):
    for i in range(1, 10):
        q.put("baozi%s" % i)
        print("%s produce baozi%s" % (name, i))
    print("waiting....")
    # 队列阻塞直至包子全部消费完毕
    q.join()
    print("all baozi has gone...")


def Consumer(name):
    while q.qsize() > 0:
        q.get()
        print("[%s] get baozi and eat it" % name)
        # 告知队列已消费一个包子
        q.task_done()


p = threading.Thread(target=Producer, args=("Nana", ))
p.start()

c = Consumer("Cici")

三、边生产边消费

import time
import random
import Queue
import threading
q = Queue.Queue()


def Producer(name):
    count = 0
    while count < 20:
        time.sleep(random.randrange(3))
        q.put(count)
        print('Producer %s has produced %s baozi..' % (name, count))
        count += 1


def Consumer(name):
  count = 0
  while count < 20:
    time.sleep(random.randrange(4))
    if not q.empty():
        data = q.get()
        print(data)
        print('\033[32;1mConsumer %s has eat %s baozi...\033[0m' % (name, data))
    else:
        print("-----no baozi anymore----")
    count += 1

p1 = threading.Thread(target=Producer, args=('A',))
c1 = threading.Thread(target=Consumer, args=('B',))
p1.start()
c1.start()