dalbodeule/chibot-chzzk-bot
1
0
Fork 0
mirror of https://github.com/dalbodeule/chibot-chzzk-bot.git synced 2025-08-07 21:01:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

asdf11

Browse Source
This commit is contained in:
dalbodeule
2024-08-14 20:28:17 +09:00
parent cb5bbf73c3
commit 0e3264a9f3
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/utils/DiscordGuildCache.kt
View File

@@ -30,6 +30,9 @@ object DiscordGuildCache {
}
private suspend fun fetchGuilds(beforeGuildId: String? = null, limit: Int = 100): List<DiscordGuildListAPI> {
if(DiscordRatelimits.isLimited()) {
delay(DiscordRatelimits.getRateReset().takeIf { it > 1000L } ?: 3000L)
}
val result = applicationHttpClient.get("https://discord.com/api/users/@me/guilds") {
headers {
append(HttpHeaders.Authorization, "Bot ${dotenv["DISCORD_TOKEN"]}")
@@ -52,9 +55,6 @@ object DiscordGuildCache {
private suspend fun fetchAllGuilds() {
var lastGuildId: String? = null
while (true) {
if(DiscordRatelimits.isLimited()) {
delay(DiscordRatelimits.getRateReset().takeIf { it > 1000L } ?: 3000L)
}
try {
val guilds = fetchGuilds(lastGuildId)
if (guilds.isEmpty()) {
@@ -63,7 +63,8 @@ object DiscordGuildCache {
guilds.forEach {
cache[it.id] = CachedGuilds(
Guild(it.id, it.name, it.icon, it.banner)
Guild(it.id, it.name, it.icon, it.banner),
Instant.now().plusSeconds(EXP_SECONDS),
)
}
lastGuildId = guilds.last().id
@@ -76,7 +77,7 @@ object DiscordGuildCache {
fun addGuild(guilds: Map<String, Guild>) {
cache.putAll(guilds.map {
it.key to CachedGuilds(it.value, Instant.now())
it.key to CachedGuilds(it.value, Instant.now().plusSeconds(EXP_SECONDS))
})
}
}