61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
# ================================================================
|
|
# Gateway Integration: Update gateway/app/routes/forecasting.py
|
|
# ================================================================
|
|
"""Forecasting service routes for API Gateway"""
|
|
|
|
from fastapi import APIRouter, Request
|
|
import httpx
|
|
import logging
|
|
|
|
from app.core.config import settings
|
|
|
|
logger = logging.getLogger(__name__)
|
|
router = APIRouter()
|
|
|
|
@router.api_route("/forecasts/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
|
|
async def proxy_forecasts(request: Request, path: str):
|
|
"""Proxy forecast requests to forecasting service"""
|
|
return await _proxy_request(request, f"/api/v1/forecasts/{path}")
|
|
|
|
@router.api_route("/predictions/{path:path}", methods=["GET", "POST"])
|
|
async def proxy_predictions(request: Request, path: str):
|
|
"""Proxy prediction requests to forecasting service"""
|
|
return await _proxy_request(request, f"/api/v1/predictions/{path}")
|
|
|
|
async def _proxy_request(request: Request, target_path: str):
|
|
"""Proxy request to forecasting service with user context"""
|
|
try:
|
|
url = f"{settings.FORECASTING_SERVICE_URL}{target_path}"
|
|
|
|
# Forward headers and add user context
|
|
headers = dict(request.headers)
|
|
headers.pop("host", None)
|
|
|
|
# Add user context from gateway authentication
|
|
if hasattr(request.state, 'user'):
|
|
headers["X-User-ID"] = str(request.state.user.get("user_id"))
|
|
headers["X-User-Email"] = request.state.user.get("email", "")
|
|
headers["X-Tenant-ID"] = str(request.state.user.get("tenant_id"))
|
|
headers["X-User-Roles"] = ",".join(request.state.user.get("roles", []))
|
|
|
|
# Get request body if present
|
|
body = None
|
|
if request.method in ["POST", "PUT", "PATCH"]:
|
|
body = await request.body()
|
|
|
|
async with httpx.AsyncClient(timeout=30.0) as client:
|
|
response = await client.request(
|
|
method=request.method,
|
|
url=url,
|
|
headers=headers,
|
|
content=body,
|
|
params=request.query_params
|
|
)
|
|
|
|
# Return response
|
|
return response.json() if response.headers.get("content-type", "").startswith("application/json") else response.text
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error proxying to forecasting service: {e}")
|
|
raise
|