Queue是python多进程的安全队列,可以使用Queue实现多进程之间的数据传递。
Queue.put()方法说明:
Queue.put(self, obj, block=True, timeout=None) # 用于插入数据到队列中
当block为True,timeout为正值时,该方法会阻塞timeout指定的时间,直到该队列有剩余的空闲时间。如果超时,会抛出Queue.Full异常
当block为False,但队列已满,会抛出Queue.Full异常
Queue.get()方法说明:
Queue.get(self, block=True, timeout=None) # 从队列中取出数据并删除
当block为True,timeout为正值时,在timeout时间内没用取到数据就会抛出Queue.Empty异常
当block为False时,有两种情况:若队列中有数据则取出;若队列为空则抛出Queue.Empty异常。
示例:
from multiprocessing import Process, Queue
import time
def ProducerA(q):
count = 1
while True:
q.put(f"冷饮 {count}")
print(f"{time.strftime('%H:%M:%S')} A 放入:[冷饮 {count}]")
count += 1
time.sleep(1)
def ConsumerB(q):
while True:
print(f"{time.strftime('%H:%M:%S')} B 取出 [{q.get()}]")
time.sleep(5)
if __name__ == '__main__':
q = Queue(maxsize=5)
p = Process(target=ProducerA, args=(q,))
c = Process(target=ConsumerB, args=(q,))
c.start()
p.start()
c.join()
p.join()
输出:
19:07:09 A 放入:[冷饮 1]
19:07:09 B 取出 [冷饮 1]
19:07:10 A 放入:[冷饮 2]
19:07:11 A 放入:[冷饮 3]
19:07:12 A 放入:[冷饮 4]
19:07:13 A 放入:[冷饮 5]
19:07:14 B 取出 [冷饮 2]
19:07:14 A 放入:[冷饮 6]
19:07:15 A 放入:[冷饮 7]
19:07:19 B 取出 [冷饮 3]
19:07:19 A 放入:[冷饮 8]
19:07:24 B 取出 [冷饮 4]
19:07:24 A 放入:[冷饮 9]
19:07:29 B 取出 [冷饮 5]
19:07:29 A 放入:[冷饮 10]
说明:上述代码定义了生产者和消费者函数,设置队列最大容量为5,生产者调用Queue.put()方法放入数据,消费调用Queue.get()方法取出数据,当队列满时,生产者等待,当队列空时,消费者等待。生产的速度和消费的速度可能不一致,但是队列能让生产和消费有条不紊地进行。