Add role-based filtering and imporve code

services/demo_session/app/api/demo_operations.py

@@ -8,7 +8,7 @@ import jwt
 
 from app.api.schemas import DemoSessionResponse, DemoSessionStats
 from app.services import DemoSessionManager, DemoCleanupService
-from app.core import get_db, get_redis, RedisClient
+from app.core import get_db, get_redis, DemoRedisWrapper
 from sqlalchemy.ext.asyncio import AsyncSession
 from shared.routing import RouteBuilder
 
@@ -25,7 +25,7 @@ route_builder = RouteBuilder('demo')
 async def extend_demo_session(
     session_id: str = Path(...),
     db: AsyncSession = Depends(get_db),
-    redis: RedisClient = Depends(get_redis)
+    redis: DemoRedisWrapper = Depends(get_redis)
 ):
     """Extend demo session expiration (BUSINESS OPERATION)"""
     try:
@@ -67,7 +67,7 @@ async def extend_demo_session(
 )
 async def get_demo_stats(
     db: AsyncSession = Depends(get_db),
-    redis: RedisClient = Depends(get_redis)
+    redis: DemoRedisWrapper = Depends(get_redis)
 ):
     """Get demo session statistics (BUSINESS OPERATION)"""
     session_manager = DemoSessionManager(db, redis)
@@ -81,7 +81,7 @@ async def get_demo_stats(
 )
 async def run_cleanup(
     db: AsyncSession = Depends(get_db),
-    redis: RedisClient = Depends(get_redis)
+    redis: DemoRedisWrapper = Depends(get_redis)
 ):
     """Manually trigger session cleanup (BUSINESS OPERATION - Internal endpoint for CronJob)"""
     cleanup_service = DemoCleanupService(db, redis)

services/demo_session/app/api/demo_sessions.py

@@ -10,7 +10,8 @@ import jwt
 
 from app.api.schemas import DemoSessionCreate, DemoSessionResponse
 from app.services import DemoSessionManager
-from app.core import get_db, get_redis, RedisClient
+from app.core import get_db
+from app.core.redis_wrapper import get_redis, DemoRedisWrapper
 from sqlalchemy.ext.asyncio import AsyncSession
 from shared.routing import RouteBuilder
 
@@ -64,7 +65,7 @@ async def create_demo_session(
     request: DemoSessionCreate,
     http_request: Request,
     db: AsyncSession = Depends(get_db),
-    redis: RedisClient = Depends(get_redis)
+    redis: DemoRedisWrapper = Depends(get_redis)
 ):
     """Create a new isolated demo session (ATOMIC)"""
     logger.info("Creating demo session", demo_account_type=request.demo_account_type)
@@ -130,7 +131,7 @@ async def create_demo_session(
 async def get_session_info(
     session_id: str = Path(...),
     db: AsyncSession = Depends(get_db),
-    redis: RedisClient = Depends(get_redis)
+    redis: DemoRedisWrapper = Depends(get_redis)
 ):
     """Get demo session information (ATOMIC READ)"""
     session_manager = DemoSessionManager(db, redis)
@@ -149,7 +150,7 @@ async def get_session_info(
 async def get_session_status(
     session_id: str = Path(...),
     db: AsyncSession = Depends(get_db),
-    redis: RedisClient = Depends(get_redis)
+    redis: DemoRedisWrapper = Depends(get_redis)
 ):
     """
     Get demo session provisioning status
@@ -173,7 +174,7 @@ async def get_session_status(
 async def retry_session_cloning(
     session_id: str = Path(...),
     db: AsyncSession = Depends(get_db),
-    redis: RedisClient = Depends(get_redis)
+    redis: DemoRedisWrapper = Depends(get_redis)
 ):
     """
     Retry failed cloning operations
@@ -204,7 +205,7 @@ async def retry_session_cloning(
 async def destroy_demo_session(
     session_id: str = Path(...),
     db: AsyncSession = Depends(get_db),
-    redis: RedisClient = Depends(get_redis)
+    redis: DemoRedisWrapper = Depends(get_redis)
 ):
     """Destroy demo session and cleanup resources (ATOMIC DELETE)"""
     try:
@@ -225,7 +226,7 @@ async def destroy_demo_session(
 async def destroy_demo_session_post(
     session_id: str = Path(...),
     db: AsyncSession = Depends(get_db),
-    redis: RedisClient = Depends(get_redis)
+    redis: DemoRedisWrapper = Depends(get_redis)
 ):
     """Destroy demo session via POST (for frontend compatibility)"""
     try: