【测开能力提升-fastapi框架】fastapi能力提升 - 中间件与CORS

1. 中间件

1.1 介绍（ChatGPT抄的，大致可以理解）

一种机制，用于在处理请求和响应之前对其进行拦截、处理或修改。中间件可以在应用程序的请求处理管道中插入自定义逻辑，以实现一些通用的功能，如身份验证、日志记录、错误处理等。

中间件在请求到达应用程序之前被执行，并在应用程序生成响应之前被执行。它们可以对请求进行修改、记录请求信息、添加额外的响应头，甚至可以拒绝请求或提前返回响应。

在FastAPI中，可以通过编写中间件函数来定义中间件逻辑。这些函数接收一个Request对象和一个Callable对象（代表下一个中间件或请求处理器）作为参数，并返回一个Response对象。

注释：
我理解就是在中间件里写一个函数，所有路由函数在执行前都会执行此函数

1.2 示例

import uvicorn
from fastapi import FastAPI, Request

app = FastAPI()

# 中间件
@app.middleware("http")
async def m2(request:Request, call_next):
    # 请求代码块
    print("m2 request")
    response = await call_next(request)
    # 响应代码块
    print("m2 response")

    return response


@app.middleware("http")
async def m1(request: Request, call_next):
    # 请求代码块
    print("m1 request")
    response = await call_next(request)
    # 响应代码块
    print("m1 response")

    return response


@app.get("/user")
async def get_user():
    print("get_user函数执行")
    return {"user":"current user"}

@app.get("/item/{item_id}")
async def get_item(item_id:int):
    print("get_item函数执行")
    return {"item_id":item_id}

if __name__ == "__main__":
    uvicorn.run("main:app", port=5050, reload=True)

注释：
执行顺序为：request先下后上，response先上后下，先执行request，再执行response

1.3 示例

import time

import uvicorn
from fastapi import FastAPI, Request

from fastapi.responses import Response

app = FastAPI()

# 中间件
@app.middleware("http")
async def m2(request:Request, call_next):
    # 请求代码块
    print("m2 request")
    response = await call_next(request)
    # 响应代码块
    print("m2 response")

    return response


@app.middleware("http")
async def m1(request: Request, call_next):
    # 请求代码块
    print("m1 request")
    # 示例：访问url限制
    if request.client.host in ["127.0.0.1",]:
        return Response(status_code=403, content="visit forbidden")
    # 示例：访问路径限制
    if request.url.path in ["/user"]:
        return Response(status_code=403, content="visit forbidden")

    response = await call_next(request)
    # 响应代码块
    print("m1 response")
    return response


@app.get("/user")
async def get_user():
    print("get_user函数执行")
    return {"user":"current user"}

@app.get("/item/{item_id}")
async def get_item(item_id:int):
    print("get_item函数执行")
    return {"item_id":item_id}

if __name__ == "__main__":
    uvicorn.run("main:app", host="127.0.0.1", port=5050, reload=True)

1.4 response响应示例

import time

import uvicorn
from fastapi import FastAPI, Request

from fastapi.responses import Response

app = FastAPI()

# 中间件
@app.middleware("http")
async def m2(request:Request, call_next):
    # 请求代码块
    print("m2 request")
    response = await call_next(request)
    # 响应代码块
    print("m2 response")

    return response


@app.middleware("http")
async def m1(request: Request, call_next):
    # 请求代码块
    print("m1 request")
    # 示例：访问url限制
    # if request.client.host in ["127.0.0.1",]:
    #     return Response(status_code=403, content="visit forbidden")
    # # 示例：访问路径限制
    # if request.url.path in ["/user"]:
    #     return Response(status_code=403, content="visit forbidden")

    start = time.time()
    response = await call_next(request)
    # 响应代码块
    print("m1 response")
    end = time.time()
    response.headers["ProcessTimer"] = str(end - start)
    return response


@app.get("/user")
async def get_user():
    time.sleep(3)
    print("get_user函数执行")
    return {"user":"current user"}

@app.get("/item/{item_id}")
async def get_item(item_id:int):
    time.sleep(2)
    print("get_item函数执行")
    return {"item_id":item_id}

if __name__ == "__main__":
    uvicorn.run("main:app", host="127.0.0.1", port=5050, reload=True)

2. CORS跨域请求

import uvicorn
from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_methods=['GET', 'POST'],
    allow_headers=['*']
)

# @app.middleware("http")
# async def HelloMiddleware(request:Request, call_next):
#     response = await call_next(request)
#     response.headers["Access-Control-Allow-Origin"] = "*"
#     return response

@app.get("/hello")
async def get_hello():
    return {"message": "hello xujie"}

if __name__ == "__main__":
    uvicorn.run("main:app", port=5050, reload=True)

注释：
fastapi框架（测试开发用）差不多就这些了，请求、响应、assert、数据库、中间件、跨域、template、基本上合作开发一套简单自用测试平台已经够用（至少我是这样的），填坑完成，撒花