mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
be4fa60970
Implements complete M3 milestone with TelemetryService, two-phase scheduling, collection/upload loops, Replicated SDK integration, and FastAPI lifespan. - TelemetryService with 3-min bootstrap and 1-hour normal intervals - Optional Replicated SDK with fallback UUID generation - 85 unit tests and 14 integration tests passing - Updated design doc checklist Co-authored-by: openhands <openhands@all-hands.dev>
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""FastAPI lifespan integration for the embedded telemetry service."""
|
|
|
|
from contextlib import asynccontextmanager
|
|
from typing import AsyncIterator
|
|
|
|
from fastapi import FastAPI
|
|
|
|
from server.logger import logger
|
|
from server.telemetry.service import telemetry_service
|
|
|
|
|
|
@asynccontextmanager
|
|
async def telemetry_lifespan(app: FastAPI) -> AsyncIterator[None]:
|
|
"""FastAPI lifespan context manager for telemetry service.
|
|
|
|
This is called automatically during FastAPI application startup and shutdown,
|
|
managing the lifecycle of the telemetry background tasks.
|
|
|
|
Startup: Initializes and starts background collection and upload tasks
|
|
Shutdown: Gracefully stops background tasks
|
|
"""
|
|
logger.info('Starting telemetry service lifespan')
|
|
|
|
# Startup - start background tasks
|
|
try:
|
|
await telemetry_service.start()
|
|
logger.info('Telemetry service started successfully')
|
|
except Exception as e:
|
|
logger.error(f'Error starting telemetry service: {e}', exc_info=True)
|
|
# Don't fail server startup if telemetry fails
|
|
|
|
yield # Server runs here
|
|
|
|
# Shutdown - stop background tasks
|
|
try:
|
|
await telemetry_service.stop()
|
|
logger.info('Telemetry service stopped successfully')
|
|
except Exception as e:
|
|
logger.error(f'Error stopping telemetry service: {e}', exc_info=True)
|