mirror of
https://github.com/Rhinemann/IoT-Systems.git
synced 2026-03-14 20:50:39 +02:00
add initial server->client update with all current DB data
This commit is contained in:
parent
1c856dca0e
commit
da9fe69d4e
@ -40,10 +40,24 @@ subscriptions: Dict[int, Set[WebSocket]] = {}
|
|||||||
@app.websocket("/ws/{user_id}")
|
@app.websocket("/ws/{user_id}")
|
||||||
async def websocket_endpoint(websocket: WebSocket, user_id: int):
|
async def websocket_endpoint(websocket: WebSocket, user_id: int):
|
||||||
await websocket.accept()
|
await websocket.accept()
|
||||||
|
|
||||||
if user_id not in subscriptions:
|
if user_id not in subscriptions:
|
||||||
subscriptions[user_id] = set()
|
subscriptions[user_id] = set()
|
||||||
|
|
||||||
subscriptions[user_id].add(websocket)
|
subscriptions[user_id].add(websocket)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
# send already available data
|
||||||
|
r = processed_agent_data.select()
|
||||||
|
stored_data = SessionLocal().execute(r).fetchall()
|
||||||
|
|
||||||
|
jsonable_data = [{c.name: getattr(i, c.name) for c in processed_agent_data.columns} for i in stored_data]
|
||||||
|
for i in jsonable_data:
|
||||||
|
i['timestamp'] = i['timestamp'].strftime("%Y-%m-%dT%H:%M:%SZ")
|
||||||
|
|
||||||
|
await websocket.send_json(json.dumps(jsonable_data))
|
||||||
|
|
||||||
|
# receive forever
|
||||||
while True:
|
while True:
|
||||||
await websocket.receive_text()
|
await websocket.receive_text()
|
||||||
except WebSocketDisconnect:
|
except WebSocketDisconnect:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user