Add role-based filtering and imporve code

services/alert_processor/app/main.py

@@ -11,7 +11,7 @@ import sys
 from datetime import datetime
 from typing import Dict, Any
 import structlog
-import redis.asyncio as aioredis
+from shared.redis_utils import initialize_redis, close_redis, get_redis_client
 from aio_pika import connect_robust, IncomingMessage, ExchangeType
 
 from app.config import AlertProcessorConfig
@@ -92,9 +92,10 @@ class AlertProcessorService:
         """Start the alert processor service"""
         try:
             logger.info("Starting Alert Processor Service")
-            
-            # Connect to Redis for SSE publishing
-            self.redis = aioredis.from_url(self.config.REDIS_URL)
+
+            # Initialize shared Redis connection for SSE publishing
+            await initialize_redis(self.config.REDIS_URL, db=0, max_connections=20)
+            self.redis = await get_redis_client()
             logger.info("Connected to Redis")
             
             # Connect to RabbitMQ
@@ -306,18 +307,17 @@ class AlertProcessorService:
         """Stop the alert processor service"""
         self.running = False
         logger.info("Stopping Alert Processor Service")
-        
+
         try:
             # Close RabbitMQ connection
             if self.connection and not self.connection.is_closed:
                 await self.connection.close()
-            
-            # Close Redis connection
-            if self.redis:
-                await self.redis.close()
-                
+
+            # Close shared Redis connection
+            await close_redis()
+
             logger.info("Alert Processor Service stopped")
-            
+
         except Exception as e:
             logger.error("Error stopping service", error=str(e))
     

services/alert_processor/app/models/__init__.py

@@ -4,6 +4,13 @@ Alert Processor Service Models Package
 Import all models to ensure they are registered with SQLAlchemy Base.
 """
 
+# Import AuditLog model for this service
+from shared.security import create_audit_log_model
+from shared.database.base import Base
+
+# Create audit log model for this service
+AuditLog = create_audit_log_model(Base)
+
 # Import all models to register them with the Base metadata
 from .alerts import Alert, AlertStatus, AlertSeverity
 
@@ -12,4 +19,5 @@ __all__ = [
     "Alert",
     "AlertStatus",
     "AlertSeverity",
+    "AuditLog",
 ]