【吴师傅的Python面包店】Day 6

"""吴师傅为每位师傅分配了专属的一个售卖窗口，客户可以在喜欢师傅的窗口买到想要的面包了。"""

import time
import threading
import random

import zmq

TOTAL = []
END_FLAG = False
PRODUCER_NUM = 3
CONSUMER_NUM = 3

def producer(topic, pub_socket):
    """生产面包。"""
    num = 0
    while True:
        time.sleep(0.2)
        pub_socket.send(topic.encode('utf8'))
        # print('Producer: I produced one:', topic)
        num += 1
        if num >= 3:
            # 师傅做3个面包就要下班了。
            break

def consumer(topic, context, freq):
    global END_FLAG
    sub_socket = context.socket(zmq.SUB)
    sub_socket.connect("tcp://localhost:5555")
    sub_socket.setsockopt(zmq.SUBSCRIBE, topic.encode('utf8'))
    sub_socket.setsockopt(zmq.RCVTIMEO, 2000)    # 设置接受超时
    
    while not END_FLAG:
        msg = None
        try:
            msg = sub_socket.recv()
        except Exception as e:
            pass
        else:
            if msg:
                print('Consumer: I had buy:', msg.decode())
        time.sleep(freq)

def run():
    context = zmq.Context()
    pub_socket = context.socket(zmq.PUB)
    pub_socket.bind("tcp://*:5555")
    producers = []
    consumers = []
    for i in range(PRODUCER_NUM):
        # 每个师傅对应一条买面包店队伍
        topic = 'P:%s' % i
        p = threading.Thread(target=producer, args=(topic, pub_socket))
        producers.append(p)
        p.start()

        c = threading.Thread(target=consumer, args=(topic, context, i))
        consumers.append(c)
        c.start()
    for p in producers:
        p.join()
    global END_FLAG
    END_FLAG = True    # 关门了，顾客消费也要停止
    for c in consumers:
        c.join()

if __name__ == "__main__":
    run()