我有一个同步迭代器,它来自第三方包。迭代器查询外部服务并产生一些数据。如果没有数据,则迭代器等待它。我将WebSocketEndpoint
Starlette 的子类化为通过 websocket 从迭代器发送新数据。不幸的是,我似乎不明白某些东西,而且我的代码没有按预期工作。这是一个稍微简化的代码:
import time
from starlette.endpoints import WebSocketEndpoint
from starlette.websockets import WebSocket
class Iterator:
"""This is a third-party object, not asynchronous at all."""
def __init__(self):
self._stop = False
def __iter__(self):
self.i = 0
return self
def __next__(self):
if self._stop:
raise StopIteration
time.sleep(5)
self.i += 1
print(self.i)
return self.i
def cancel(self):
self._stop = True
class MyWebSocket(WebSocketEndpoint):
def __init__(self, scope, receive, send) -> None:
super().__init__(scope, receive, send)
self.iterator = Iterator()
async def on_connect(self, websocket: WebSocket) -> None:
await super().on_connect(websocket)
for message in self.iterator:
await websocket.send_json({"message": message})
async def on_disconnect(self, websocket: WebSocket, close_code: int) -> None:
await super().on_disconnect(websocket, close_code)
self.iterator.cancel()
第一个问题 - 代码不通过 websocket 发送任何数据。print 语句表明,迭代器产生数据,但实际上没有发送任何数据。如果我放return
after websocket.send_json()
,它将正确发送迭代器的第一个结果,但循环将在之后完成。为什么?
另一个问题是迭代器完全阻塞了应用程序的执行。我理解它为什么会发生,但由于它是一个 Web 服务,并且迭代器旨在工作,直到客户端与 Websocket 断开连接,它很容易阻塞我的整个应用程序。如果我有 10 个工作人员,那么 10 个 websocket 客户端将阻止应用程序,并且在其中一个 websocket 断开连接之前将无法执行任何操作。我该如何解决?