diff --git a/.deploy/deploy-dev.yaml b/.deploy/deploy-dev.yaml index f1816fd..54064d7 100644 --- a/.deploy/deploy-dev.yaml +++ b/.deploy/deploy-dev.yaml @@ -6,6 +6,7 @@ services: image: mathwave/sprint-repo:configurator networks: - configurator-development + - monitoring environment: MONGO_HOST: "mongo.develop.sprinthub.ru" MONGO_PASSWORD: $MONGO_PASSWORD_DEV @@ -20,3 +21,5 @@ services: networks: configurator-development: external: true + monitoring: + external: true diff --git a/.deploy/deploy-prod.yaml b/.deploy/deploy-prod.yaml index 84153e5..dcfffc0 100644 --- a/.deploy/deploy-prod.yaml +++ b/.deploy/deploy-prod.yaml @@ -6,6 +6,7 @@ services: image: mathwave/sprint-repo:configurator networks: - configurator + - monitoring environment: MONGO_HOST: "mongo.sprinthub.ru" MONGO_PASSWORD: $MONGO_PASSWORD_PROD @@ -20,3 +21,5 @@ services: networks: configurator: external: true + monitoring: + external: true diff --git a/app/middlewares/__init__.py b/app/middlewares/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/app/middlewares/metrics.py b/app/middlewares/metrics.py new file mode 100644 index 0000000..b1b1b5d --- /dev/null +++ b/app/middlewares/metrics.py @@ -0,0 +1,14 @@ +import datetime +import zoneinfo +from fastapi import Request, Response +from starlette.middleware.base import BaseHTTPMiddleware + +from app.utils.monitoring import monitoring + +class MetricsMiddleware(BaseHTTPMiddleware): + async def dispatch(self, request: Request, call_next): + start = datetime.datetime.now(zoneinfo.ZoneInfo("Europe/Moscow")) + response: Response = await call_next(request) + end = datetime.datetime.now(zoneinfo.ZoneInfo("Europe/Moscow")) + monitoring.send_metric(start, end, request.url.path, response.status_code, request.method) + return response \ No newline at end of file diff --git a/app/routers/experiments.py b/app/routers/experiments.py index a531114..4dcd35b 100644 --- a/app/routers/experiments.py +++ b/app/routers/experiments.py @@ -1,3 +1,4 @@ +import bson import fastapi import pydantic @@ -37,14 +38,14 @@ async def post(body: RequestPostBody): @router.put('/api/v1/experiments', status_code=fastapi.status.HTTP_202_ACCEPTED, responses={404: {'description': 'Not found'}}) async def put(body: RequestPutBody): - changed = await experiments.update(id=body.id, enabled=body.enabled, condition=body.condition) + changed = await experiments.update(id=bson.ObjectId(body.id), enabled=body.enabled, condition=body.condition) if not changed: raise fastapi.HTTPException(404) @router.delete('/api/v1/experiments', status_code=fastapi.status.HTTP_202_ACCEPTED, responses={404: {'description': 'Not found'}}) async def delete(body: RequestDeleteBody): - changed = await experiments.delete(id=body.id) + changed = await experiments.delete(id=bson.ObjectId(body.id)) if not changed: raise fastapi.HTTPException(404) diff --git a/app/utils/monitoring.py b/app/utils/monitoring.py new file mode 100644 index 0000000..0eb52d5 --- /dev/null +++ b/app/utils/monitoring.py @@ -0,0 +1,24 @@ +from concurrent.futures import ThreadPoolExecutor +import datetime +import requests + + +class Monitroing: + def __init__(self): + self.executor = ThreadPoolExecutor(max_workers=1) + + def send_metric(self, start: datetime.datetime, end: datetime.datetime, endpoint: str, status_code: int, method: str): + def send(): + requests.post(f'http://monitoring:1237/api/v1/metrics/endpoint', json={ + 'timestamp': start.strftime("%Y-%m-%dT%H:%M:%S") + "Z", + 'service': 'configurator', + 'endpoint': endpoint, + 'status_code': status_code, + 'response_time': (end - start).microseconds // 1000, + 'method': method, + }) + + self.executor.submit(send) + + +monitoring = Monitroing() diff --git a/main.py b/main.py index 2c2603e..e58a5b2 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,7 @@ import fastapi import uvicorn +from app.middlewares.metrics import MetricsMiddleware from app.routers import experiments from app.routers import configs from app.routers import staff @@ -10,6 +11,7 @@ from app.storage import mongo app = fastapi.FastAPI() +app.add_middleware(MetricsMiddleware) app.include_router(experiments.router) app.include_router(configs.router) diff --git a/requirements.txt b/requirements.txt index 325371f..fdde3ac 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,8 @@ annotated-types==0.7.0 anyio==4.6.2.post1 APScheduler==3.10.4 +certifi==2025.6.15 +charset-normalizer==3.4.2 click==8.1.7 dnspython==2.7.0 fastapi==0.115.4 @@ -12,9 +14,11 @@ pydantic_core==2.23.4 pymongo==4.9.2 pytz==2024.2 redis==5.2.0 +requests==2.32.4 six==1.16.0 sniffio==1.3.1 starlette==0.41.2 typing_extensions==4.12.2 tzlocal==5.2 +urllib3==2.4.0 uvicorn==0.32.0