Initial commit - production deployment

services/distribution/app/main.py

@@ -0,0 +1,127 @@
+"""
+Distribution Service Main Application
+"""
+
+from fastapi import FastAPI
+from sqlalchemy import text
+from app.core.config import settings
+from app.core.database import database_manager
+from app.api.routes import router as distribution_router
+from app.api.shipments import router as shipments_router
+from app.api.internal_demo import router as internal_demo_router
+from app.api.vrp_optimization import router as vrp_optimization_router
+from shared.service_base import StandardFastAPIService
+
+
+class DistributionService(StandardFastAPIService):
+    """Distribution Service with standardized setup"""
+
+    async def on_startup(self, app):
+        """Custom startup logic including migration verification"""
+        await self.verify_migrations()
+        await super().on_startup(app)
+
+    async def verify_migrations(self):
+        """Verify database schema matches the latest migrations."""
+        try:
+            async with self.database_manager.get_session() as session:
+                # Check if alembic_version table exists
+                result = await session.execute(text("""
+                    SELECT EXISTS (
+                        SELECT FROM information_schema.tables
+                        WHERE table_schema = 'public'
+                        AND table_name = 'alembic_version'
+                    )
+                """))
+                table_exists = result.scalar()
+
+                if table_exists:
+                    # If table exists, check the version
+                    result = await session.execute(text("SELECT version_num FROM alembic_version"))
+                    version = result.scalar()
+                    self.logger.info(f"Migration verification successful: {version}")
+                else:
+                    # If table doesn't exist, migrations might not have run yet
+                    # This is OK - the migration job should create it
+                    self.logger.warning("alembic_version table does not exist yet - migrations may not have run")
+
+        except Exception as e:
+            self.logger.warning(f"Migration verification failed (this may be expected during initial setup): {e}")
+
+    def __init__(self):
+        # Define expected database tables for health checks
+        # Must match tables created in migrations/versions/001_initial_schema.py
+        distribution_expected_tables = [
+            'delivery_routes', 'shipments', 'delivery_schedules'
+        ]
+
+        # Define custom metrics for distribution service
+        distribution_custom_metrics = {
+            "routes_generated_total": {
+                "type": "counter",
+                "description": "Total delivery routes generated"
+            },
+            "shipments_processed_total": {
+                "type": "counter",
+                "description": "Total shipments processed"
+            },
+            "route_optimization_time_seconds": {
+                "type": "histogram",
+                "description": "Time to optimize delivery routes"
+            },
+            "shipment_processing_time_seconds": {
+                "type": "histogram",
+                "description": "Time to process shipment request"
+            },
+            "delivery_completion_rate": {
+                "type": "counter",
+                "description": "Delivery completion rate by status",
+                "labels": ["status"]
+            }
+        }
+
+        super().__init__(
+            service_name="distribution-service",
+            app_name="Distribution Service",
+            description="Distribution and logistics service for enterprise tier bakery management",
+            version="1.0.0",
+            log_level=settings.LOG_LEVEL,
+            api_prefix="",  # Empty because RouteBuilder already includes /api/v1
+            database_manager=database_manager,
+            expected_tables=distribution_expected_tables,
+            custom_metrics=distribution_custom_metrics
+        )
+
+    async def on_shutdown(self, app: FastAPI):
+        """Custom shutdown logic for distribution service"""
+        self.logger.info("Distribution Service shutdown complete")
+
+    def get_service_features(self):
+        """Return distribution-specific features"""
+        return [
+            "delivery_route_optimization",
+            "shipment_tracking",
+            "vehicle_assignment",
+            "distribution_planning",
+            "delivery_point_management"
+        ]
+
+
+# Create service instance
+service = DistributionService()
+
+# Create FastAPI app with standardized setup
+app = service.create_app(
+    docs_url="/docs",
+    redoc_url="/redoc"
+)
+
+# Setup standard endpoints
+service.setup_standard_endpoints()
+
+# Include routers with specific configurations
+# Note: Routes now use RouteBuilder which includes full paths, so no prefix needed
+service.add_router(distribution_router, tags=["distribution"])
+service.add_router(shipments_router, tags=["shipments"])
+service.add_router(internal_demo_router, tags=["internal-demo"])
+service.add_router(vrp_optimization_router, tags=["vrp-optimization"])