进程之间使用队列Queue()通信

多进程之间通信可以使用multiprocessing中的Queue()来完成,下面简单介绍一下Queue()里面的函数及功能

初始化Queue()对象时（例如：q=Queue()），若括号中没有指定最大可接收的消息数量，或数量为负值，那么就代表可接受的消息数量没有上限（直到内存的尽头）；

• Queue.qsize()：返回当前队列包含的消息数量；

• Queue.empty()：如果队列为空，返回True，反之False ；

• Queue.full()：如果队列满了，返回True,反之False；

• Queue.get([block[, timeout]])：获取队列中的一条消息，然后将其从列队中移除，block默认值为True；

1）如果block使用默认值，且没有设置timeout（单位秒），消息列队如果为空，此时程序将被阻塞（停在读取状态），直到从消息列队读到消息为止，如果设置了timeout，则会等待timeout秒，若还没读取到任何消息，则抛出"Queue.Empty"异常；

2）如果block值为False，消息列队如果为空，则会立刻抛出"Queue.Empty"异常；

• Queue.get_nowait()：相当Queue.get(False)；

• Queue.put(item,[block[, timeout]])：将item消息写入队列，block默认值为True；

1）如果block使用默认值，且没有设置timeout（单位秒），消息列队如果已经没有空间可写入，此时程序将被阻塞（停在写入状态），直到从消息列队腾出空间为止，如果设置了timeout，则会等待timeout秒，若还没空间，则抛出"Queue.Full"异常；

2）如果block值为False，消息列队如果没有空间可写入，则会立刻抛出"Queue.Full"异常；

• Queue.put_nowait(item)：相当Queue.put(item, False)；

下面请看一个简单的例子:

其中进程p_get执行函数get_data从队列q中获取数据,进程p_put执行函数put_data向队列q写入数据.

from multiprocessing import Queue, Process

import time


def get_data(q):
    """从队列q里面获取数据"""
    while True:
        if q.qsize:
            print('取出%s' % q.get())
        time.sleep(0.6)


def put_data(q):
    """往队列q里面添加数据"""
    count = 0
    while True:
        count += 1
        if q.qsize() < 100:
            q.put("第%s个数据" % count)
            print("放入第%s个数据" % count)
        time.sleep(0.5)


def main():
    q = Queue()
    p_get = Process(target=get_data, args=(q,))
    p_get.start()
    p_put = Process(target=put_data, args=(q,))
    p_put.start()
    p_put.join()
    p_get.join()


if __name__ == '__main__':
    main()