Add more services

services/orders/app/core/database.py

@@ -0,0 +1,80 @@
+# ================================================================
+# services/orders/app/core/database.py
+# ================================================================
+"""
+Orders Service Database Configuration
+"""
+
+from sqlalchemy import create_engine
+from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker, AsyncSession
+from sqlalchemy.orm import sessionmaker, DeclarativeBase
+import structlog
+from typing import AsyncGenerator
+
+from app.core.config import settings
+
+logger = structlog.get_logger()
+
+# Create async engine
+async_engine = create_async_engine(
+    settings.DATABASE_URL,
+    echo=settings.DEBUG,
+    pool_size=10,
+    max_overflow=20,
+    pool_pre_ping=True,
+    pool_recycle=3600
+)
+
+# Create async session factory
+AsyncSessionLocal = async_sessionmaker(
+    bind=async_engine,
+    class_=AsyncSession,
+    expire_on_commit=False
+)
+
+# Base class for models
+class Base(DeclarativeBase):
+    pass
+
+
+async def get_db() -> AsyncGenerator[AsyncSession, None]:
+    """Get database session"""
+    async with AsyncSessionLocal() as session:
+        try:
+            yield session
+        except Exception as e:
+            await session.rollback()
+            logger.error("Database session error", error=str(e))
+            raise
+        finally:
+            await session.close()
+
+
+async def init_database():
+    """Initialize database tables"""
+    try:
+        async with async_engine.begin() as conn:
+            # Import all models to ensure they are registered
+            from app.models.order import CustomerOrder, OrderItem, OrderStatusHistory
+            from app.models.customer import Customer, CustomerContact
+            from app.models.procurement import ProcurementPlan, ProcurementRequirement
+            from app.models.alerts import OrderAlert
+            
+            # Create all tables
+            await conn.run_sync(Base.metadata.create_all)
+            
+        logger.info("Orders database initialized successfully")
+    except Exception as e:
+        logger.error("Failed to initialize orders database", error=str(e))
+        raise
+
+
+async def get_db_health() -> bool:
+    """Check database health"""
+    try:
+        async with async_engine.begin() as conn:
+            await conn.execute("SELECT 1")
+        return True
+    except Exception as e:
+        logger.error("Database health check failed", error=str(e))
+        return False