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

Better Logging

Browse source
This commit is contained in:
Alex Emmet 2025-10-26 11:20:00 +01:00
commit ab707c1907
13 changed files with 176 additions and 53 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/omega/omega_connection.rs
View file

@ -1,6 +1,10 @@
use std::sync::Arc;
use std::time::Duration;
use crate::data::communication::{CommunicationType, CommunicationValue};
use crate::util::print::PrintType;
use crate::util::print::line;
use crate::util::print::line_err;
use crate::{
data::{
communication::DataTypes,
@ -19,8 +23,6 @@ use tokio_tungstenite::{connect_async, tungstenite::protocol::Message};
use tungstenite::Utf8Bytes;
use uuid::Uuid;
use crate::data::communication::{CommunicationType, CommunicationValue};
static WAITING_TASKS: Lazy<DashMap<Uuid, Box<dyn Fn(CommunicationValue) -> bool + Send + Sync>>> =
Lazy::new(DashMap::new);
@ -41,7 +43,10 @@ impl OmegaConnection {
async fn connect_internal(&self, mut retry: usize) {
loop {
if retry > 5 {
eprintln!("Max retry attempts reached, giving up.");
line_err(
PrintType::OmegaIn,
&"Max retry attempts reached, giving up.",
);
return;
}
@ -61,7 +66,10 @@ impl OmegaConnection {
});
}
Err(e) => {
eprintln!("WebSocket connection failed (attempt {}): {}", retry, e);
line_err(
PrintType::OmegaIn,
&format!("WebSocket connection failed (attempt {}): {}", retry, e),
);
retry += 1;
sleep(Duration::from_secs(2)).await;
continue;