saltstack系列(三)——zmq订阅/发布模式

zmq订阅发布模式

server端代码：

#coding=utf-8  
''''' 
服务端，发布模式 
'''  
import zmq  
from random import randrange  
  
context = zmq.Context()  
socket = context.socket(zmq.PUB)  
socket.bind("tcp://127.0.0.1:8000")  
  
while True:  
    zipcode = randrange(1, 100000)  
    temperature = randrange(-80, 135)  
    relhumidity = randrange(10, 60)  
  
    socket.send("%i %i %i" % (zipcode,temperature , relhumidity))

客户端代码：

#coding=utf-8  
''''' 
订阅模式，如果设置了过滤条件，那么只会接收到以过滤条件开头的消息 
'''  
import sys  
import zmq  
  
  
#  Socket to talk to server  
context = zmq.Context()  
socket = context.socket(zmq.SUB)  
  
print("Collecting updates from weather server...")  
socket.connect("tcp://127.0.0.1:8000")  
  
# Subscribe to zipcode, default is NYC, 10001  
zip_filter = sys.argv[1] if len(sys.argv) > 1 else "10002"  
  
#此处设置过滤条件，只有以 zip_filter 开头的消息才会被接收  
socket.setsockopt(zmq.SUBSCRIBE, zip_filter)  
  
# Process 5 updates  
total_temp = 0  
for update_nbr in range(5):  
    string = socket.recv()  
    print string  
    zipcode, temperature, relhumidity = string.split()  
    total_temp += int(temperature)  
  
print("Average temperature for zipcode '%s' was %dF" % (  
      zip_filter, total_temp / update_nbr)  
)

总结　　

1、 zmq的程序，也是要分清服务端和客户端的，服务端也是要绑定ip和端口的

2、如果我们先启动客户端，后启动服务端，那么程序是可以正常运行的，换成socket，就不行，socket只能先启动服务端，后启动客户端

3、学习zmq的过程，千万别总想着socket，你能用socket传输文件，但是如果用zmq做同样的事情，那你就错误的使用了zmq，记住，这是一个消息通信库，它自己实现了一些协议，使得我们可以非常轻松的在节点间，进程间，线程间传递消息，如果你对我刚才说的节点间，进程间，线程间传递消息没什么兴趣，说明，你平日里写的程序都是单进程，单线程的，只管顺序执行就好了，其他的不用考虑。

4、广播，这种模式没有队列缓存，断开之后数据将丢失

下面，分析这两段程序。

1、不论是服务端还是客户端，都需要获得zmq上下文

context = zmq.Context()

2、获得socket，这个socket不是我们平日里以为的那个socket。zmq里叫socket，我猜可能是为了方便大家学习才这么命名。它的表现，已经远远的超出了我们对以前的那个socket的了解。每一个socket都是有自己的类型的，示例中，服务端的socket的类型是zmq.PUB，客户端的socket的类型是zmq.SUB，pub是发布，sub是订阅。说的通俗点，就是有一个pub节点，可以有多个sub节点，pub节点发出去的消息，如果sub节点没有设置过滤条件，那么就会接收所有的消息，如果有过滤条件，就只接收满足过滤条件的消息。想想看，有没有那么一个时刻，你希望你的程序等待一个命令，收到命令后，你让程序去做一些事情？那么pub与sub模式非常适合这种应用场景。

3、设置过滤条件很简单　

socket.setsockopt(zmq.SUBSCRIBE, zip_filter)

第二个参数就是你期望的过滤条件，只有那些以这个过滤条件开头的消息才会被接收　　

问答环节

问题1：如果想创建多个socket怎么写？

答：一个上下文可以创建任意多个socket，完全不受限制

问题2：明明先启动了客户端，后启动的服务端，为啥有些消息却没有收到呢？

答：就算你先启动了客户端，服务端pub出去的一些消息也还是可能没有被收到，因为你启动服务端时，服务端与客户端要建立连接，而这个时候，消息其实已经发出去了，所以你没收到。

问题3：在订阅发布模型中，如果客户端断开连接，或是服务端断开连接会产生什么样的影响

答：如果是客户端断开连接，没什么的，就好比一堆人在听收音机，现在离开一个人，收音机继续播放喽。如果是服务端断开了呢，比如程序死掉了，那么请放心，客户端不会发生崩溃，只是阻塞在socket.recv() 这条语句上，更神奇的是，如果你恢复了服务端

现在，我们修改一下客户端程序

#coding=utf-8  
''''' 
订阅模式，如果设置了过滤条件，那么只会接收到以过滤条件开头的消息 
'''  
import sys  
import zmq  
import time  
  
#  Socket to talk to server  
context = zmq.Context()  
socket = context.socket(zmq.SUB)  
  
print("Collecting updates from weather server...")  
socket.connect("tcp://localhost:8000")  
  
# Subscribe to zipcode, default is NYC, 10001  
zip_filter = sys.argv[1] if len(sys.argv) > 1 else "10002"  
  
#此处设置过滤条件，只有以 zip_filter 开头的消息才会被接收  
socket.setsockopt(zmq.SUBSCRIBE, zip_filter)  
  
# Process 5 updates  
total_temp = 0  
for update_nbr in range(50):  
    print 'wait recv'  
    string = socket.recv()  
    print 'has recv'  
    time.sleep(1)  
    print string  
    zipcode, temperature, relhumidity = string.split()  
    total_temp += int(temperature)  
  
print("Average temperature for zipcode '%s' was %dF" % (  
      zip_filter, total_temp / update_nbr)  
)

服务端

#coding=utf-8  
''''' 
服务端，发布模式 
'''  
import zmq  
import time  
from random import randrange  
  
context = zmq.Context()  
socket = context.socket(zmq.PUB)  
socket.bind("tcp://*:8000")  
  
while True:  
    zipcode = randrange(1, 100000)  
    temperature = randrange(-80, 135)  
    relhumidity = randrange(10, 60)  
  
    socket.send("%i %i %i" % (10002,temperature , relhumidity))

　　服务端和客户端都启动，这时候，客户端收到一条消息后会睡一秒钟，但是服务端却是一刻不停的在发送消息，那么问题来了，一个发的快，一个收的慢，那么这时候把服务端停掉会怎样呢？

　　实际的效果是，服务端停下来了，客户端依然在接收消息，因为有一些消息被缓存起来了，虽然服务端不再发送了，客户端却依然可以接收得到，但这种接收，只是从之前接收的缓冲区里取数据。

　　现在，我们在服务端最后加上一条语句，time.sleep(2)，这样，服务端发送一条消息后，睡两秒钟，发的慢，收的快了，我们再次启动服务端和客户端，当客户端收到一些消息后，关掉服务端，这次，客户端很快就停止接收了，因为发的慢，所以缓冲区里没有数据，现在，我们再次启动服务端，你会发现，客户端又开始接收数据了，哈哈，神奇吧！