FastAPI 中间件(二) 高级中间件

作者:麦克煎蛋   出处:https://www.cnblogs.com/mazhiyong/ 转载请保留这段声明,谢谢!

 

除了自定义中间件,我们也可以直接使用其他的中间件。

FastAPI通过 app.add_middleware() 操作来引入已定义的中间件。

from fastapi import FastAPI
from unicorn import UnicornMiddleware

app = FastAPI()

app.add_middleware(UnicornMiddleware, some_config="rainbow")

app.add_middleware()第一个参数是中间件类,同时还支持中间件的其他附加信息。

 

FastAPI提供了一些常用的中间件。

一、HTTPSRedirectMiddleware

HTTPSRedirectMiddleware强制发来的请求协议必须是https或者wss。

from fastapi import FastAPI
from fastapi.middleware.httpsredirect import HTTPSRedirectMiddleware

app = FastAPI()

app.add_middleware(HTTPSRedirectMiddleware)


@app.get("/")
async def main():
    return {"message": "Hello World"}

二、TrustedHostMiddleware

TrustedHostMiddleware强制发来的请求必须在Header信息中设置了Host选项,为了避免HTTP Host Header攻击。

from fastapi import FastAPI
from fastapi.middleware.trustedhost import TrustedHostMiddleware

app = FastAPI()

app.add_middleware(
    TrustedHostMiddleware, allowed_hosts=["example.com", "*.example.com"]
)


@app.get("/")
async def main():
    return {"message": "Hello World"}

三、GZipMiddleware

当请求的头信息 Accept-Encoding 字段带有"gzip"时,GZipMiddleware负责完成相应的返回结果处理。

GZipMiddleware 支持 minimum_size 参数:当返回结果大小小于指定值时不启用压缩。(单位为字节,默认值为500)

from fastapi import FastAPI
from fastapi.middleware.gzip import GZipMiddleware

app = FastAPI()

app.add_middleware(GZipMiddleware, minimum_size=1000)


@app.get("/")
async def main():
    return "somebigcontent"

 

更多中间件信息可参考文档:

Starlette‘s Middleware docs 

ASGI Awesome List

FastAPI 中间件(二) 高级中间件

上一篇:C# 使用Dev控件为gridView添加复选框,实现多选和单选


下一篇:洛谷P5443 【[APIO2019]桥梁】