Я пытаюсь создать веб-сокет, используя старлетку, которая получает сообщение, выполняет вычисления в синхронной функции и возвращает ответ.
@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
await websocket.accept()
while True:
stock = await websocket.receive_text()
stock = stock.upper()
data = sentiment_analysis(stock=stock)
await websocket.send_json({"score": data})
анализ настроений - синхронная функция.
поэтому текст получен, и вычисления выполняются, и если я распечатаю переменную данных (словарь), я вижу, что она вернулась правильно, но когда я пытаюсь отправить ее обратно клиенту, я получаю следующую ошибку:
ERROR: Exception in ASGI application
Traceback (most recent call last):
File "/usr/lib/python3/dist-packages/uvicorn/protocols/websockets/websockets_impl.py", line 153, in run_asgi
result = await self.app(self.scope, self.asgi_receive, self.asgi_send)
File "/usr/lib/python3/dist-packages/uvicorn/middleware/proxy_headers.py", line 45, in __call__
return await self.app(scope, receive, send)
File "/usr/local/lib/python3.8/dist-packages/fastapi/applications.py", line 179, in __call__
await super().__call__(scope, receive, send)
File "/usr/local/lib/python3.8/dist-packages/starlette/applications.py", line 111, in __call__
await self.middleware_stack(scope, receive, send)
File "/usr/local/lib/python3.8/dist-packages/starlette/middleware/errors.py", line 146, in __call__
await self.app(scope, receive, send)
File "/usr/local/lib/python3.8/dist-packages/starlette/exceptions.py", line 58, in __call__
await self.app(scope, receive, send)
File "/usr/local/lib/python3.8/dist-packages/starlette/routing.py", line 566, in __call__
await route.handle(scope, receive, send)
File "/usr/local/lib/python3.8/dist-packages/starlette/routing.py", line 283, in handle
await self.app(scope, receive, send)
File "/usr/local/lib/python3.8/dist-packages/starlette/routing.py", line 57, in app
await func(session)
File "/usr/local/lib/python3.8/dist-packages/fastapi/routing.py", line 228, in app
await dependant.call(**values)
File "./app.py", line 13, in websocket_endpoint
stock = await websocket.receive_text()
File "/usr/local/lib/python3.8/dist-packages/starlette/websockets.py", line 85, in receive_text
self._raise_on_disconnect(message)
File "/usr/local/lib/python3.8/dist-packages/starlette/websockets.py", line 80, in _raise_on_disconnect
raise WebSocketDisconnect(message["code"])
starlette.websockets.WebSocketDisconnect: 1011