Add cache max size and sweepers (#255)

wise-old-man · Jul 16, 2024 · 3ff9c44 · 3ff9c44
1 parent 907339e
commit 3ff9c44
Show file tree

Hide file tree

Showing 2 changed files with 56 additions and 2 deletions.
diff --git a/src/bot.ts b/src/bot.ts
@@ -6,7 +6,8 @@ import {
   TextChannel,
   GatewayIntentBits,
   ChannelType,
-  PermissionFlagsBits
+  PermissionFlagsBits,
+  Options
 } from 'discord.js';
 import config from './config';
 import * as router from './commands/router';
@@ -18,19 +19,55 @@ import {
   setupPatreonTrigger
 } from './patreon-trigger';
 
+const CACHED_ACTIVE_USER_IDS = new Set<string>(config.discord.cache.excludeUsers);
+const CACHED_ACTIVE_GUILD_IDS = new Set<string>(config.discord.cache.excludeGuilds);
+
 class Bot {
   client: Client;
 
   constructor() {
     this.client = new Client({
+      shards: 'auto',
       intents: [
         GatewayIntentBits.Guilds,
         GatewayIntentBits.GuildMessages,
         GatewayIntentBits.GuildMessageTyping,
         GatewayIntentBits.GuildMessageReactions,
         GatewayIntentBits.DirectMessageTyping
       ],
-      shards: 'auto'
+      makeCache: Options.cacheWithLimits({
+        // Disable caching for these
+        ThreadManager: { maxSize: 0 },
+        MessageManager: { maxSize: 0 },
+        PresenceManager: { maxSize: 0 },
+        VoiceStateManager: { maxSize: 0 },
+        GuildInviteManager: { maxSize: 0 },
+        GuildStickerManager: { maxSize: 0 },
+        ThreadMemberManager: { maxSize: 0 },
+        // Keep some cached items
+        UserManager: {
+          maxSize: 1000,
+          keepOverLimit: user => CACHED_ACTIVE_USER_IDS.has(user.id)
+        },
+        GuildMemberManager: {
+          maxSize: 200,
+          keepOverLimit: member => CACHED_ACTIVE_USER_IDS.has(member.user.id)
+        },
+        GuildEmojiManager: {
+          maxSize: 1,
+          keepOverLimit: i => CACHED_ACTIVE_GUILD_IDS.has(i.guild.id)
+        }
+      }),
+      sweepers: {
+        guildMembers: {
+          interval: 60 * 60,
+          filter: () => member => !CACHED_ACTIVE_USER_IDS.has(member.user.id)
+        },
+        users: {
+          interval: 60 * 60,
+          filter: () => user => !CACHED_ACTIVE_USER_IDS.has(user.id)
+        }
+      }
     });
   }
 

diff --git a/src/config.ts b/src/config.ts
@@ -32,6 +32,23 @@ export default {
       modLogs: env.DISCORD_DEV_MODSLOGS_CHANNEL_ID || '830199626630955039',
       patreonInfo: env.DISCORD_DEV_PATREON_INFO_CHANNEL_ID || '1173680059526152272',
       flaggedPlayerReviews: env.DISCORD_DEV_FLAGGED_REVIEWS_CHANNEL_ID || '1086637095415722169'
+    },
+    cache: {
+      excludeUsers: [
+        '719720369241718837' // The bot itself
+      ],
+      excludeGuilds: [
+        env.DISCORD_DEV_GUILD_ID ?? '',
+        '679454777708380161', // The WOM Discord server
+        '719722124301828138', // Psikoi's Emoji/Test server
+        '847787694213824512', // Rorro's Emoji/Test server
+        '1159473162342371458', // Rorro's Role server 1
+        '1159473499585400923', // Rorro's Role server 2
+        '1159473768859709450', // Rorro's Role server 3
+        '1159473857237897267', // Rorro's Role server 4
+        '1159473909708636161', // Rorro's Role server 5
+        '1159473951781683240' // Rorro's Role server 6
+      ]
     }
   }
 };