Improve teh securty of teh DB

services/inventory/app/api/internal_demo.py

@@ -20,6 +20,7 @@ sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent.parent))
 from app.core.database import get_db
 from app.models.inventory import Ingredient, Stock
 from shared.utils.demo_dates import adjust_date_for_demo, BASE_REFERENCE_DATE
+from shared.messaging.rabbitmq import RabbitMQClient
 
 logger = structlog.get_logger()
 router = APIRouter(prefix="/internal/demo", tags=["internal"])
@@ -231,16 +232,42 @@ async def clone_demo_data(
         # Commit all changes
         await db.commit()
 
-        # Generate inventory alerts
+        # Generate inventory alerts with RabbitMQ publishing
+        rabbitmq_client = None
         try:
             from shared.utils.alert_generator import generate_inventory_alerts
-            alerts_count = await generate_inventory_alerts(db, virtual_uuid, session_created_at)
+
+            # Initialize RabbitMQ client for alert publishing
+            rabbitmq_host = os.getenv("RABBITMQ_HOST", "rabbitmq-service")
+            rabbitmq_user = os.getenv("RABBITMQ_USER", "bakery")
+            rabbitmq_password = os.getenv("RABBITMQ_PASSWORD", "forecast123")
+            rabbitmq_port = os.getenv("RABBITMQ_PORT", "5672")
+            rabbitmq_vhost = os.getenv("RABBITMQ_VHOST", "/")
+            rabbitmq_url = f"amqp://{rabbitmq_user}:{rabbitmq_password}@{rabbitmq_host}:{rabbitmq_port}{rabbitmq_vhost}"
+
+            rabbitmq_client = RabbitMQClient(rabbitmq_url, service_name="inventory")
+            await rabbitmq_client.connect()
+
+            # Generate alerts and publish to RabbitMQ
+            alerts_count = await generate_inventory_alerts(
+                db,
+                virtual_uuid,
+                session_created_at,
+                rabbitmq_client=rabbitmq_client
+            )
             stats["alerts_generated"] = alerts_count
-            await db.commit()  # Commit alerts
+            await db.commit()
             logger.info(f"Generated {alerts_count} inventory alerts", virtual_tenant_id=virtual_tenant_id)
         except Exception as e:
             logger.warning(f"Failed to generate alerts: {str(e)}", exc_info=True)
             stats["alerts_generated"] = 0
+        finally:
+            # Clean up RabbitMQ connection
+            if rabbitmq_client:
+                try:
+                    await rabbitmq_client.disconnect()
+                except Exception as cleanup_error:
+                    logger.warning(f"Error disconnecting RabbitMQ: {cleanup_error}")
 
         total_records = sum(stats.values())
         duration_ms = int((datetime.now(timezone.utc) - start_time).total_seconds() * 1000)