-
Notifications
You must be signed in to change notification settings - Fork 0
/
demo.py
61 lines (43 loc) · 1.25 KB
/
demo.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#!/usr/bin/python
import os
import sys
import argparse
import json
import time
import urlparse
import zmq
from zmq.eventloop.ioloop import PeriodicCallback
from notery.base import ZMQProcess, MessageHandler
class DemoHandler (MessageHandler):
def handle_ping (self, data):
print 'RECEIVE PING', data
class DemoApp(ZMQProcess):
def __init__ (self):
super(DemoApp, self).__init__()
def setup(self):
super(DemoApp, self).setup()
self.pub, self.pub_addr = self.stream(zmq.PUB, 'tcp://127.0.0.1:%(port)s', True)
self.sub, sub_addr = self.stream(zmq.SUB, self.pub_addr, False,
callback=DemoHandler())
self.heartbeat = PeriodicCallback(self.ping, 1000, self.loop)
def ping(self):
print 'SEND PING'
self.pub.send_multipart(['ping', json.dumps(['ping', time.time()])])
def local_run(self):
print 'START HEARTBEAT'
self.heartbeat.start()
def stop(self):
self.heartbeat.stop()
self.loop.stop()
def parse_args():
p = argparse.ArgumentParser()
return p.parse_args()
def echo(msg):
print 'ECHO:', msg
def main():
opts = parse_args()
app = DemoApp()
app.start()
app.join()
if __name__ == '__main__':
main()