Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -63,7 +63,7 @@ async def get_messages(code: str,websocket:WebSocket,start,brand_name: Optional[
        
             
         if "messages" in gmail_data:
-            with ThreadPoolExecutor(max_workers=15) as executor:
+            with ThreadPoolExecutor(max_workers=20) as executor:
 
                             
                     
@@ -76,10 +76,7 @@ async def get_messages(code: str,websocket:WebSocket,start,brand_name: Optional[
                     # print(message)
                     if message:
                         # Process and send the message immediately
-                        end1 = time.time()
-                        print("time 1")
-                        print(end1-start1)
-                        await process_message(message, websocket, 30000)
+                        await process_message(message, websocket, 50000)
                         
                       
                     # if message:
@@ -101,13 +98,12 @@ async def get_messages(code: str,websocket:WebSocket,start,brand_name: Optional[
 
 
 async def process_message(message:Message, websocket:WebSocket, chunk_size:int):
-    logging.info("process_message")
+    # logging.info("process_message")
     # print(message)
     if message:
         message_json = message.to_json()
         # logging.info(f"{message_json}")
-        await send_message_in_chunks(websocket, message_json, 50000)
-        # await websocket.send_text(str(message_json))
+        await send_message_in_chunks(websocket, message_json, chunk_size)
         await websocket.send_text("NEXT_MESSAGE")
 
 
@@ -272,23 +268,7 @@ async def websocket_main(code: str,  websocket: WebSocket,start,brand_name: Opti
     await websocket.close()
 
 
-async def send_message_in_chunks(websocket: WebSocket, message_json: dict, chunk_size: int):
-
-    # if message_json['attachments'] is not None :
-    #     for attch in message_json['attachments']:
-    #         attachment_len = attch['attachment_len']
-            
-
-    # print(body_len)
-    # print(attachment_len)
-    # if attachment_len == 0:
-    #     attachment_len = None
-    # await websocket.send_json({"body_len":body_len ,"attachment_len":attachment_len})        
-    
+async def send_message_in_chunks(websocket: WebSocket, message_json: dict, chunk_size: int):         
     message_str = json.dumps(message_json)
-    # print("Printing message_str")
-    # print(message_str)
-    # logging.info(message_str)
-    # await websocket.send_json({"file_len":len(file)})
     for i in range(0, len(message_str), chunk_size):
         await websocket.send_text(message_str[i:i + chunk_size])