18、在FastAPI中实现WebSocket双向交互通信

在FastAPI中实现WebSocket双向交互通信

1. 处理并发问题

在之前的示例中，我们假设客户端总是先发送消息，服务器在接收到消息后再将其返回。但在实际场景中，服务器可能主动向客户端发送数据。例如在聊天应用里，其他用户发送的消息需要立即转发给第一个用户。而之前示例中的 receive_text 阻塞调用就会产生问题，因为在等待客户端消息时，服务器可能有消息要转发给客户端。

为了解决这个问题，我们可以使用 asyncio 模块的高级工具。该模块提供了函数，允许我们并发调度多个协程，并等待其中一个完成。在我们的场景中，可以有一个协程等待客户端消息，另一个协程在有数据时将其发送给客户端。哪个协程先完成，就开始下一轮循环。

以下是一个示例代码：

import asyncio
from fastapi import FastAPI, WebSocket
import datetime

app = FastAPI()

async def echo_message(websocket: WebSocket):
    data = await websocket.receive_text()
    await websocket.send_text(f"Message text was: {data}")

async def send_time(websocket: WebSocket):
    await asyncio.sleep(10)
    await websocket.send_text(f"It is: {datetim