Store both user and member data on cache for member operations

kordlib · Aug 11, 2024 · b4fff47 · b4fff47
1 parent eb98ccc
commit b4fff47
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/core/src/commonMain/kotlin/supplier/StoreEntitySupplier.kt b/core/src/commonMain/kotlin/supplier/StoreEntitySupplier.kt
@@ -60,9 +60,11 @@ public class StoreEntitySupplier(
 
     }
 
-    override suspend fun getMemberOrNull(guildId: Snowflake, userId: Snowflake): Member? {
-        return storeAndReturn(supplier.getMemberOrNull(guildId, userId)) { it.memberData }
-    }
+    override suspend fun getMemberOrNull(guildId: Snowflake, userId: Snowflake): Member? =
+        supplier.getMemberOrNull(guildId, userId)?.also { member ->
+            cache.put(member.data)
+            cache.put(member.memberData)
+        }
 
     override suspend fun getMessageOrNull(channelId: Snowflake, messageId: Snowflake): Message? {
         return storeAndReturn(supplier.getMessageOrNull(channelId, messageId)) { it.data }
@@ -104,9 +106,11 @@ public class StoreEntitySupplier(
         return storeOnEach(supplier.getGuildBans(guildId, limit)) { it.data }
     }
 
-    override fun getGuildMembers(guildId: Snowflake, limit: Int?): Flow<Member> {
-        return storeOnEach(supplier.getGuildMembers(guildId, limit)) { it.memberData }
-    }
+    override fun getGuildMembers(guildId: Snowflake, limit: Int?): Flow<Member> =
+        supplier.getGuildMembers(guildId, limit).onEach { member ->
+            cache.put(member.data)
+            cache.put(member.memberData)
+        }
 
     override fun getGuildVoiceRegions(guildId: Snowflake): Flow<Region> {
         return storeOnEach(supplier.getGuildVoiceRegions(guildId)) { it.data }