wangdongxu

jira:NYJ-1460 desc:init

import json
import time
import logging
logger = logging.getLogger(__name__)
from fastapi import FastAPI, Request
from fastapi import Request
from starlette.middleware.base import BaseHTTPMiddleware
from starlette.responses import StreamingResponse
from config import settings
class LoggingMiddleware(BaseHTTPMiddleware):
"""HTTP请求/响应日志中间件."""
# 日志记录路径前缀
async def dispatch(self, request: Request, call_next):
# 记录请求开始时间
start_time = time.time()
# 获取请求信息
method = request.method
url = str(request.url)
client_host = request.client.host if request.client else "unknown"
# 判断是否需要记录日志
should_log = request.url.path.startswith(settings.url_prefix)
if should_log:
# 记录请求开始时间
start_time = time.time()
# 读取请求体
request_body = None
if method in ("POST", "PUT", "PATCH"):
... ... @@ -77,3 +84,6 @@ class LoggingMiddleware(BaseHTTPMiddleware):
logger.info(f" Response Body: {response_body}")
return response
else:
# 不需要记录日志的请求,直接处理
return await call_next(request)
... ...
"""API路由聚合模块."""
from fastapi import APIRouter
from config import settings
api_router = APIRouter(prefix=settings.url_prefix)
# 创建主路由
api_router = APIRouter(prefix="/aigc-embedding/api")
def register_router(router: APIRouter, prefix: str):
api_router.include_router(router, prefix=prefix)
... ...
... ... @@ -2,12 +2,10 @@ app_name: "Embedding Service"
app_version: "1.0.0"
port: 8000
debug: false
url_prefix: "/aigc-embedding/api"
db_es_enable: true
db_es_url: "http://localhost:9200"
db_postgres_enable: false
db_postgres_url: "postgresql://postgres:postgres@localhost:5432/postgres"
service:
face:
db_client: "es"
#db_postgres_enable: false
#db_postgres_url: "postgresql://postgres:postgres@localhost:5432/postgres"
\ No newline at end of file
... ...