fix websocket disconnection error

api/main.py

@@ -1,6 +1,7 @@
-from fastapi import Depends, FastAPI, WebSocket
+from fastapi import Depends, FastAPI, WebSocket, WebSocketDisconnect
 from fastapi.middleware.cors import CORSMiddleware
 from datetime import datetime
+from fastapi.websockets import WebSocketState
 from fastapi_sqlalchemy import DBSessionMiddleware, db
 from BDD.models import Coordonnee
 from BDD.schema import Coordonnee  as Model_coordonnee
@@ -44,13 +45,21 @@ async def add_coordonnees(coor: Model_coordonnee, connections= Depends(get_conne
     db_coor = Coordonnee(latitude = coor.latitude, longitude = coor.longitude, ip = coor.ip, date = datetime.strptime(coor.date, date_format))
     db.session.add(db_coor)
     db.session.commit()
-    await connections[0].send_json({'latitude': coor.latitude, 'longitude':coor.longitude, 'ip': coor.ip, 'date': coor.date})
+
+    for i in range(len(connections)):
+        if connections[i].client_state == WebSocketState.DISCONNECTED :
+            connections.remove(connections[i])
+        else :
+            await connections[i].send_json({'latitude': coor.latitude, 'longitude':coor.longitude, 'ip': coor.ip, 'date': coor.date})
     return True
 
 @app.websocket('/ws')
 async def websocket_endpoint(websocket: WebSocket, connections = Depends(get_connections)):
     await websocket.accept()
     connections.append(websocket)
-    while True:
-        data = await websocket.receive_json()
-        await websocket.send_json(data)
+    try:
+        while True:
+            data = await websocket.receive_json()
+            await websocket.send_json(data)
+    except WebSocketDisconnect :
+        connections.remove(websocket)

producer/producer.py

@@ -56,7 +56,7 @@ def produce_messages(bootstrap_servers, topic, num_messages):
         partition = get_machine_partition()
         # partition = 0
         producer.produce(topic, value=message, partition=partition, callback=delivery_report)
-        time.sleep(0.5)
+        time.sleep(5)
 
     producer.flush()