了解saltstack的通信协议zeromq(二)
import sysimport time
import zmq
context = zmq.Context()
# Socket to receive messages on
receiver = context.socket(zmq.PULL)
receiver.bind("tcp://*:5558")
# Wait for start of batch
s = receiver.recv()
sum = 0
flag = 0
# Start our clock now
tstart = time.time()
while True:
a_str = receiver.recv_string()
num = int(a_str)
sum += num
if a_str == '0':
flag += 1
if flag == 10:
break
tend = time.time()
tdiff = tend - tstart
total_msec = tdiff * 1000
print "Total elapsed time: %d msec" % total_msec
页:
[1]