mirror of
https://github.com/borbann-platform/backend-api.git
synced 2025-12-18 20:24:05 +01:00
77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
from fastapi import FastAPI
|
|
from contextlib import asynccontextmanager
|
|
import platform
|
|
import asyncio
|
|
from loguru import logger
|
|
|
|
from stores.memory import InMemoryPipelineStore
|
|
from stores.base import PipelineStore
|
|
from services.pipeline_service import PipelineService
|
|
from scheduler.manager import SchedulerManager
|
|
from routers.pipelines import router as pipelines_router
|
|
|
|
# ! Window specific asyncio policy
|
|
if platform.system() == "Windows":
|
|
logger.info("Setting WindowsProactorEventLoopPolicy for asyncio.")
|
|
asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
|
|
|
|
# --- Resource Initialization ---
|
|
pipeline_store: PipelineStore = InMemoryPipelineStore()
|
|
pipeline_service = PipelineService(store=pipeline_store)
|
|
scheduler_manager = SchedulerManager(pipeline_service=pipeline_service)
|
|
|
|
# to avoid circular import
|
|
pipeline_service.set_scheduler_manager(scheduler_manager)
|
|
|
|
|
|
# --- Lifespan Management (for startup/shutdown) ---
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
logger.info("Application startup...")
|
|
# Store instances in app state for dependency injection
|
|
app.state.pipeline_store = pipeline_store
|
|
app.state.scheduler_manager = scheduler_manager
|
|
app.state.pipeline_service = pipeline_service
|
|
|
|
# Initialize and start the scheduler
|
|
logger.info("Initializing and starting SchedulerManager...")
|
|
|
|
scheduler_manager.start()
|
|
logger.info("SchedulerManager started.")
|
|
|
|
yield
|
|
|
|
# --- Shutdown ---
|
|
logger.info("Application shutdown...")
|
|
logger.info("Shutting down SchedulerManager...")
|
|
scheduler_manager.stop()
|
|
logger.info("SchedulerManager stopped.")
|
|
logger.info("Cleanup complete.")
|
|
|
|
|
|
# --- FastAPI App ---
|
|
app = FastAPI(
|
|
title="Data Integration Pipeline API",
|
|
description="API for managing and running data integration pipelines.",
|
|
version="0.1.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
# Include the pipelines router
|
|
app.include_router(pipelines_router)
|
|
|
|
|
|
# --- Root Endpoint (Optional) ---
|
|
@app.get("/", tags=["Root"])
|
|
async def read_root():
|
|
return {"message": "Welcome to the Data Integration Pipeline API"}
|
|
|
|
|
|
# --- Run with Uvicorn (Example) ---
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
|
|
logger.info("Starting Uvicorn server...")
|
|
# ! use reload=True only for development
|
|
uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True, loop="asyncio")
|