shiroyashik/sculptor
1
0
Fork 0
mirror of https://github.com/shiroyashik/sculptor.git synced 2025-12-06 04:51:13 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Temporary fix

Browse source
This commit is contained in:
Shiroyasha 2024-07-24 23:57:25 +03:00
parent 584983810b
commit f402d3d441
Signed by: shiroyashik
GPG key ID: E4953D3940D7860A
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/api/figura/websocket.rs
View file

@ -1,4 +1,4 @@
use std::sync::Arc; use std::{sync::Arc, thread};
use axum::{ use axum::{
extract::{ extract::{
@ -99,7 +99,7 @@ async fn handle_socket(mut socket: WebSocket, state: AppState) {
}; };
}, },
None => { None => {
warn!("[WebSocket] Authenticaton error! Connection terminated!"); warn!("[WebSocket] Authentication error! Connection terminated!");
debug!("[WebSocket] Tried to log in with {token}"); // Tried to log in with token: {token} debug!("[WebSocket] Tried to log in with {token}"); // Tried to log in with token: {token}
break; break;
}, },
@ -173,9 +173,9 @@ async fn handle_socket(mut socket: WebSocket, state: AppState) {
} }
// Closing connection // Closing connection
if let Some(u) = owner { if let Some(u) = owner {
state.session.remove(&u.uuid); // state.session.remove(&u.uuid); // FIXME: Temporary solution
// state.broadcasts.remove(&u.uuid); // NOTE: Create broadcasts manager ?? // state.broadcasts.remove(&u.uuid); // NOTE: Create broadcasts manager ??
state.user_manager.remove(&u.uuid); // state.user_manager.remove(&u.uuid);
} }
} }
@ -188,7 +188,7 @@ async fn subscribe(
loop { loop {
tokio::select! { tokio::select! {
_ = shutdown.notified() => { _ = shutdown.notified() => {
debug!("Shutdown SUB!"); debug!("SUB successfully closed!");
return; return;
} }
msg = rx.recv() => { msg = rx.recv() => {
@ -196,11 +196,11 @@ async fn subscribe(
if let Some(msg) = msg { if let Some(msg) = msg {
if socket.send(msg.clone()).await.is_err() { if socket.send(msg.clone()).await.is_err() {
error!("Forced shutdown SUB! Reciever closed connection?"); debug!("Forced shutdown SUB! Client died?");
return; return;
}; };
} else { } else {
error!("Forced shutdown SUB! Sender closed connection?"); debug!("Forced shutdown SUB! Source died?");
return; return;
} }
} }