Merge pull request #31 from tahzeer/1.0

Implement role-based cache key builder

Author: venky-ganapathy

iam-staff-portal-api/src/iam_staff_portal_api/cache.py

@@ -4,3 +4,17 @@
 
 def init_cache():
     FastAPICache.init(InMemoryBackend(), prefix="iam-staff-cache")
+
+
+def role_cache_key(func, namespace: str, *args, **kwargs):
+    """
+    Build a cache key scoped only by role mnemonic.
+    """
+    call_args = kwargs.get("args") or args
+    call_kwargs = kwargs.get("kwargs") or {}
+
+    role_mnemonic = call_kwargs.get("role_mnemonic")
+    if role_mnemonic is None and len(call_args) > 1:
+        role_mnemonic = call_args[1]
+
+    return f"{namespace}:{role_mnemonic}"

iam-staff-portal-api/src/iam_staff_portal_api/controllers/user_access_controller.py

@@ -9,6 +9,7 @@
 from sqlalchemy import select
 from sqlalchemy.ext.asyncio import async_sessionmaker
 
+from ..cache import role_cache_key
 from ..config import Settings
 from ..models import (
     StaffApplicationPermission,
@@ -184,7 +185,7 @@ async def get_permissions_for_roles(
 
         return PermissionsResponse(permissions=sorted(set(permissions)))
 
-    @cache(expire=_config.cache_expire_seconds)
+    @cache(expire=_config.cache_expire_seconds, key_builder=role_cache_key)
     async def get_permission_mnemonics_for_role(
         self,
         role_mnemonic: str,