Cache some values

Author: acrylic-style

src/main/kotlin/net/azisaba/guildchatdiscord/Main.kt

@@ -19,7 +19,6 @@ import net.azisaba.interchat.api.network.Protocol
 import net.azisaba.interchat.api.network.protocol.GuildMessagePacket
 import org.slf4j.LoggerFactory
 import java.io.File
-import java.util.UUID
 
 private val logger = LoggerFactory.getLogger("GuildChatDiscord")
 
@@ -48,32 +47,13 @@ suspend fun main() {
 
     client.on<MessageCreateEvent> {
         if (message.author?.isBot != false) return@on
-        InterChatDiscord.asyncExecutor.execute {
-            val minecraftUuid = DatabaseManager.query("SELECT `minecraft_uuid` FROM `users` WHERE `discord_id` = ?") {
-                it.setLong(1, message.author!!.id.value.toLong())
-                it.executeQuery().use { rs ->
-                    if (rs.next()) {
-                        UUID.fromString(rs.getString("minecraft_uuid"))
-                    } else {
-                        null
-                    }
-                }
-            } ?: return@execute
-            val guildId = DatabaseManager.query("SELECT `guild_id` FROM `channels` WHERE `channel_id` = ?") {
-                it.setLong(1, message.channelId.value.toLong())
-                it.executeQuery().use { rs ->
-                    if (rs.next()) {
-                        rs.getLong("guild_id")
-                    } else {
-                        null
-                    }
-                }
-            } ?: return@execute
-            val isMember = InterChatDiscord.guildManager.getMember(guildId, minecraftUuid).exceptionally { null }.join() != null
-            if (!isMember) return@execute
-            val packet = GuildMessagePacket(guildId, "Discord", minecraftUuid, message.content, null)
-            JedisBoxProvider.get().pubSubHandler.publish(Protocol.GUILD_MESSAGE.name, packet)
-        }
+        val guildId = DatabaseManager.getGuildIdByChannelId(message.channelId.value.toLong()) ?: return@on
+        val minecraftUuid = DatabaseManager.getMinecraftUUIDByDiscordId(message.author!!.id.value.toLong()) ?: return@on
+        // return if the author is not member of the guild
+        InterChatDiscord.guildManager.getMember(guildId, minecraftUuid).exceptionally { null }.join() ?: return@on
+        // send packet
+        val packet = GuildMessagePacket(guildId, "Discord", minecraftUuid, message.content, null)
+        JedisBoxProvider.get().pubSubHandler.publish(Protocol.GUILD_MESSAGE.name, packet)
     }
 
     client.on<ApplicationCommandInteractionCreateEvent> {

src/main/kotlin/net/azisaba/guildchatdiscord/util/DatabaseManager.kt

@@ -5,6 +5,7 @@ import net.azisaba.interchat.api.util.Functions
 import net.azisaba.interchat.api.util.QueryExecutor
 import org.intellij.lang.annotations.Language
 import java.sql.PreparedStatement
+import java.util.UUID
 
 object DatabaseManager {
     val interChatQueryExecutor = QueryExecutor { sql, action -> queryInterChat(sql) { stmt -> action.accept(stmt) } }
@@ -46,6 +47,32 @@ object DatabaseManager {
         }
     }.toKotlin()
 
+    val getGuildIdByChannelId: (Long) -> Long? = Functions.memoize<Long, Long>(1000 * 60) { channelId ->
+        query("SELECT `guild_id` FROM `channels` WHERE `channel_id` = ?") {
+            it.setLong(1, channelId)
+            it.executeQuery().use { rs ->
+                if (rs.next()) {
+                    rs.getLong("guild_id")
+                } else {
+                    null
+                }
+            }
+        }
+    }.toKotlin()
+
+    val getMinecraftUUIDByDiscordId: (Long) -> UUID? = Functions.memoize<Long, UUID>(1000 * 30) { discordId ->
+        query("SELECT `minecraft_uuid` FROM `users` WHERE `discord_id` = ?") {
+            it.setLong(1, discordId)
+            it.executeQuery().use { rs ->
+                if (rs.next()) {
+                    UUID.fromString(rs.getString("minecraft_uuid"))
+                } else {
+                    null
+                }
+            }
+        }
+    }.toKotlin()
+
     inline fun <R> query(@Language("SQL") sql: String, block: (PreparedStatement) -> R): R =
         dataSource.connection.use { connection ->
             connection.prepareStatement(sql).use(block)