nki/youmubot
1
0
Fork 0
mirror of https://github.com/natsukagami/youmubot.git synced 2025-04-19 00:38:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Don't retry the members endpoint too much

Browse source
This commit is contained in:
Natsu Kagami 2024-03-06 16:38:50 +01:00
parent 4ba64a466d
commit d23c86fb99
Signed by: nki
GPG key ID: 55A032EB38B49ADB
1 changed files with 21 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
youmubot-prelude/src/member_cache.rs
View file

@ -1,3 +1,4 @@
use anyhow::bail;
use chrono::{DateTime, Utc};
use serenity::model::{
guild::Member,
@ -9,6 +10,7 @@ use std::sync::Arc;
use tokio::sync::Mutex;
const VALID_CACHE_SECONDS: i64 = 15 * 60; // 15 minutes
const INVALID_CACHE_SECONDS: i64 = 2 * 60; // 2 minutes
type Map<K, V> = Mutex<HashMap<K, V>>;
@ -16,7 +18,7 @@ type Map<K, V> = Mutex<HashMap<K, V>>;
#[derive(Debug, Default)]
pub struct MemberCache {
per_user: Map<(UserId, GuildId), Expiring<Option<Member>>>,
per_guild: Map<GuildId, Expiring<Arc<[Member]>>>,
per_guild: Map<GuildId, Expiring<Option<Arc<[Member]>>>>,
}
#[derive(Debug)]
@ -90,21 +92,32 @@ impl MemberCache {
// Check cache
if let Entry::Occupied(oe) = &entry {
if oe.get().timeout > now {
return Ok(oe.get().value.clone());
return match &oe.get().value {
Some(v) => Ok(v.clone()),
None => bail!("guild members for {} unavailable", guild_id),
};
}
}
// query
eprintln!("querying members of {}", guild_id);
let members: Arc<[Member]> = guild_id
let members: Option<Arc<[Member]>> = guild_id
.members(cache_http.http(), None, None)
.await?
.into();
Ok(entry
.await
.ok()
.map(|v| v.into());
match &entry
.or_insert(Expiring::new(
members.clone(),
now + chrono::Duration::seconds(VALID_CACHE_SECONDS),
now + chrono::Duration::seconds(if members.is_some() {
VALID_CACHE_SECONDS
} else {
INVALID_CACHE_SECONDS
}),
))
.value
.clone())
{
Some(v) => Ok(v.clone()),
None => bail!("guild members for {} unavailable", guild_id),
}
}
}