Merge pull request #156 from PROCOLLAB-github/feature/caching-views-and-likes

Caching for likes & views

Author: yakser

core/constants.py

@@ -1,2 +1,4 @@
 ONE_DAY_IN_SECONDS = 60 * 60 * 24
 ONE_WEEK_IN_SECONDS = ONE_DAY_IN_SECONDS * 7
+VIEWS_CACHING_TIMEOUT = 60 * 15
+LIKES_CACHING_TIMEOUT = 60 * 15

core/services.py

@@ -1,6 +1,8 @@
 from django.contrib.auth import get_user_model
 from django.contrib.contenttypes.models import ContentType
+from django.core.cache import cache
 
+from core.constants import VIEWS_CACHING_TIMEOUT, LIKES_CACHING_TIMEOUT
 from core.models import Like, View, Link
 
 User = get_user_model()
@@ -34,9 +36,15 @@ def get_fans(obj):
 
 def get_likes_count(obj):
     obj_type = ContentType.objects.get_for_model(obj)
-    return User.objects.filter(
-        likes__content_type=obj_type, likes__object_id=obj.id
-    ).count()
+    # cache this
+    likes_count = cache.get(f"likes_count_{obj_type}_{obj.id}")
+    if likes_count is None:
+        likes_count = User.objects.filter(
+            likes__content_type=obj_type, likes__object_id=obj.id
+        ).count()
+        # cache for LIKES_CACHING_TIMEOUT seconds
+        cache.set(f"views_count_{obj_type}_{obj.id}", likes_count, LIKES_CACHING_TIMEOUT)
+    return likes_count
 
 
 def set_like(obj, user, is_liked):
@@ -76,9 +84,16 @@ def get_viewers(obj):
 
 def get_views_count(obj):
     obj_type = ContentType.objects.get_for_model(obj)
-    return User.objects.filter(
-        views__content_type=obj_type, views__object_id=obj.id
-    ).count()
+    # cache this
+    views_count = cache.get(f"views_count_{obj_type}_{obj.id}")
+    if views_count is None:
+        views_count = User.objects.filter(
+            views__content_type=obj_type, views__object_id=obj.id
+        ).count()
+        # cache for VIEWS_CACHING_TIMEOUT seconds
+        cache.set(f"views_count_{obj_type}_{obj.id}", views_count, VIEWS_CACHING_TIMEOUT)
+
+    return views_count
 
 
 def set_viewed(obj, user, is_viewed):