1

fastapi-demo/controller/base_controller.py

@@ -0,0 +1,10 @@
+# -*- coding:utf-8 -*-
+from fastapi import APIRouter
+from fastapi.responses import JSONResponse
+
+router = APIRouter()
+
+
+@router.get("/check", name="服务状态校验", tags=['BASE'])
+async def listener():
+    return JSONResponse({"message": "success", "statusCode": 200}, status_code=200)

fastapi-demo/debug.py

@@ -0,0 +1,5 @@
+# -*- coding:utf-8 -*-
+import uvicorn
+
+if __name__ == '__main__':
+    uvicorn.run(app='api_main:app', host="0.0.0.0", port=8087, reload=True, debug=True)

fastapi-demo/main.py

@@ -1,7 +1,100 @@
+import importlib
+import json
+import os
+import time
+
 from fastapi import FastAPI
+from fastapi import HTTPException, Request
+from fastapi.exceptions import RequestValidationError
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import PlainTextResponse
 from pydantic import BaseModel  # 参数校验
+from starlette.exceptions import HTTPException as StarletteHTTPException
+from starlette.middleware.sessions import SessionMiddleware
+from starlette.responses import JSONResponse
+from starlette.staticfiles import StaticFiles
+
+from config import logger, CfgBaseInit
+
+app = FastAPI(title="EIP REC ENGINE ", description="通用系统 ", version="v 0.0.0")
+
+app.routes.pop(1)  # drop default docs_url router
+
+# 添加 session 中间键，使项目中可以使用session
+app.add_middleware(SessionMiddleware, secret_key='123456')
+
+# 跨域白名单, 解决跨域问题
+origins = [
+    "*",
+]
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+static_obj = StaticFiles(directory="static", html=True)
+app.mount("/static", static_obj, "static")
+
+
+@app.exception_handler(StarletteHTTPException)
+async def http_exception_handler(request, exc):
+    """接口服务异常捕获; text/plain"""
+    return PlainTextResponse(str(exc.detail), status_code=exc.status_code)
+
+
+@app.exception_handler(RequestValidationError)
+async def validation_exception_handler(request, exc):
+    """请求校验异常捕获; application/json """
+    json_response = JSONResponse({'message': '【 {} 】接口触发了 RequestValidationError 错误.'
+                                 .format(request.scope.get("endpoint").__name__, exc.errors()), 'status_code': 999,
+                                  "error_message": exc.errors(), "queryParasm": exc.body})
+    logger.error(json.loads(json_response.body))
+    return json_response
+
+
+@app.exception_handler(Exception)
+async def validation_exception_handler(request, exc):
+    """请求校验异常捕获; application/json """
+    return JSONResponse({'message': "服务器内部错误", 'status_code': 500})
+
+
+@app.middleware("http")
+async def add_process_time_header(request: Request, call_next):
+    """接口响应中间键; 当前只支持 http 请求"""
+    logger.info(f'【 {request.scope.get("path").replace("/", ".").lstrip(".")} 】API query start...')
+    start_time = time.time()
+    response = await call_next(request)
+    process_time = time.time() - start_time
+    response.headers["API-Process-Time"] = f"{process_time} seconds"
+    response.headers["service"] = "EIP REC ENGINE"
+    response.headers["Request-Client"] = str(request.scope.get("client")[0]) + ":" + str(
+        request.scope.get("client")[1])
+    response.headers["User-Agent"] = str(request.headers.get("user-agent"))
+    logger.info(f'Request-Client is :【 {response.headers.get("Request-Client")} 】')
+    logger.info(f'Request-Client User-Agent is :【 {response.headers.get("User-Agent")} 】')
+    request.scope.get("path").replace("/", "")
+    logger.info(f'【 {request.scope.get("path")} 】API '
+                f'Process-Time:【 {process_time} seconds 】...')
+
+    return response
+
 
-app = FastAPI()
+# 动态注册路由
+for project_path in os.path.join(CfgBaseInit.project_path, "controller"):
+    files = [file.split(".")[0] for file in files if file.endswith(".py")]
+    # 获取每一个文件的 module。
+    for file in files:
+        router_module = importlib.import_module(f"controller.{file}")
+        if hasattr(router_module, "router"):
+            # 注册路由
+            app.include_router(
+                router_module.router,
+                # dependencies=[Depends(get_token_header)],
+                responses={404: {"description": "Not found"}},
+            )
 
 
 @app.get("/")
@@ -34,4 +127,4 @@ class People(BaseModel):
 @app.post("/insert")
 def insert(people: People):
     msg = f"名字：{people.name},年龄{people.age}"
-    return msg
+    return msg

tmp/tmp.py

@@ -0,0 +1,15 @@
+import pickle
+import sys
+
+from cuckoopy import CuckooFilter
+from pybloom_live import BloomFilter
+
+b = BloomFilter(capacity=100000, error_rate=0.0001)
+print(sys.getsizeof(pickle.dumps(b).decode("latin1")))
+print(sys.getsizeof(b))
+
+ck = CuckooFilter(capacity=100000, bucket_size=4, fingerprint_size=1)
+print(sys.getsizeof(pickle.dumps(ck).decode("latin1")))
+print(sys.getsizeof(ck))
+
+print(sys.getsizeof(bytes[10000]))