python多进程中使用生产者消费者模式对进程的执行顺序进行控制

代码

from multiprocessing import Process, Queue


def consumer(q):
    while True:
        res = q.get()
        if res is None: break  # 收到结束信号则结束
        print('\033[45m 吃 %s\033[0m' % ( res))


def producer(name, q):
        res = '%s' % (name)
        q.put(res)
        print('\033[44m 生产了 %s\033[0m' % ( res))


if __name__ == '__main__':
    q = Queue()
    # 生产者们:即厨师们
    p1 = Process(target=producer, args=('包子', q))
    p2 = Process(target=producer, args=('馒头', q))
    p3 = Process(target=producer, args=('花卷', q))

    # 消费者们:即吃货们
    c1 = Process(target=consumer, args=(q,))
    c2 = Process(target=consumer, args=(q,))

    # 开始
    p1.start()
    p2.start()
    p3.start()
    c1.start()
    c2.start()

    p1.join()  # 必须保证生产者全部生产完毕,才应该发送结束信号
    p2.join()
    p3.join()
    q.put(None)  # 有几个消费者就应该发送几次结束信号None
    q.put(None)  # 发送结束信号

输出