🎨 follow count增加缓存

FishPiOffical · Dec 21, 2024 · eb57e01 · eb57e01
1 parent fa7f025
commit eb57e01
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 2 deletions.
diff --git a/src/main/java/org/b3log/symphony/cache/FollowingCountCache.java b/src/main/java/org/b3log/symphony/cache/FollowingCountCache.java
@@ -0,0 +1,19 @@
+package org.b3log.symphony.cache;
+
+import org.b3log.latke.ioc.Singleton;
+import org.json.JSONObject;
+
+import java.util.Collections;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+@Singleton
+public class FollowingCountCache {
+
+    public static final Map<String, JSONObject> COUNT_CACHE = Collections.synchronizedMap(new LinkedHashMap<>() {
+        @Override
+        protected boolean removeEldestEntry(Map.Entry eldest) {
+            return size() > 2000;
+        }
+    });
+}
diff --git a/src/main/java/org/b3log/symphony/service/FollowQueryService.java b/src/main/java/org/b3log/symphony/service/FollowQueryService.java
@@ -29,6 +29,7 @@
 import org.b3log.latke.repository.*;
 import org.b3log.latke.service.annotation.Service;
 import org.b3log.latke.util.Stopwatchs;
+import org.b3log.symphony.cache.FollowingCountCache;
 import org.b3log.symphony.model.Follow;
 import org.b3log.symphony.repository.ArticleRepository;
 import org.b3log.symphony.repository.FollowRepository;
@@ -395,9 +396,23 @@ public long getFollowingCount(final String followerId, final int followingType)
             final List<Filter> filters = new ArrayList<>();
             filters.add(new PropertyFilter(Follow.FOLLOWER_ID, FilterOperator.EQUAL, followerId));
             filters.add(new PropertyFilter(Follow.FOLLOWING_TYPE, FilterOperator.EQUAL, followingType));
+            JSONObject cacheJSON = new JSONObject();
+            cacheJSON.put(Follow.FOLLOWER_ID, followerId);
+            cacheJSON.put(Follow.FOLLOWING_TYPE, followingType);
+            JSONObject result = FollowingCountCache.COUNT_CACHE.get(cacheJSON.toString());
+            if (result != null) {
+                if (result.optLong("time") > System.currentTimeMillis() - FIVE_MINUTES_IN_MILLIS) {
+                    return result.optLong("count");
+                }
+            }
             final Query query = new Query().setFilter(new CompositeFilter(CompositeFilterOperator.AND, filters));
             try {
-                return followRepository.count(query);
+                long count = followRepository.count(query);
+                JSONObject json = new JSONObject();
+                json.put("count", count);
+                json.put("time", System.currentTimeMillis());
+                FollowingCountCache.COUNT_CACHE.put(cacheJSON.toString(), json);
+                return count;
             } catch (final RepositoryException e) {
                 LOGGER.log(Level.ERROR, "Counts following count failed", e);
                 return 0;
@@ -407,6 +422,8 @@ public long getFollowingCount(final String followerId, final int followingType)
         }
     }
 
+    private static final long FIVE_MINUTES_IN_MILLIS = 1000 * 60 * 5;
+
     /**
      * Gets the follower count of a following specified by the given following id and following type.
      *
@@ -418,9 +435,23 @@ public long getFollowerCount(final String followingId, final int followingType)
         final List<Filter> filters = new ArrayList<>();
         filters.add(new PropertyFilter(Follow.FOLLOWING_ID, FilterOperator.EQUAL, followingId));
         filters.add(new PropertyFilter(Follow.FOLLOWING_TYPE, FilterOperator.EQUAL, followingType));
+        JSONObject cacheJSON = new JSONObject();
+        cacheJSON.put(Follow.FOLLOWING_ID, followingId);
+        cacheJSON.put(Follow.FOLLOWING_TYPE, followingType);
+        JSONObject result = FollowingCountCache.COUNT_CACHE.get(cacheJSON.toString());
+        if (result != null) {
+            if (result.optLong("time") > System.currentTimeMillis() - FIVE_MINUTES_IN_MILLIS) {
+                return result.optLong("count");
+            }
+        }
         final Query query = new Query().setFilter(new CompositeFilter(CompositeFilterOperator.AND, filters));
         try {
-            return followRepository.count(query);
+            long count = followRepository.count(query);
+            JSONObject json = new JSONObject();
+            json.put("count", count);
+            json.put("time", System.currentTimeMillis());
+            FollowingCountCache.COUNT_CACHE.put(cacheJSON.toString(), json);
+            return count;
         } catch (final RepositoryException e) {
             LOGGER.log(Level.ERROR, "Counts follower count failed", e);
             return 0;