Rewrite top.gg integration to update every hour

Author: TechnoVisionDev

src/main/java/technobot/TechnoBot.java

@@ -7,8 +7,6 @@
 import net.dv8tion.jda.api.requests.GatewayIntent;
 import net.dv8tion.jda.api.sharding.DefaultShardManagerBuilder;
 import net.dv8tion.jda.api.sharding.ShardManager;
-import net.dv8tion.jda.api.utils.ChunkingFilter;
-import net.dv8tion.jda.api.utils.MemberCachePolicy;
 import okhttp3.OkHttpClient;
 import org.jetbrains.annotations.NotNull;
 import technobot.commands.CommandRegistry;

src/main/java/technobot/listeners/GuildListener.java

@@ -1,5 +1,6 @@
 package technobot.listeners;
 
+import net.dv8tion.jda.api.JDA;
 import net.dv8tion.jda.api.entities.Guild;
 import net.dv8tion.jda.api.entities.Member;
 import net.dv8tion.jda.api.entities.Role;
@@ -15,6 +16,8 @@
 import technobot.data.GuildData;
 
 import java.io.IOException;
+import java.util.Timer;
+import java.util.TimerTask;
 
 /**
  * Listens for guild events.
@@ -24,10 +27,14 @@
  */
 public class GuildListener extends ListenerAdapter {
 
+    private static final long STATS_UPDATE_PERIOD = 3600000; //1 hour in millis
+
     private final TechnoBot bot;
+    private final Timer timer;
 
     public GuildListener(TechnoBot bot) {
         this.bot = bot;
+        this.timer = new Timer();
     }
 
     /**
@@ -53,30 +60,9 @@ public void onGuildJoin(@NotNull GuildJoinEvent event) {
     public void onReady(@NotNull ReadyEvent event) {
         // Get Top.GG bearer token from .env
         String TOPGG_TOKEN = bot.config.get("TOPGG_TOKEN");
-        if (TOPGG_TOKEN == null) return;
-
-        // Build post request to update server count
-        RequestBody formBody = new FormBody.Builder()
-                .add("server_count", String.valueOf(event.getGuildTotalCount()))
-                .add("shard_count", String.valueOf(event.getJDA().getShardManager().getShardsTotal()))
-                .build();
-        String url = "https://top.gg/api/bots/" + event.getJDA().getSelfUser().getId() + "/stats";
-        Request request = new Request.Builder()
-                .url(url)
-                .post(formBody)
-                .addHeader("Authorization", "Bearer " + TOPGG_TOKEN)
-                .build();
-        Call call = bot.httpClient.newCall(request);
-
-        // Execute post request
-        try {
-            Response response = call.execute();
-            if (response.code() != 200) {
-                System.out.println("Warning: Unable to update Top.GG statistics! [code="+response.code()+"]");
-            }
-            response.close();
-        } catch (IOException e) {
-            System.out.println("ERROR IOException: Unable to update Top.GG statistics!");
+        if (TOPGG_TOKEN != null) {
+            // Update bot statistics at a fixed rate
+            timer.scheduleAtFixedRate(new StatUpdateTask(event.getJDA(), TOPGG_TOKEN), 0, STATS_UPDATE_PERIOD);
         }
     }
 
@@ -104,4 +90,49 @@ public void onGuildMemberJoin(@NotNull GuildMemberJoinEvent event) {
             }
         }
     }
+
+    /**
+     * Timer task that updates bot statistics to Top.GG
+     * Use in OnReady event every hour to send post request.
+     */
+    private class StatUpdateTask extends TimerTask {
+
+        private final JDA jda;
+        private final String TOPGG_TOKEN;
+
+        public StatUpdateTask(JDA jda, String TOPGG_TOKEN) {
+            this.jda = jda;
+            this.TOPGG_TOKEN = TOPGG_TOKEN;
+        }
+
+        /**
+         * Sends a post request to top.gg to update stats.
+         */
+        @Override
+        public void run() {
+            // Build post request to update server count
+            RequestBody formBody = new FormBody.Builder()
+                    .add("server_count", String.valueOf(jda.getGuilds().size()))
+                    .add("shard_count", String.valueOf(jda.getShardManager().getShardsTotal()))
+                    .build();
+            String url = "https://top.gg/api/bots/" + jda.getSelfUser().getId() + "/stats";
+            Request request = new Request.Builder()
+                    .url(url)
+                    .post(formBody)
+                    .addHeader("Authorization", "Bearer " + TOPGG_TOKEN)
+                    .build();
+            Call call = bot.httpClient.newCall(request);
+
+            // Execute post request
+            try {
+                Response response = call.execute();
+                if (response.code() != 200) {
+                    System.out.println("Warning: Unable to update Top.GG statistics! [code="+response.code()+"]");
+                }
+                response.close();
+            } catch (IOException e) {
+                System.out.println("ERROR IOException: Unable to update Top.GG statistics!");
+            }
+        }
+    }
 }