-
Notifications
You must be signed in to change notification settings - Fork 0
/
asyncio_publisher.py
69 lines (53 loc) · 1.66 KB
/
asyncio_publisher.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
61
62
63
64
65
66
67
68
import struct
import random
import asyncio
clients = {}
@asyncio.coroutine
def publish_stuff():
idx = 0
while True:
payload_size = random.randint(4, 10000)
fullmsg = bytearray(4 + payload_size)
struct.pack_into('>I', fullmsg, 0, payload_size) # 4-byte header
struct.pack_into('I', fullmsg, 4, idx)
remove_list = []
for connid in clients.keys():
reader, writer = clients[connid]
do_close = False
hiccup = random.randint(1, 20)==1 # 1 in 20 chance
msg = memoryview(fullmsg)
if hiccup:
# simulate send of partial message before disconnection
partial_size = random.randint(1, len(msg))
msg = msg[:partial_size]
do_close = True
writer.write(msg)
if do_close:
writer.close()
remove_list.append(connid)
for connid in remove_list:
del clients[connid]
idx += 1
yield from asyncio.sleep(0.2)
@asyncio.coroutine
def client_connected(reader, writer):
sock = writer.get_extra_info('socket')
connid = sock.fileno()
clients[connid] = (reader, writer)
while True:
data = yield from reader.read(1)
if not data:
break
clients.pop(connid, None)
writer.close()
loop = asyncio.get_event_loop()
coro = asyncio.start_server(client_connected, '127.0.0.1', 10000)
server = loop.run_until_complete(coro)
asyncio.async(publish_stuff())
try:
loop.run_forever()
except KeyboardInterrupt:
pass
server.close()
loop.run_until_complete(server.wait_closed())
loop.close()