importzmqimporttimeimportmultiprocessingfromzmqimportFlag,PollEvent,Eventfromzmq.utils.monitorimportrecv_monitor_messagedefpub_process():# Create a context and a PUB socketcontext=zmq.Context()pub_socket=context.socket(zmq.PUB)pub_socket.bind("tcp://*:5555")# Create monitor socketmonitor_socket=pub_socket.get_monitor_socket()print("Waiting for subscriber to connect...")# Wait for the subscriber to connectwhileTrue:result=monitor_socket.poll(10)ifresult!=PollEvent.POLLIN:continueevent=recv_monitor_message(monitor_socket,Flag.DONTWAIT)ifevent["event"]==Event.ACCEPTED:print("Subscriber connected!")monitor_socket.close()break# Now send messages after the subscriber is connectedforiinrange(5):message=f"Message{i}"print(f"Publishing:{message}")pub_socket.send_string(message)time.sleep(1)# Send a stop message to the subscriber, then close the socketpub_socket.send_string("STOP")pub_socket.close()context.term()print("Publisher process finished")defsub_process():# Create a context and a SUB socketcontext=zmq.Context()sub_socket=context.socket(zmq.SUB)sub_socket.setsockopt_string(zmq.SUBSCRIBE,"")sub_socket.connect("tcp://localhost:5555")# Poller setup to check for messagespoller=zmq.Poller()poller.register(sub_socket,zmq.POLLIN)print("Waiting for messages...")# Wait for messageswhileTrue:socks=dict(poller.poll())ifsub_socketinsocks:message=sub_socket.recv_string()ifmessage=="STOP":breakprint(f"Received:{message}")# Close the socket and terminate the contextsub_socket.close()context.term()print("Subscriber process finished")defmain():pub_proc=multiprocessing.Process(target=pub_process,daemon=True)sub_proc=multiprocessing.Process(target=sub_process,daemon=True)pub_proc.start()sub_proc.start()pub_proc.join()sub_proc.join()if__name__=="__main__":main()