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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
# this handler responds to every request with "hello world"
import os
import tnetstring
import zmq
instance_id = 'streamhandler.{}'.format(os.getpid()).encode('utf-8')
ctx = zmq.Context()
in_sock = ctx.socket(zmq.PULL)
in_sock.connect('ipc://client-out')
in_stream_sock = ctx.socket(zmq.ROUTER)
in_stream_sock.identity = instance_id
in_stream_sock.connect('ipc://client-out-stream')
out_sock = ctx.socket(zmq.PUB)
out_sock.connect('ipc://client-in')
poller = zmq.Poller()
poller.register(in_sock, zmq.POLLIN)
poller.register(in_stream_sock, zmq.POLLIN)
while True:
socks = dict(poller.poll(None))
if socks.get(in_sock) == zmq.POLLIN:
m_raw = in_sock.recv()
elif socks.get(in_stream_sock) == zmq.POLLIN:
m_list = in_stream_sock.recv_multipart()
m_raw = m_list[2]
else:
continue
req = tnetstring.loads(m_raw[1:])
print('IN {}'.format(req))
if req.get(b'type'):
# skip all non-data messages
continue
if req.get(b'uri', b'').startswith(b'ws'):
resp = {}
resp[b'from'] = instance_id
resp[b'id'] = req[b'id']
resp[b'seq'] = 0
resp[b'code'] = 101
resp[b'reason'] = b'Switching Protocols'
resp[b'credits'] = 1024
print('OUT {} {}'.format(req[b'from'], resp))
out_sock.send(req[b'from'] + b' T' + tnetstring.dumps(resp))
resp = {}
resp[b'from'] = instance_id
resp[b'id'] = req[b'id']
resp[b'seq'] = 1
resp[b'body'] = b'hello world'
print('OUT {} {}'.format(req[b'from'], resp))
out_sock.send(req[b'from'] + b' T' + tnetstring.dumps(resp))
resp = {}
resp[b'from'] = instance_id
resp[b'id'] = req[b'id']
resp[b'seq'] = 2
resp[b'type'] = b'close'
print('OUT {} {}'.format(req[b'from'], resp))
out_sock.send(req[b'from'] + b' T' + tnetstring.dumps(resp))
else:
resp = {}
resp[b'from'] = instance_id
resp[b'id'] = req[b'id']
resp[b'seq'] = 0
resp[b'code'] = 200
resp[b'reason'] = b'OK'
resp[b'headers'] = [[b'Content-Type', b'text/plain']]
resp[b'more'] = True
resp[b'credits'] = 1024
print('OUT {} {}'.format(req[b'from'], resp))
out_sock.send(req[b'from'] + b' T' + tnetstring.dumps(resp))
resp = {}
resp[b'from'] = instance_id
resp[b'id'] = req[b'id']
resp[b'seq'] = 1
resp[b'body'] = b'hello world\n'
print('OUT {} {}'.format(req[b'from'], resp))
out_sock.send(req[b'from'] + b' T' + tnetstring.dumps(resp))
|
