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

kingforce

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 ,>　　

　　

　　客户端代码：
　　

#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,>　　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,>　　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 ,>　　

　　

　　服务端和客户端都启动，这时候，客户端收到一条消息后会睡一秒钟，但是服务端却是一刻不停的在发送消息，那么问题来了，一个发的快，一个收的慢，那么这时候把服务端停掉会怎样呢？
　　实际的效果是，服务端停下来了，客户端依然在接收消息，因为有一些消息被缓存起来了，虽然服务端不再发送了，客户端却依然可以接收得到，但这种接收，只是从之前接收的缓冲区里取数据。
　　现在，我们在服务端最后加上一条语句，time.sleep(2)，这样，服务端发送一条消息后，睡两秒钟，发的慢，收的快了，我们再次启动服务端和客户端，当客户端收到一些消息后，关掉服务端，这次，客户端很快就停止接收了，因为发的慢，所以缓冲区里没有数据，现在，我们再次启动服务端，你会发现，客户端又开始接收数据了，哈哈，神奇吧！