Question

我看了各种ZMQ消息传递模式，我不确定哪一个会为我的项目做什么。我想做的就是能够连接到服务器并发送命令（客户端永远不会收到任何东西）。在服务器端，我希望能够检查是否有消息，如果有消息，则处理消息，否则继续执行其他内容不阻止。这样，即使没有连接客户端，服务器也可以继续工作。

#client.py
while(True):
    select = raw_input()
    if select == "1":
       socket.send(msg1)
    elif select == "2":
       socket.send(msg2)
    ...

#server.py
while(True):
    msg = socket.recv() #should not block
    if msg == ...
        #do stuff
    #do other stuff

那么我应该与ZMQ一起使用哪种模式呢？示例代码将不胜感激。

Answer 1

首先，由于您只希望与一个接收消息的套接字进行单向通信，这通常意味着PUSH-PULL。这是客户端的一个版本：

import zmq

ctx = zmq.Context.instance()
s = ctx.socket(zmq.PUSH)
url = 'tcp://127.0.0.1:5555'
s.connect(url)

while True:
    msg = raw_input("msg > ")
    s.send(msg)
    if msg == 'quit':
        break

所以PUSH套接字发送我们从raw_input获得的消息。应该清楚如何更改该逻辑以生成所需的消息。一点好处是如果你输入'quit'，客户端和服务器都会退出。

根据应用程序的复杂程度，有多种方法可以执行非阻塞服务器。我将展示一些例子，从最基本的一个到最强大的/可扩展的。

所有这些服务器示例都假设在顶部，设置服务器的PULL套接字：

import time
import zmq

ctx = zmq.Context.instance()
s = ctx.socket(zmq.PULL)
url = 'tcp://127.0.0.1:5555'
s.bind(url)

第一个示例是简单的非阻塞recv，如果没有准备好接收消息，则会引发zmq.Again异常：

# server0.py

while True:
    try:
        msg = s.recv(zmq.NOBLOCK) # note NOBLOCK here
    except zmq.Again:
        # no message to recv, do other things
        time.sleep(1)
    else:
        print("received %r" % msg)
        if msg == 'quit':
            break

但是这种模式很难超越非常简单的情况。第二个示例使用Poller来检查套接字上的事件：

# server1.py

poller = zmq.Poller()
poller.register(s)

while True:
    events = dict(poller.poll(0))
    if s in events:
        msg = s.recv()
        print("received %r" % msg)
        if msg == 'quit':
            break
    else:
        # no message to recv, do other things
        time.sleep(1)

在这个玩具示例中，这与第一个非常相似。但是，与第一个不同的是，通过进一步调用poller.register或将除0之外的超时传递给poller.poll，很容易扩展到许多套接字或事件。

最后一个示例使用了一个eventloop，实际上是在消息到达时注册回调。您可以使用这种模式构建非常复杂的应用程序，这是一种编写代码的相当简单的方法，只有在有工作要做时才能正常工作。

# server2.py

from zmq.eventloop import ioloop
from zmq.eventloop.zmqstream import ZMQStream

def print_msg(msg):
    print("received %r" % ' '.join(msg))
    if msg[0] == 'quit':
        ioloop.IOLoop.instance().stop()

# register the print_msg callback to be fired
# whenever there is a message on our socket
stream = ZMQStream(s)
stream.on_recv(print_msg)

# do other things in the meantime
tic = time.time()
def do_other_things():
    print("%.3f" % (time.time() - tic))

pc = ioloop.PeriodicCallback(do_other_things, 1000)
pc.start()

# start the eventloop
ioloop.IOLoop.instance().start()

这是一些处理zmq消息而不会阻塞的基本方法。您可以抓住这些示例together as a gist。

PyZMQ单向通信

1 个答案: