Initial commit - production deployment
This commit is contained in:
346
services/orchestrator/app/api/orchestration.py
Normal file
346
services/orchestrator/app/api/orchestration.py
Normal file
@@ -0,0 +1,346 @@
|
||||
# ================================================================
|
||||
# services/orchestrator/app/api/orchestration.py
|
||||
# ================================================================
|
||||
"""
|
||||
Orchestration API Endpoints
|
||||
Testing and manual trigger endpoints for orchestration
|
||||
"""
|
||||
|
||||
import uuid
|
||||
from typing import Optional
|
||||
from fastapi import APIRouter, Depends, HTTPException, Request
|
||||
from pydantic import BaseModel, Field
|
||||
import structlog
|
||||
|
||||
from app.core.database import get_db
|
||||
from app.repositories.orchestration_run_repository import OrchestrationRunRepository
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
logger = structlog.get_logger()
|
||||
|
||||
router = APIRouter(prefix="/api/v1/tenants/{tenant_id}/orchestrator", tags=["Orchestration"])
|
||||
|
||||
|
||||
# ================================================================
|
||||
# REQUEST/RESPONSE SCHEMAS
|
||||
# ================================================================
|
||||
|
||||
class OrchestratorTestRequest(BaseModel):
|
||||
"""Request schema for testing orchestrator"""
|
||||
test_scenario: Optional[str] = Field(None, description="Test scenario: full, production_only, procurement_only")
|
||||
dry_run: bool = Field(False, description="Dry run mode (no actual changes)")
|
||||
|
||||
|
||||
class OrchestratorTestResponse(BaseModel):
|
||||
"""Response schema for orchestrator test"""
|
||||
success: bool
|
||||
message: str
|
||||
tenant_id: str
|
||||
forecasting_completed: bool = False
|
||||
production_completed: bool = False
|
||||
procurement_completed: bool = False
|
||||
notifications_sent: bool = False
|
||||
summary: dict = {}
|
||||
|
||||
|
||||
class OrchestratorWorkflowRequest(BaseModel):
|
||||
"""Request schema for daily workflow trigger"""
|
||||
dry_run: bool = Field(False, description="Dry run mode (no actual changes)")
|
||||
|
||||
|
||||
class OrchestratorWorkflowResponse(BaseModel):
|
||||
"""Response schema for daily workflow trigger"""
|
||||
success: bool
|
||||
message: str
|
||||
tenant_id: str
|
||||
run_id: Optional[str] = None
|
||||
forecasting_completed: bool = False
|
||||
production_completed: bool = False
|
||||
procurement_completed: bool = False
|
||||
notifications_sent: bool = False
|
||||
summary: dict = {}
|
||||
|
||||
|
||||
# ================================================================
|
||||
# API ENDPOINTS
|
||||
# ================================================================
|
||||
|
||||
@router.post("/test", response_model=OrchestratorTestResponse)
|
||||
async def trigger_orchestrator_test(
|
||||
tenant_id: str,
|
||||
request_data: OrchestratorTestRequest,
|
||||
request: Request,
|
||||
db: AsyncSession = Depends(get_db)
|
||||
):
|
||||
"""
|
||||
Trigger orchestrator for testing purposes
|
||||
|
||||
This endpoint allows manual triggering of the orchestration workflow
|
||||
for a specific tenant, useful for testing during development.
|
||||
|
||||
Args:
|
||||
tenant_id: Tenant ID to orchestrate
|
||||
request_data: Test request with scenario and dry_run options
|
||||
request: FastAPI request object
|
||||
db: Database session
|
||||
|
||||
Returns:
|
||||
OrchestratorTestResponse with results
|
||||
"""
|
||||
logger.info("Orchestrator test trigger requested",
|
||||
tenant_id=tenant_id,
|
||||
test_scenario=request_data.test_scenario,
|
||||
dry_run=request_data.dry_run)
|
||||
|
||||
try:
|
||||
# Get scheduler service from app state
|
||||
if not hasattr(request.app.state, 'scheduler_service'):
|
||||
raise HTTPException(
|
||||
status_code=503,
|
||||
detail="Orchestrator scheduler service not available"
|
||||
)
|
||||
|
||||
scheduler_service = request.app.state.scheduler_service
|
||||
|
||||
# Trigger orchestration
|
||||
tenant_uuid = uuid.UUID(tenant_id)
|
||||
result = await scheduler_service.trigger_orchestration_for_tenant(
|
||||
tenant_id=tenant_uuid,
|
||||
test_scenario=request_data.test_scenario
|
||||
)
|
||||
|
||||
# Get the latest run for this tenant
|
||||
repo = OrchestrationRunRepository(db)
|
||||
latest_run = await repo.get_latest_run_for_tenant(tenant_uuid)
|
||||
|
||||
# Build response
|
||||
response = OrchestratorTestResponse(
|
||||
success=result.get('success', False),
|
||||
message=result.get('message', 'Orchestration completed'),
|
||||
tenant_id=tenant_id,
|
||||
forecasting_completed=latest_run.forecasting_status == 'success' if latest_run else False,
|
||||
production_completed=latest_run.production_status == 'success' if latest_run else False,
|
||||
procurement_completed=latest_run.procurement_status == 'success' if latest_run else False,
|
||||
notifications_sent=latest_run.notification_status == 'success' if latest_run else False,
|
||||
summary={
|
||||
'forecasts_generated': latest_run.forecasts_generated if latest_run else 0,
|
||||
'batches_created': latest_run.production_batches_created if latest_run else 0,
|
||||
'pos_created': latest_run.purchase_orders_created if latest_run else 0,
|
||||
'notifications_sent': latest_run.notifications_sent if latest_run else 0
|
||||
}
|
||||
)
|
||||
|
||||
logger.info("Orchestrator test completed",
|
||||
tenant_id=tenant_id,
|
||||
success=response.success)
|
||||
|
||||
return response
|
||||
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=f"Invalid tenant ID: {str(e)}")
|
||||
except Exception as e:
|
||||
logger.error("Orchestrator test failed",
|
||||
tenant_id=tenant_id,
|
||||
error=str(e),
|
||||
exc_info=True)
|
||||
raise HTTPException(status_code=500, detail=f"Orchestrator test failed: {str(e)}")
|
||||
|
||||
|
||||
@router.post("/run-daily-workflow", response_model=OrchestratorWorkflowResponse)
|
||||
async def run_daily_workflow(
|
||||
tenant_id: str,
|
||||
request_data: Optional[OrchestratorWorkflowRequest] = None,
|
||||
request: Request = None,
|
||||
db: AsyncSession = Depends(get_db)
|
||||
):
|
||||
"""
|
||||
Trigger the daily orchestrated workflow for a tenant
|
||||
|
||||
This endpoint runs the complete daily workflow which includes:
|
||||
1. Forecasting Service: Generate demand forecasts
|
||||
2. Production Service: Create production schedule from forecasts
|
||||
3. Procurement Service: Generate procurement plan
|
||||
4. Notification Service: Send relevant notifications
|
||||
|
||||
This is the production endpoint used by the dashboard scheduler button.
|
||||
|
||||
Args:
|
||||
tenant_id: Tenant ID to orchestrate
|
||||
request_data: Optional request data with dry_run flag
|
||||
request: FastAPI request object
|
||||
db: Database session
|
||||
|
||||
Returns:
|
||||
OrchestratorWorkflowResponse with workflow execution results
|
||||
"""
|
||||
logger.info("Daily workflow trigger requested", tenant_id=tenant_id)
|
||||
|
||||
# Handle optional request_data
|
||||
if request_data is None:
|
||||
request_data = OrchestratorWorkflowRequest()
|
||||
|
||||
try:
|
||||
# Get scheduler service from app state
|
||||
if not hasattr(request.app.state, 'scheduler_service'):
|
||||
raise HTTPException(
|
||||
status_code=503,
|
||||
detail="Orchestrator scheduler service not available"
|
||||
)
|
||||
|
||||
scheduler_service = request.app.state.scheduler_service
|
||||
|
||||
# Trigger orchestration (use full workflow, not test scenario)
|
||||
tenant_uuid = uuid.UUID(tenant_id)
|
||||
result = await scheduler_service.trigger_orchestration_for_tenant(
|
||||
tenant_id=tenant_uuid,
|
||||
test_scenario=None # Full production workflow
|
||||
)
|
||||
|
||||
# Get the latest run for this tenant
|
||||
repo = OrchestrationRunRepository(db)
|
||||
latest_run = await repo.get_latest_run_for_tenant(tenant_uuid)
|
||||
|
||||
# Build response
|
||||
response = OrchestratorWorkflowResponse(
|
||||
success=result.get('success', False),
|
||||
message=result.get('message', 'Daily workflow completed successfully'),
|
||||
tenant_id=tenant_id,
|
||||
run_id=str(latest_run.id) if latest_run else None,
|
||||
forecasting_completed=latest_run.forecasting_status == 'success' if latest_run else False,
|
||||
production_completed=latest_run.production_status == 'success' if latest_run else False,
|
||||
procurement_completed=latest_run.procurement_status == 'success' if latest_run else False,
|
||||
notifications_sent=latest_run.notification_status == 'success' if latest_run else False,
|
||||
summary={
|
||||
'run_number': latest_run.run_number if latest_run else 0,
|
||||
'forecasts_generated': latest_run.forecasts_generated if latest_run else 0,
|
||||
'production_batches_created': latest_run.production_batches_created if latest_run else 0,
|
||||
'purchase_orders_created': latest_run.purchase_orders_created if latest_run else 0,
|
||||
'notifications_sent': latest_run.notifications_sent if latest_run else 0,
|
||||
'duration_seconds': latest_run.duration_seconds if latest_run else 0
|
||||
}
|
||||
)
|
||||
|
||||
logger.info("Daily workflow completed",
|
||||
tenant_id=tenant_id,
|
||||
success=response.success,
|
||||
run_id=response.run_id)
|
||||
|
||||
return response
|
||||
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=f"Invalid tenant ID: {str(e)}")
|
||||
except Exception as e:
|
||||
logger.error("Daily workflow failed",
|
||||
tenant_id=tenant_id,
|
||||
error=str(e),
|
||||
exc_info=True)
|
||||
raise HTTPException(status_code=500, detail=f"Daily workflow failed: {str(e)}")
|
||||
|
||||
|
||||
@router.get("/health")
|
||||
async def orchestrator_health():
|
||||
"""Check orchestrator health"""
|
||||
return {
|
||||
"status": "healthy",
|
||||
"service": "orchestrator",
|
||||
"message": "Orchestrator service is running"
|
||||
}
|
||||
|
||||
|
||||
@router.get("/runs", response_model=dict)
|
||||
async def list_orchestration_runs(
|
||||
tenant_id: str,
|
||||
limit: int = 10,
|
||||
offset: int = 0,
|
||||
db: AsyncSession = Depends(get_db)
|
||||
):
|
||||
"""
|
||||
List orchestration runs for a tenant
|
||||
|
||||
Args:
|
||||
tenant_id: Tenant ID
|
||||
limit: Maximum number of runs to return
|
||||
offset: Number of runs to skip
|
||||
db: Database session
|
||||
|
||||
Returns:
|
||||
List of orchestration runs
|
||||
"""
|
||||
try:
|
||||
tenant_uuid = uuid.UUID(tenant_id)
|
||||
repo = OrchestrationRunRepository(db)
|
||||
|
||||
runs = await repo.list_runs(
|
||||
tenant_id=tenant_uuid,
|
||||
limit=limit,
|
||||
offset=offset
|
||||
)
|
||||
|
||||
return {
|
||||
"runs": [
|
||||
{
|
||||
"id": str(run.id),
|
||||
"run_number": run.run_number,
|
||||
"status": run.status.value,
|
||||
"started_at": run.started_at.isoformat() if run.started_at else None,
|
||||
"completed_at": run.completed_at.isoformat() if run.completed_at else None,
|
||||
"duration_seconds": run.duration_seconds,
|
||||
"forecasts_generated": run.forecasts_generated,
|
||||
"batches_created": run.production_batches_created,
|
||||
"pos_created": run.purchase_orders_created
|
||||
}
|
||||
for run in runs
|
||||
],
|
||||
"total": len(runs),
|
||||
"limit": limit,
|
||||
"offset": offset
|
||||
}
|
||||
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=f"Invalid tenant ID: {str(e)}")
|
||||
except Exception as e:
|
||||
logger.error("Error listing orchestration runs",
|
||||
tenant_id=tenant_id,
|
||||
error=str(e))
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/last-run")
|
||||
async def get_last_orchestration_run(
|
||||
tenant_id: str,
|
||||
db: AsyncSession = Depends(get_db)
|
||||
):
|
||||
"""
|
||||
Get timestamp of last orchestration run
|
||||
|
||||
Lightweight endpoint for health status frontend migration (Phase 4).
|
||||
Returns only timestamp and run number for the most recent completed run.
|
||||
|
||||
Args:
|
||||
tenant_id: Tenant ID
|
||||
|
||||
Returns:
|
||||
Dict with timestamp and runNumber (or None if no runs)
|
||||
"""
|
||||
try:
|
||||
tenant_uuid = uuid.UUID(tenant_id)
|
||||
repo = OrchestrationRunRepository(db)
|
||||
|
||||
# Get most recent completed run
|
||||
latest_run = await repo.get_latest_run_for_tenant(tenant_uuid)
|
||||
|
||||
if not latest_run:
|
||||
return {"timestamp": None, "runNumber": None}
|
||||
|
||||
return {
|
||||
"timestamp": latest_run.started_at.isoformat() if latest_run.started_at else None,
|
||||
"runNumber": latest_run.run_number
|
||||
}
|
||||
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=f"Invalid tenant ID: {str(e)}")
|
||||
except Exception as e:
|
||||
logger.error("Error getting last orchestration run",
|
||||
tenant_id=tenant_id,
|
||||
error=str(e))
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
Reference in New Issue
Block a user