mirror of
https://github.com/trycua/computer.git
synced 2026-01-04 12:30:08 -06:00
Use to_thread instead
This commit is contained in:
@@ -238,8 +238,7 @@ async def websocket_endpoint(websocket: WebSocket):
|
||||
result = await handler_func(**filtered_params)
|
||||
else:
|
||||
# Run sync functions in thread pool to avoid blocking event loop
|
||||
loop = asyncio.get_event_loop()
|
||||
result = await loop.run_in_executor(None, lambda: handler_func(**filtered_params))
|
||||
result = await asyncio.to_thread(handler_func, **filtered_params)
|
||||
await websocket.send_json({"success": True, **result})
|
||||
except Exception as cmd_error:
|
||||
logger.error(f"Error executing command {command}: {str(cmd_error)}")
|
||||
@@ -367,8 +366,7 @@ async def cmd_endpoint(
|
||||
result = await handler_func(**filtered_params)
|
||||
else:
|
||||
# Run sync functions in thread pool to avoid blocking event loop
|
||||
loop = asyncio.get_event_loop()
|
||||
result = await loop.run_in_executor(None, lambda: handler_func(**filtered_params))
|
||||
result = await asyncio.to_thread(handler_func, **filtered_params)
|
||||
|
||||
# Stream the successful result
|
||||
response_data = {"success": True, **result}
|
||||
|
||||
Reference in New Issue
Block a user