Add forecasting service

services/forecasting/app/services/messaging.py

@@ -0,0 +1,98 @@
+# ================================================================
+# services/forecasting/app/services/messaging.py  
+# ================================================================
+"""
+Messaging service for event publishing and consuming
+"""
+
+import structlog
+import json
+from typing import Dict, Any
+import asyncio
+
+from shared.messaging.rabbitmq import RabbitMQPublisher, RabbitMQConsumer
+from app.core.config import settings
+
+logger = structlog.get_logger()
+
+# Global messaging instances
+publisher = None
+consumer = None
+
+async def setup_messaging():
+    """Initialize messaging services"""
+    global publisher, consumer
+    
+    try:
+        # Initialize publisher
+        publisher = RabbitMQPublisher(settings.RABBITMQ_URL)
+        await publisher.connect()
+        
+        # Initialize consumer
+        consumer = RabbitMQConsumer(settings.RABBITMQ_URL)
+        await consumer.connect()
+        
+        # Set up event handlers
+        await consumer.subscribe("training.model.updated", handle_model_updated)
+        await consumer.subscribe("data.weather.updated", handle_weather_updated)
+        
+        logger.info("Messaging setup completed")
+        
+    except Exception as e:
+        logger.error("Failed to setup messaging", error=str(e))
+        raise
+
+async def cleanup_messaging():
+    """Cleanup messaging connections"""
+    global publisher, consumer
+    
+    try:
+        if consumer:
+            await consumer.close()
+        if publisher:
+            await publisher.close()
+        
+        logger.info("Messaging cleanup completed")
+        
+    except Exception as e:
+        logger.error("Error during messaging cleanup", error=str(e))
+
+async def publish_forecast_completed(data: Dict[str, Any]):
+    """Publish forecast completed event"""
+    if publisher:
+        await publisher.publish("forecasting.forecast.completed", data)
+
+async def publish_alert_created(data: Dict[str, Any]):
+    """Publish alert created event"""
+    if publisher:
+        await publisher.publish("forecasting.alert.created", data)
+
+async def publish_batch_completed(data: Dict[str, Any]):
+    """Publish batch forecast completed event"""
+    if publisher:
+        await publisher.publish("forecasting.batch.completed", data)
+
+# Event handlers
+async def handle_model_updated(data: Dict[str, Any]):
+    """Handle model updated event from training service"""
+    try:
+        logger.info("Received model updated event", 
+                   model_id=data.get("model_id"),
+                   tenant_id=data.get("tenant_id"))
+        
+        # Clear model cache for this model
+        # This will be handled by PredictionService
+        
+    except Exception as e:
+        logger.error("Error handling model updated event", error=str(e))
+
+async def handle_weather_updated(data: Dict[str, Any]):
+    """Handle weather data updated event"""
+    try:
+        logger.info("Received weather updated event", 
+                   date=data.get("date"))
+        
+        # Could trigger re-forecasting if needed
+        
+    except Exception as e:
+        logger.error("Error handling weather updated event", error=str(e))