tensamin/omega
Watch
6
0
Fork
You've already forked omega
 0
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

(fix): fix connection issues

Browse source
This commit is contained in:
Alois 2026-07-28 01:02:05 +02:00
commit be0cad295d
2 changed files with 11 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/transport/omikron_connection.rs
View file

@ -245,16 +245,20 @@ impl OmikronConnection {
}
pub async fn close(self: Arc<Self>) {
log_in!(
self.get_omikron_id().await.unwrap_or(0),
self.id as i64,
PrintType::Omega,
"Omikron connection Closed"
);
if let Some(sender) = self.sender.lock().await.take() {
sender.close();
}
}
async fn cleanup(self: Arc<Self>) {
if self.id != 0 {
log_in!(self.id as i64, PrintType::Omega, "Omikron disconnected");
omikron_manager::remove_omikron(self.id as i64).await;
crate::sql::user_online_tracker::untrack_omikron(self.id as i64).await;
if omikron_manager::remove_omikron(self.id as i64, &self).await {
crate::sql::user_online_tracker::untrack_omikron(self.id as i64).await;
}
}
if let Some(handle) = self.cleanup_handle.lock().unwrap().take() {
handle.abort();

6
src/transport/omikron_manager.rs
View file

@ -22,8 +22,10 @@ pub async fn add_omikron(conn: Arc<OmikronConnection>) {
}
}
pub async fn remove_omikron(omikron_id: i64) {
OMIKRON_CONNECTIONS.remove(&omikron_id);
pub async fn remove_omikron(omikron_id: i64, connection: &Arc<OmikronConnection>) -> bool {
OMIKRON_CONNECTIONS
.remove_if(&omikron_id, |_, current| Arc::ptr_eq(current, connection))
.is_some()
}
pub fn get_connected_omikron(omikron_id: i64) -> Option<Arc<OmikronConnection>> {