FastAPI异步编程与await关键字：ws.send_text()和load_dataset()的执行顺序
本文探讨在FastAPI框架中使用async/await进行异步编程时，ws.send_text()和load_dataset()函数的执行顺序问题。 之前的代码示例中，存在一个误解：ws.send_text("1")似乎需要等待load_dataset("beans")完成才能执行。实际上并非如此。
关键在于理解await关键字的作用和load_dataset()函数的特性。 await 仅用于等待异步操作完成。ws.send_text()是一个异步操作，因此await ws.send_text("1")会等待消息发送完成。然而，load_dataset("beans")是一个同步阻塞操作，它会阻塞当前协程直到数据集加载完成。
代码执行流程分析:
-
await ws.accept()： 等待WebSocket连接建立。 -
await ws.send_text("1")： 异步发送消息"1"，此操作完成后继续执行。 -
dataset = load_dataset("beans")： 同步阻塞操作开始，程序在此处暂停，直到load_dataset("beans")从远程下载并加载数据集完成。 -
await ws.send_text("2")： 异步发送消息"2"，同样需要等待发送完成。
实验验证与结果解读:
实验结果清晰地表明，浏览器端先接收到"1"，然后才是"2"， 这与load_dataset("beans")的阻塞特性相符。 虽然ws.send_text("1")先执行，但load_dataset("beans")的阻塞导致"2"的发送被延迟到数据集加载完成之后。
改进代码以实现并发:
如果需要load_dataset("beans")与ws.send_text("1")并发执行，需要将load_dataset("beans")改造成异步操作。这通常需要使用异步IO库，例如aiohttp来下载数据。 以下是一个改进后的示例（假设已使用aiohttp并实现了一个异步的load_dataset_async函数）：
import asyncio
from datetime import datetime
from datasets import load_dataset
from fastapi import FastAPI, WebSocket
from fastapi.responses import HTMLResponse
app = FastAPI()
# ... (HTML code remains the same) ...
@app.websocket("/ws")
async def h(ws: WebSocket):
await ws.accept()
task = asyncio.create_task(load_dataset_async("beans")) # Asynchronously load dataset
await ws.send_text(f"1: {datetime.now()}")
dataset = await task # Await the dataset loading task
print(f"time: {datetime.now()} => dataset: {dataset}")
await ws.send_text(f"2: {datetime.now()}")
# ... (rest of the code remains the same) ...
通过asyncio.create_task()创建异步任务，load_dataset_async("beans")可以在后台并发执行，而不会阻塞主协程。
总结： 原代码中ws.send_text("1")先执行，但load_dataset("beans")的同步阻塞特性决定了后续操作的执行顺序。 要实现并发，必须将数据加载操作异步化。
