84 lines
2.3 KiB
Python
84 lines
2.3 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import os
|
||
|
|
from typing import Any, Dict, List, Optional
|
||
|
|
|
||
|
|
from fastapi import FastAPI, Body, HTTPException
|
||
|
|
from fastapi.responses import JSONResponse
|
||
|
|
from pydantic import BaseModel
|
||
|
|
import httpx
|
||
|
|
|
||
|
|
from .enrichment import Enricher
|
||
|
|
from .adapter import normalize_telemetry
|
||
|
|
|
||
|
|
app = FastAPI(title="HPCSim Enrichment API", version="0.1.0")
|
||
|
|
|
||
|
|
# Single Enricher instance keeps state across laps
|
||
|
|
_enricher = Enricher()
|
||
|
|
|
||
|
|
# Simple in-memory store of recent enriched records
|
||
|
|
_recent: List[Dict[str, Any]] = []
|
||
|
|
_MAX_RECENT = 200
|
||
|
|
|
||
|
|
# Optional callback URL to forward enriched data to next stage
|
||
|
|
_CALLBACK_URL = os.getenv("NEXT_STAGE_CALLBACK_URL")
|
||
|
|
|
||
|
|
|
||
|
|
class EnrichedRecord(BaseModel):
|
||
|
|
lap: int
|
||
|
|
aero_efficiency: float
|
||
|
|
tire_degradation_index: float
|
||
|
|
ers_charge: float
|
||
|
|
fuel_optimization_score: float
|
||
|
|
driver_consistency: float
|
||
|
|
weather_impact: str
|
||
|
|
|
||
|
|
|
||
|
|
@app.post("/ingest/telemetry")
|
||
|
|
async def ingest_telemetry(payload: Dict[str, Any] = Body(...)):
|
||
|
|
"""Receive raw telemetry (from Pi), normalize, enrich, return enriched.
|
||
|
|
|
||
|
|
Optionally forwards to NEXT_STAGE_CALLBACK_URL if set.
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
normalized = normalize_telemetry(payload)
|
||
|
|
enriched = _enricher.enrich(normalized)
|
||
|
|
except Exception as e:
|
||
|
|
raise HTTPException(status_code=400, detail=f"Failed to enrich: {e}")
|
||
|
|
|
||
|
|
_recent.append(enriched)
|
||
|
|
if len(_recent) > _MAX_RECENT:
|
||
|
|
del _recent[: len(_recent) - _MAX_RECENT]
|
||
|
|
|
||
|
|
# Async forward to next stage if configured
|
||
|
|
if _CALLBACK_URL:
|
||
|
|
try:
|
||
|
|
async with httpx.AsyncClient(timeout=5.0) as client:
|
||
|
|
await client.post(_CALLBACK_URL, json=enriched)
|
||
|
|
except Exception:
|
||
|
|
# Don't fail ingestion if forwarding fails; log could be added here
|
||
|
|
pass
|
||
|
|
|
||
|
|
return JSONResponse(enriched)
|
||
|
|
|
||
|
|
|
||
|
|
@app.post("/enriched")
|
||
|
|
async def post_enriched(enriched: EnrichedRecord):
|
||
|
|
"""Allow posting externally enriched records (bypass local computation)."""
|
||
|
|
rec = enriched.model_dump()
|
||
|
|
_recent.append(rec)
|
||
|
|
if len(_recent) > _MAX_RECENT:
|
||
|
|
del _recent[: len(_recent) - _MAX_RECENT]
|
||
|
|
return JSONResponse(rec)
|
||
|
|
|
||
|
|
|
||
|
|
@app.get("/enriched")
|
||
|
|
async def list_enriched(limit: int = 50):
|
||
|
|
limit = max(1, min(200, limit))
|
||
|
|
return JSONResponse(_recent[-limit:])
|
||
|
|
|
||
|
|
|
||
|
|
@app.get("/healthz")
|
||
|
|
async def healthz():
|
||
|
|
return {"status": "ok", "stored": len(_recent)}
|