2

我用starlette framework 我在下面写了一条路线:

@app.websocket_route("/wss_back", name="run_in_backend")
async def wss_back(session: WebSocket):
    await session.accept()
    data = await session.receive_json()
    await run_code_in_subprocess(session)
    await session.send_json({"msgType": 1, "data": "good"})
    await session.close(1000)

但我不能确保客户端会话将消息发送到 wss 服务器。

那么,我想设置接收超时的时间吗?我怎样才能做到这一点?

如果await websocket.receive_json()超过 5s,则关闭会话。

提前致谢 :)。

4

1 回答 1

2

https://docs.python.org/3/library/asyncio-task.html#asyncio.wait_for

await asyncio.wait_for(session.receive_json(), timeout=5)
于 2019-09-25T10:51:59.917 回答