Stats cache

3 files changed, 80 insertions, 6 deletions

diff --git a/src/main/java/com/juick/service/InfoService.java b/src/main/java/com/juick/service/InfoService.java
new file mode 100644
index 00000000..fbf75b86
--- /dev/null
+++ b/src/main/java/com/juick/service/InfoService.java
@@ -0,0 +1,74 @@
+/*
+ * Copyright (C) 2008-2023, Juick
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package com.juick.service;
+
+import com.juick.www.api.xnodeinfo2.model.*;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.cache.annotation.Cacheable;
+import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.stereotype.Service;
+
+import javax.inject.Inject;
+import java.time.ZonedDateTime;
+import java.time.temporal.ChronoUnit;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.Map;
+
+@Service
+public class InfoService {
+    @Value("${ap_base_uri:http://localhost:8080/}")
+    private String baseUri;
+    @Inject
+    JdbcTemplate jdbcTemplate;
+
+    @Cacheable("stats")
+    public NodeInfo getCurrentNodeInfo(String version) {
+        NodeInfo nodeInfo = new NodeInfo();
+        nodeInfo.setVersion(version);
+        Server server = new Server();
+        server.setBaseUrl(baseUri);
+        server.setName("Juick");
+        server.setSoftware("Juick");
+        server.setVersion("2.x");
+        nodeInfo.setServer(server);
+        nodeInfo.setProtocols(Arrays.asList("xmpp", "activitypub", "smtp"));
+        Map<String, String> metadata = new HashMap<>();
+        metadata.put("email", "support@juick.com");
+        nodeInfo.setMetadata(metadata);
+        ServiceInfo serviceInfo = new ServiceInfo();
+        serviceInfo.setInbound(Arrays.asList("jabber", "mastodon", "email", "telegram"));
+        serviceInfo.setOutbound(Arrays.asList("jabber", "mastodon", "telegram", "twitter", "email", "rss"));
+        nodeInfo.setServices(serviceInfo);
+        UserStats userStats = new UserStats();
+        userStats.setTotal(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM users WHERE banned=0", Integer.class));
+        userStats.setActiveMonth(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM users WHERE banned=0 AND last_seen > ?",
+                Integer.class, ZonedDateTime.now().minus(1, ChronoUnit.MONTHS).toLocalDateTime()));
+        userStats.setActiveHalfyear(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM users WHERE banned=0 AND last_seen > ?",
+                Integer.class, ZonedDateTime.now().minus(6, ChronoUnit.MONTHS).toLocalDateTime()));
+        Usage usage = new Usage();
+        usage.setUsers(userStats);
+        usage.setLocalPosts(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM messages",
+                Integer.class));
+        usage.setLocalComments(jdbcTemplate.queryForObject("SELECT COUNT(*) FROM replies WHERE reply_uri IS NULL",
+                Integer.class));
+        nodeInfo.setUsage(usage);
+        return nodeInfo;
+    }
+
+}
diff --git a/src/main/java/com/juick/service/MessagesService.java b/src/main/java/com/juick/service/MessagesService.java
index 91788a3c..5574dbbf 100644
--- a/src/main/java/com/juick/service/MessagesService.java
+++ b/src/main/java/com/juick/service/MessagesService.java
@@ -34,10 +34,10 @@ import java.util.*;
  * Created by aalexeev on 11/13/16.
  */
 public interface MessagesService {
-    @CacheEvict(value = { "discover", "discussions", "messages", "replies" }, allEntries = true)
+    @CacheEvict(value = { "discover", "discussions", "messages", "replies", "stats" }, allEntries = true)
     int createMessage(int uid, String txt, String attachment, @NonNull Set<Tag> tags);
 
-    @CacheEvict(value = { "discover", "discussions", "messages", "replies" }, allEntries = true)
+    @CacheEvict(value = { "discover", "discussions", "messages", "replies", "stats" }, allEntries = true)
     int createReply(int mid, int rid, User user, String txt, String attachment);
     enum RecommendStatus {
         Error,
@@ -127,10 +127,10 @@ public interface MessagesService {
 
     boolean setMessagePrivacy(int mid);
 
-    @CacheEvict(value = { "discover", "discussions", "messages", "replies" }, allEntries = true)
+    @CacheEvict(value = { "discover", "discussions", "messages", "replies", "stats" }, allEntries = true)
     boolean deleteMessage(int uid, int mid);
 
-    @CacheEvict(value = { "discover", "discussions", "messages", "replies" }, allEntries = true)
+    @CacheEvict(value = { "discover", "discussions", "messages", "replies", "stats" }, allEntries = true)
     boolean deleteReply(int uid, int mid, int rid);
 
     List<Integer> getLastMessages(int hours);
@@ -155,7 +155,7 @@ public interface MessagesService {
 
     boolean replyExists(URI replyUri);
 
-    @CacheEvict(value = { "discover", "discussions", "messages", "replies" }, allEntries = true)
+    @CacheEvict(value = { "discover", "discussions", "messages", "replies", "stats" }, allEntries = true)
     boolean deleteReply(URI userUri, URI replyUri);
 
     String getMessageProperty(Integer mid, Integer rid, String key);
diff --git a/src/main/java/com/juick/service/UserService.java b/src/main/java/com/juick/service/UserService.java
index 31822b01..9313bba9 100644
--- a/src/main/java/com/juick/service/UserService.java
+++ b/src/main/java/com/juick/service/UserService.java
@@ -46,7 +46,7 @@ public interface UserService {
 
     String getSignUpHashByTelegramID(Long telegramId, String username);
 
-    @CacheEvict(value = { "usernames", "users_by_name" }, allEntries = true)
+    @CacheEvict(value = { "usernames", "users_by_name", "stats" }, allEntries = true)
     Optional<User> createUser(String username, String password) throws UsernameTakenException;
 
     Optional<User> getUserByUID(int uid);