Fix Tests

Cache Updater

Author: surajkumar

api/src/main/java/com/javadiscord/jdi/core/api/DiscordResponseParser.java

@@ -7,6 +7,8 @@
 import com.javadiscord.jdi.internal.api.DiscordRequestDispatcher;
 import com.javadiscord.jdi.internal.api.DiscordResponse;
 import com.javadiscord.jdi.internal.api.DiscordResponseFuture;
+import com.javadiscord.jdi.internal.cache.Cache;
+import com.javadiscord.jdi.internal.utils.CacheUpdater;
 
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
@@ -17,9 +19,11 @@ public class DiscordResponseParser {
     private static final ObjectMapper OBJECT_MAPPER =
         JsonMapper.builder().addModule(new JavaTimeModule()).build();
     private final DiscordRequestDispatcher dispatcher;
+    private final CacheUpdater cacheUpdater;
 
-    public DiscordResponseParser(DiscordRequestDispatcher dispatcher) {
+    public DiscordResponseParser(DiscordRequestDispatcher dispatcher, Cache cache) {
         this.dispatcher = dispatcher;
+        this.cacheUpdater = new CacheUpdater(cache);
     }
 
     public <T> AsyncResponse<List<T>> callAndParseList(Class<T> clazz, DiscordRequest request) {
@@ -31,6 +35,7 @@ public <T> AsyncResponse<List<T>> callAndParseList(Class<T> clazz, DiscordReques
                     try {
                         List<T> resultList = parseResponseFromList(clazz, response.body());
                         asyncResponse.setResult(resultList);
+                        cacheUpdater.updateCache(resultList);
                     } catch (Exception e) {
                         asyncResponse.setException(e);
                     }
@@ -52,6 +57,7 @@ public <T> AsyncResponse<List<T>> callAndParseMap(String key, DiscordRequest req
                     try {
                         List<T> resultList = parseResponseFromMap(key, response.body());
                         asyncResponse.setResult(resultList);
+                        cacheUpdater.updateCache(resultList);
                     } catch (Exception e) {
                         asyncResponse.setException(e);
                     }
@@ -108,6 +114,7 @@ private <T> void success(
                     result = OBJECT_MAPPER.readValue(response.body(), type);
                 }
                 asyncResponse.setResult(result);
+                cacheUpdater.updateCache(result);
             } catch (JsonProcessingException e) {
                 asyncResponse.setException(e);
             }

api/src/main/java/com/javadiscord/jdi/internal/api/DiscordRequestDispatcher.java

@@ -11,8 +11,6 @@
 import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.stream.Stream;
 
-import com.javadiscord.jdi.RateLimit;
-
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 

api/src/main/java/com/javadiscord/jdi/RateLimit.java renamed to api/src/main/java/com/javadiscord/jdi/internal/api/RateLimit.java

@@ -1,4 +1,4 @@
-package com.javadiscord.jdi;
+package com.javadiscord.jdi.internal.api;
 
 public class RateLimit {
     private String bucket;

api/src/main/java/com/javadiscord/jdi/internal/utils/CacheUpdater.java

@@ -0,0 +1,63 @@
+package com.javadiscord.jdi.internal.utils;
+
+import java.lang.reflect.Field;
+import java.util.List;
+
+import com.javadiscord.jdi.internal.cache.Cache;
+
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+
+public class CacheUpdater {
+
+    private final Cache cache;
+
+    private static final Logger LOGGER = LogManager.getLogger(CacheUpdater.class);
+
+    public CacheUpdater(Cache cache) {
+        this.cache = cache;
+    }
+
+    public <T> void updateCache(T result) {
+        if (result == null) {
+            return;
+        }
+        try {
+            Field guildIdField = result.getClass().getDeclaredField("guildId");
+            Field idField = result.getClass().getDeclaredField("id");
+
+            long guildId = getLongFromField(guildIdField, result);
+            long id = getLongFromField(idField, result);
+
+            if (cache.getCacheForGuild(guildId) == null) {
+                LOGGER.trace(
+                    "Failed to cache result of type {} with guildId of {}",
+                    result.getClass().getName(), guildId
+                );
+            } else {
+                cache.getCacheForGuild(guildId).add(id, result);
+            }
+
+        } catch (IllegalAccessException | NoSuchFieldException | NumberFormatException e) {
+            LOGGER.trace(
+                "Failed to cache result of type {}, cause: {}",
+                result.getClass().getName(), e.getMessage()
+            );
+        }
+    }
+
+    public <T> void updateCache(List<T> resultList) {
+        resultList.forEach(this::updateCache);
+    }
+
+    private <T> long getLongFromField(
+        Field field,
+        T result
+    ) throws IllegalAccessException, NumberFormatException {
+        field.setAccessible(true);
+        if (field.getType() == String.class) {
+            return Long.parseLong((String) field.get(result));
+        }
+        return (long) field.get(result);
+    }
+}

api/src/test/integration/helpers/LiveDiscordHelper.java

@@ -17,7 +17,7 @@ public class LiveDiscordHelper {
     private static final CountDownLatch STARTUP_LATCH = new CountDownLatch(1);
     private static Guild guild;
 
-    private static class TestListener implements EventListener {
+    public static class TestListener implements EventListener {
 
         @Override
         public void onGuildCreate(Guild guild) {

core/src/main/java/com/javadiscord/jdi/core/Guild.java

@@ -34,7 +34,7 @@ public Guild(GuildModel guild, Cache cache, Discord discord) {
         long guildId = guild.id();
 
         DiscordResponseParser discordResponseParser =
-            new DiscordResponseParser(discord.getDiscordRequestDispatcher());
+            new DiscordResponseParser(discord.getDiscordRequestDispatcher(), cache);
 
         this.applicationRequest = new ApplicationRequest(discordResponseParser, guildId);
         this.applicationRoleConnectionMetaRequest =