FilesExpand file tree

 jon

/

sse.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

33 lines (26 loc) · 980 Bytes

 jon

/

sse.py

Top

File metadata and controls

• Code
• Blame

33 lines (26 loc) · 980 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

import asyncio

import asyncio_redis

import json

import websockets

from datetime import datetime

clients = set()

@asyncio.coroutine

def publisher(websocket, path):

global clients

clients.add(websocket)

redis_con = yield from asyncio_redis.Connection.create()

sub = yield from redis_con.start_subscribe()

yield from sub.subscribe(['live'])

while True:

try:

data = yield from asyncio.wait_for(sub.next_published(), 5)

data = json.loads(data.value)

yield from asyncio.wait([ws.send(json.dumps(data)) for ws in clients])

except asyncio.TimeoutError:

out = json.dumps({'type': 'keepalive', 'ts': datetime.now().timestamp()})

yield from asyncio.wait([ws.send(out) for ws in clients])

redis_con.close()

ws_server = websockets.serve(publisher, 'localhost', 6000)

loop = asyncio.get_event_loop()

loop.run_until_complete(ws_server)

asyncio.get_event_loop().run_forever()