FastAPI 的 async/await

什么是 async/await

在介绍 FastAPI 中 async/await 的使用方法之前，我们先来了解一下 async/await 是什么。

async/await 是 Python 3.5 引入的一种异步编程的语法糖。通过使用 async 关键字定义异步函数，并在需要异步调用的地方使用 await 关键字，可以方便地编写高效的异步程序。

以前的 Python 版本中，我们需要使用回调函数或者使用线程来实现异步操作，而使用 async/await 可以更加简洁和直观地实现异步操作。

FastAPI 中的 async/await

FastAPI 是一个基于 Starlette 框架的现代化高性能 Web 框架，它可以与 async/await 完美地结合起来，提供高效的异步编程体验。

在 FastAPI 中，我们可以将路由处理函数定义为异步函数，并在其中使用 async/await 来实现异步操作。

以下是一个简单的示例，展示了如何在 FastAPI 中使用 async/await 来处理一个异步请求：

import asyncio
from fastapi import FastAPI

app = FastAPI()


@app.get("/")
async def root():
    result = await fetch_data_from_database()
    return {"data": result}

async def fetch_data_from_database():
    # 假设这是一个耗时的数据库操作
    # 使用 sleep 模拟耗时操作
    await asyncio.sleep(1)
    return "data from database"

在上述示例中，fetch_data_from_database 函数模拟了一个耗时的数据库操作，通过使用 await asyncio.sleep(1) 来模拟耗时。

在 root 路由处理函数中，我们使用 await fetch_data_from_database() 来等待异步操作的结果，并将结果作为 JSON 数据返回给客户端。

异常处理

在使用 async/await 进行异步编程时，我们也需要考虑异常处理的情况。

在 FastAPI 中，我们可以使用 try/except 块来捕获异步函数中的异常，并进行相应的处理。

以下是一个示例，展示了如何在 FastAPI 中处理异步函数中的异常：

import asyncio
from fastapi import FastAPI

app = FastAPI()


@app.get("/")
async def root():
    try:
        result = await fetch_data_from_database()
        return {"data": result}
    except Exception as e:
        return {"error": str(e)}

async def fetch_data_from_database():
    # 假设这是一个可能出现异常的耗时数据库操作
    if error_occurs:
        raise Exception("Error occurred")
    await asyncio.sleep(1)
    return "data from database"