mat/azalea
1
2
Fork 0
mirror of https://github.com/mat-1/azalea.git synced 2025-08-02 06:16:04 +00:00
Code Issues Projects Releases Packages Activity

don't keep firing disconnectevent

Browse source
This commit is contained in:
mat 2023-02-21 22:21:43 -06:00
parent 48640ca303
commit c52cee3a69
1 changed files with 24 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
azalea-client/src/disconnect.rs
View file

@ -2,11 +2,15 @@
use azalea_ecs::{
app::{App, CoreStage, Plugin},
component::{BevyComponent, Component},
entity::Entity,
event::{EventReader, EventWriter},
query::Changed,
schedule::IntoSystemDescriptor,
system::{Commands, Query},
AppTickExt,
};
use derive_more::Deref;
use crate::{client::JoinedClientBundle, LocalPlayer};
@ -15,6 +19,9 @@ impl Plugin for DisconnectPlugin {
fn build(&self, app: &mut App) {
app.add_event::<DisconnectEvent>()
.add_system_to_stage(CoreStage::PostUpdate, handle_disconnect)
.add_tick_system(
update_read_packets_task_running_component.before(disconnect_on_read_packets_ended),
)
.add_tick_system(disconnect_on_read_packets_ended);
}
}
@ -32,12 +39,26 @@ pub fn handle_disconnect(mut commands: Commands, mut events: EventReader<Disconn
}
}
fn disconnect_on_read_packets_ended(
#[derive(Component, Clone, Copy, Debug, Deref)]
pub struct ReadPacketsTaskRunning(bool);
fn update_read_packets_task_running_component(
mut commands: Commands,
local_player: Query<(Entity, &LocalPlayer)>,
mut disconnect_events: EventWriter<DisconnectEvent>,
) {
for (entity, local_player) in &local_player {
if local_player.read_packets_task.is_finished() {
let running = !local_player.read_packets_task.is_finished();
commands
.entity(entity)
.insert(ReadPacketsTaskRunning(running));
}
}
fn disconnect_on_read_packets_ended(
local_player: Query<(Entity, &ReadPacketsTaskRunning), Changed<ReadPacketsTaskRunning>>,
mut disconnect_events: EventWriter<DisconnectEvent>,
) {
for (entity, &read_packets_task_running) in &local_player {
if *read_packets_task_running {
disconnect_events.send(DisconnectEvent { entity });
}
}