add caching to stats query

Author: Deepthi-Chand

api/schema/stats_schema.py

@@ -2,6 +2,7 @@
 
 import strawberry
 import strawberry_django
+from django.core.cache import cache
 from django.db.models import Count, Q
 from strawberry.types import Info
 
@@ -12,6 +13,9 @@
 from api.utils.graphql_telemetry import trace_resolver
 from authorization.models import User
 
+STATS_CACHE_KEY = "platform_stats"
+STATS_CACHE_TTL = 60 * 10  # 10 minutes
+
 
 @strawberry.type
 class StatsType:
@@ -31,20 +35,20 @@ class Query:
     @trace_resolver(name="stats", attributes={"component": "stats"})
     def stats(self, info: Info) -> StatsType:
         """Get platform statistics"""
+        cached = cache.get(STATS_CACHE_KEY)
+        if cached:
+            return StatsType(**cached)
+
         # Count total users
         total_users = User.objects.count()
 
         # Count published datasets
-        total_published_datasets = Dataset.objects.filter(
-            status=DatasetStatus.PUBLISHED
-        ).count()
+        total_published_datasets = Dataset.objects.filter(status=DatasetStatus.PUBLISHED).count()
 
         # Count publishers (organizations and individuals who have published datasets)
         # First, get organizations that have published datasets
         org_publishers = (
-            Organization.objects.filter(datasets__status=DatasetStatus.PUBLISHED)
-            .distinct()
-            .count()
+            Organization.objects.filter(datasets__status=DatasetStatus.PUBLISHED).distinct().count()
         )
 
         # Then, get individual users who have published datasets
@@ -61,15 +65,16 @@ def stats(self, info: Info) -> StatsType:
         total_publishers = org_publishers + individual_publishers
 
         # Count published usecases
-        total_published_usecases = UseCase.objects.filter(
-            status=UseCaseStatus.PUBLISHED
-        ).count()
-
-        return StatsType(
-            total_users=total_users,
-            total_published_datasets=total_published_datasets,
-            total_publishers=total_publishers,
-            total_organizations=org_publishers,
-            total_individuals=individual_publishers,
-            total_published_usecases=total_published_usecases,
-        )
+        total_published_usecases = UseCase.objects.filter(status=UseCaseStatus.PUBLISHED).count()
+
+        stats_data = {
+            "total_users": total_users,
+            "total_published_datasets": total_published_datasets,
+            "total_publishers": total_publishers,
+            "total_organizations": org_publishers,
+            "total_individuals": individual_publishers,
+            "total_published_usecases": total_published_usecases,
+        }
+        cache.set(STATS_CACHE_KEY, stats_data, timeout=STATS_CACHE_TTL)
+
+        return StatsType(**stats_data)