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'):
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))