Python线程安全 - 使用生产者消费者模式加消息队列

代码

# -*- coding: utf-8 -*-
#  @Author  : markadc

from queue import Queue
import threading
from concurrent.futures import ThreadPoolExecutor

q = Queue(100)
mark = False  # 生产者是否结束
tname = lambda: threading.currentThread().getName()


def prod():
    for n in range(10000):
        q.put(str(n))
    global mark
    mark = True
    print('({}) 生产者结束 \n'.format(tname()))


def cons():
    while True:
        if q.empty() and mark:  # 消费者已全部消费完成，且生产者已结束
            print('({}) 消费者结束 \n'.format(tname()))
            break
        v = q.get()
        print('({}) 拿到了 {} \n'.format(tname(), v))


def cons_():
    cons_num = 3
    pool = ThreadPoolExecutor(max_workers=cons_num)
    for _ in range(cons_num):
        pool.submit(cons)


if __name__ == '__main__':
    threading.Thread(target=prod).start()
    threading.Thread(target=cons_).start()