saltstack系列(六)——zmq扩展(二)

问题

　　我们已经熟练的掌握了REQ/REP模式，它是一个一对多的模式，一个REP对应多个REQ。

　　但是现实工作中，我们会遇到这样的难题，一个REP无法满足REQ的提问，因为REQ太多了，虽然可以增加一个REP，但是，这样做会带来很多问题。两个REP的端口不可能是一个，那么就需要将原来的一些REQ与这个新的REQ连接，这里面的工作量可想而知。那么我们能不能采取一种简单方便的方式，使得REQ和REP都可以灵活的增加减少而不对系统造成负面影响导致工作量上升呢？

　　面对这样的要求，我们需要一个代理。让REQ和REP都连接这个代理，一切繁杂的事情都由这个代理进行。我从《ZeroMQ 云时代极速消息通信库》这本书里拍下了扩展的请求-应答解析图。

　　在扩展的请求-应答系统里，REQ和REP并不直接通信，而是通过ROUTER和DEALER，ROUTER是路由器，DEALER是经销商，所有的请求到达路由器以后公平的排队，然后由经销商负载均衡后发送给REP服务器，服务器应答的结果再由经销商和路由器返回给REQ客户端。

　　在这个系统里，你可以随意的增加和减少REQ，可以随意的增加和减少REP，而不必为此付出额外的工作量。

　　到了这一步，聪明的朋友已经发现了问题了，一个DEALER怎么可以连接多个REP呢？没错，一个DEALER不能连接多个REP，但多个REP可以连接一个DEALER。在一对多的REQ/REP模型中，REP都是等待REQ来连接的，但在我们上图所示的系统中，REP是主动连接DEALER的，而不是等待DEALER来连接它。

实例代码：

客户端：

#coding=utf-8  
''''' 
发起请求 
'''  
import zmq  
  
#  Prepare our context and sockets  
context = zmq.Context()  
socket = context.socket(zmq.REQ)  
#这一次，我们不连接REP，而是连接ROUTER，多个REP连接一个ROUTER  
socket.connect("tcp://localhost:8000")  
  
#  发送问题给ROUTER  
for request in range(1,11):  
    socket.send(b"Hello")  
    message = socket.recv()  
    print("Received reply %s [%s]" % (request, message))  
socket.close()  
context.term()  

中间代理

#coding=utf-8  
 
import zmq  
  
# Prepare our context and sockets  
context = zmq.Context()  
  
frontend = context.socket(zmq.ROUTER)  
backend = context.socket(zmq.DEALER)  
frontend.bind("tcp://*:8000")  
backend.bind("tcp://*:8001")  
  
# Initialize poll set  
poller = zmq.Poller()  
poller.register(frontend, zmq.POLLIN)  
poller.register(backend, zmq.POLLIN)  
  
# Switch messages between sockets  
while True:  
    socks = dict(poller.poll())  
      
    # frontend 收到了提问后，由backend发送给REP端  
    if socks.get(frontend) == zmq.POLLIN:  
        message = frontend.recv_multipart()  
        backend.send_multipart(message)  
      
    # backend 收到了回答后，由frontend发送给REQ端  
    if socks.get(backend) == zmq.POLLIN:  
        message = backend.recv_multipart()  
        frontend.send_multipart(message)  

服务端

#coding=utf-8  
''''' 
收到请求后回复world 
'''  
import zmq  
  
context = zmq.Context()  
socket = context.socket(zmq.REP)  
# REP连接的是DEALER  
socket.connect("tcp://localhost:8001")  
  
while True:  
    message = socket.recv()  
    print("Received request: %s" % message)  
    socket.send(b"World")  

 在服务端，REP没有调用bind函数，而是调用了connect函数去连接DEALER。请求的排队和服务端的负载均衡均由中间的代理完成了，事情就是这样的简单。