Spaces:
Sleeping
Sleeping
Update controllers/ws_controller.py
Browse files
controllers/ws_controller.py
CHANGED
@@ -55,7 +55,7 @@ def get_messages(code: str,websocket:WebSocket,brand_name: Optional[str] = None)
|
|
55 |
print(gmail_data)
|
56 |
|
57 |
if "messages" in gmail_data:
|
58 |
-
with ThreadPoolExecutor(max_workers=
|
59 |
for message_data in gmail_data["messages"]:
|
60 |
|
61 |
executor.submit(fetch_message_wrapper, message_data , websocket)
|
@@ -77,8 +77,10 @@ def get_messages(code: str,websocket:WebSocket,brand_name: Optional[str] = None)
|
|
77 |
return messages
|
78 |
|
79 |
async def process_message(message:Message, websocket:WebSocket, chunk_size:int):
|
|
|
80 |
if message:
|
81 |
message_json = message.to_json()
|
|
|
82 |
await send_message_in_chunks(websocket, message_json, chunk_size)
|
83 |
await websocket.send_text("NEXT_MESSAGE")
|
84 |
|
|
|
55 |
print(gmail_data)
|
56 |
|
57 |
if "messages" in gmail_data:
|
58 |
+
with ThreadPoolExecutor(max_workers=5) as executor:
|
59 |
for message_data in gmail_data["messages"]:
|
60 |
|
61 |
executor.submit(fetch_message_wrapper, message_data , websocket)
|
|
|
77 |
return messages
|
78 |
|
79 |
async def process_message(message:Message, websocket:WebSocket, chunk_size:int):
|
80 |
+
logging.info("process_message")
|
81 |
if message:
|
82 |
message_json = message.to_json()
|
83 |
+
logging.info(f"{message_json}")
|
84 |
await send_message_in_chunks(websocket, message_json, chunk_size)
|
85 |
await websocket.send_text("NEXT_MESSAGE")
|
86 |
|