Add subcription feature 6

services/auth/app/main.py

@@ -15,10 +15,33 @@ class AuthService(StandardFastAPIService):
     """Authentication Service with standardized setup"""
 
     async def on_startup(self, app):
-        """Custom startup logic including migration verification"""
+        """Custom startup logic including migration verification and Redis initialization"""
+        self.logger.info("Starting auth service on_startup")
         await self.verify_migrations()
+
+        # Initialize Redis if not already done during service creation
+        if not self.redis_initialized:
+            try:
+                from shared.redis_utils import initialize_redis, get_redis_client
+                await initialize_redis(settings.REDIS_URL_WITH_DB, db=settings.REDIS_DB, max_connections=getattr(settings, 'REDIS_MAX_CONNECTIONS', 50))
+                self.redis_client = await get_redis_client()
+                self.redis_initialized = True
+                self.logger.info("Connected to Redis for token management")
+            except Exception as e:
+                self.logger.error(f"Failed to connect to Redis during startup: {e}")
+                raise
+
         await super().on_startup(app)
 
+    async def on_shutdown(self, app):
+        """Custom shutdown logic for Auth Service"""
+        await super().on_shutdown(app)
+
+        # Close Redis
+        from shared.redis_utils import close_redis
+        await close_redis()
+        self.logger.info("Redis connection closed")
+
     async def verify_migrations(self):
         """Verify database schema matches the latest migrations."""
         try:
@@ -47,6 +70,35 @@ class AuthService(StandardFastAPIService):
             self.logger.warning(f"Migration verification failed (this may be expected during initial setup): {e}")
 
     def __init__(self):
+        # Initialize Redis during service creation so it's available when needed
+        try:
+            import asyncio
+            # We need to run the async initialization in a sync context
+            try:
+                # Check if there's already a running event loop
+                loop = asyncio.get_running_loop()
+                # If there is, we'll initialize Redis later in on_startup
+                self.redis_initialized = False
+                self.redis_client = None
+            except RuntimeError:
+                # No event loop running, safe to run the async function
+                import asyncio
+                import nest_asyncio
+                nest_asyncio.apply()  # Allow nested event loops
+
+                async def init_redis():
+                    from shared.redis_utils import initialize_redis, get_redis_client
+                    await initialize_redis(settings.REDIS_URL_WITH_DB, db=settings.REDIS_DB, max_connections=getattr(settings, 'REDIS_MAX_CONNECTIONS', 50))
+                    return await get_redis_client()
+
+                self.redis_client = asyncio.run(init_redis())
+                self.redis_initialized = True
+                self.logger.info("Connected to Redis for token management")
+        except Exception as e:
+            self.logger.error(f"Failed to initialize Redis during service creation: {e}")
+            self.redis_initialized = False
+            self.redis_client = None
+
         # Define expected database tables for health checks
         auth_expected_tables = [
             'users', 'refresh_tokens', 'user_onboarding_progress',