zeroMQ初体验-6.多模式数据来源处理方案(multi sockets)

最新推荐文章于 2024-07-25 17:29:26 发布

最新推荐文章于 2024-07-25 17:29:26 发布 · 302 阅读

文章标签：

#Socket #Python

MQ 专栏收录该内容

37 篇文章

订阅专栏

本文探讨了在ZeroMQ中如何优雅地处理来自不同类型的多个消息发布者的数据。通过使用非阻塞模式和轮询机制，可以有效地解决多源消息接收的问题，并确保各个消息源的平衡处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

之前已经讲过，zeroMQ是可以多对多的，但需要成对匹配才行，即多个发布端都是同一种模式，而这里要涉及到的是，多个发布端模式不统一的情况。

文中先给出了一个比较"脏"的处理方式：


import zmq
import time

context = zmq.Context()

receiver = context.socket(zmq.PULL)
receiver.connect("tcp://localhost:5557")

subscriber = context.socket(zmq.SUB)
subscriber.connect("tcp://localhost:5556")
subscriber.setsockopt(zmq.SUBSCRIBE, "10001")

while True:

    while True:
        try:
            rc = receiver.recv(zmq.NOBLOCK)#这是非阻塞模式
        except zmq.ZMQError:
            break

    while True:
        try:
            rc = subscriber.recv(zmq.NOBLOCK)
        except zmq.ZMQError:
            break

显然，如此做既不优雅，还有出现单来源循环不止，另一来源又得不到响应的状况。

自然，官方也做了相应的封装，给了一个相对优雅的实现：


import zmq

context = zmq.Context()

receiver = context.socket(zmq.PULL)
receiver.connect("tcp://localhost:5557")

subscriber = context.socket(zmq.SUB)
subscriber.connect("tcp://localhost:5556")
subscriber.setsockopt(zmq.SUBSCRIBE, "10001")

poller = zmq.Poller()
poller.register(receiver, zmq.POLLIN)
poller.register(subscriber, zmq.POLLIN)

while True:
    socks = dict(poller.poll())

    if receiver in socks and socks[receiver] == zmq.POLLIN:
        message = receiver.recv()

    if subscriber in socks and socks[subscriber] == zmq.POLLIN:
        message = subscriber.recv()