为什么在FastAPI中，ws.send_text("1")会先于load_dataset("beans")执行？

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")是一个同步阻塞操作，它会阻塞当前协程直到数据集加载完成。

代码执行流程分析:

1. await ws.accept()： 等待WebSocket连接建立。
2. await ws.send_text("1")： 异步发送消息"1"，此操作完成后继续执行。
3. dataset = load_dataset("beans")： 同步阻塞操作开始，程序在此处暂停，直到load_dataset("beans")从远程下载并加载数据集完成。
4. 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")的同步阻塞特性决定了后续操作的执行顺序。 要实现并发，必须将数据加载操作异步化。

以上就是为什么在FastAPI中，ws.send_text("1")会先于load_dataset("beans")执行？的详细内容，更多请关注php中文网其它相关文章！