Omkar008 commited on
Commit
288bacd
1 Parent(s): 1723e0b

Update controllers/ws_controller.py

Browse files
Files changed (1) hide show
  1. controllers/ws_controller.py +3 -1
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=15) as executor:
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