初始提交

shierfeng-fastapi-backend/middlewares/trace_middleware/__init__.py

@@ -0,0 +1,18 @@
+from fastapi import FastAPI
+
+from .ctx import TraceCtx
+from .middle import TraceASGIMiddleware
+
+__all__ = ('TraceASGIMiddleware', 'TraceCtx')
+
+__version__ = '0.1.0'
+
+
+def add_trace_middleware(app: FastAPI) -> None:
+    """
+    添加trace中间件
+
+    :param app: FastAPI对象
+    :return:
+    """
+    app.add_middleware(TraceASGIMiddleware)