eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-14 17:50:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(gateway): log resumed event separately from ready

Browse source
This commit is contained in:
alyssa 2024-11-10 04:53:10 +09:00
parent 7c913c7b15
commit dd7a9d4464
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
services/gateway/src/discord/shard_state.rs
View file

@ -18,8 +18,8 @@ pub fn new(redis: RedisPool) -> ShardStateManager {
impl ShardStateManager { impl ShardStateManager {
pub async fn handle_event(&self, shard_id: u32, event: Event) -> anyhow::Result<()> { pub async fn handle_event(&self, shard_id: u32, event: Event) -> anyhow::Result<()> {
match event { match event {
Event::Ready(_) => self.ready_or_resumed(shard_id).await, Event::Ready(_) => self.ready_or_resumed(shard_id, false).await,
Event::Resumed => self.ready_or_resumed(shard_id).await, Event::Resumed => self.ready_or_resumed(shard_id, true).await,
Event::GatewayClose(_) => self.socket_closed(shard_id).await, Event::GatewayClose(_) => self.socket_closed(shard_id).await,
Event::GatewayHeartbeat(_) => self.heartbeated(shard_id).await, Event::GatewayHeartbeat(_) => self.heartbeated(shard_id).await,
_ => Ok(()), _ => Ok(()),
@ -53,8 +53,12 @@ impl ShardStateManager {
Ok(()) Ok(())
} }
async fn ready_or_resumed(&self, shard_id: u32) -> anyhow::Result<()> { async fn ready_or_resumed(&self, shard_id: u32, resumed: bool) -> anyhow::Result<()> {
info!("shard {} ready", shard_id); info!(
"shard {} {}",
shard_id,
if resumed { "resumed" } else { "ready" }
);
let mut info = self.get_shard(shard_id).await?; let mut info = self.get_shard(shard_id).await?;
info.last_connection = chrono::offset::Utc::now().timestamp() as i32; info.last_connection = chrono::offset::Utc::now().timestamp() as i32;
info.up = true; info.up = true;