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 asynciofrom datetime import datetimefrom datasets import load_datasetfrom fastapi import FastAPI, WebSocketfrom fastapi.responses import HTMLResponseapp = 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”)执行？的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1360152.html
微信扫一扫 
支付宝扫一扫 
