eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-15 18:20:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(gateway): add reconnect timestamp to shard state

Browse source
This commit is contained in:
asleepyskye 2025-08-27 22:31:23 -04:00
parent 2fc5f2a9d9
commit 1378379e14
4 changed files with 18 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
crates/gateway/src/discord/gateway.rs
View file

@ -6,7 +6,7 @@ use std::sync::Arc;
use tokio::sync::mpsc::Sender;
use tracing::{error, info, warn};
use twilight_gateway::{
create_iterator, ConfigBuilder, Event, EventTypeFlags, Message, Shard, ShardId,
create_iterator, ConfigBuilder, Event, EventTypeFlags, Message, Shard, ShardId, CloseFrame
};
use twilight_model::gateway::{
payload::outgoing::update_presence::UpdatePresencePayload,
@ -116,7 +116,11 @@ pub async fn runner(
let raw_event = match item {
Ok(evt) => match evt {
Message::Close(frame) => {
let mut state_event = ShardStateEvent::Closed;
let close_code = if let Some(close) = frame {
if close == CloseFrame::RESUME {
state_event = ShardStateEvent::Reconnect;
}
close.code.to_string()
} else {
"unknown".to_string()
@ -132,7 +136,7 @@ pub async fn runner(
.increment(1);
if let Err(error) =
tx_state.try_send((shard.id(), ShardStateEvent::Closed, None, None))
tx_state.try_send((shard.id(), state_event, None, None))
{
error!("failed to update shard state for socket closure: {error}");
}