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

stable omega connection & iota auth

Browse source
This commit is contained in:
Alex Emmet 2026-01-14 16:22:26 +01:00
commit 1e33ad930a
3 changed files with 105 additions and 150 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/omega/omega_connection.rs
View file

@ -33,10 +33,6 @@ pub static WAITING_TASKS: Lazy<
DashMap<Uuid, Box<dyn Fn(Arc<OmegaConnection>, CommunicationValue) -> bool + Send + Sync>>,
> = Lazy::new(DashMap::new);
static GENERIC_TASK: Lazy<
Mutex<Option<Box<dyn Fn(Arc<OmegaConnection>, CommunicationValue) -> bool + Send + Sync>>>,
> = Lazy::new(|| Mutex::new(None));
static OMEGA_CONNECTION: Lazy<Arc<OmegaConnection>> = Lazy::new(|| {
let conn = Arc::new(OmegaConnection::new());
let conn_clone = conn.clone();
@ -311,24 +307,17 @@ impl OmegaConnection {
continue;
}
let msg_id = cv.get_id();
log_in!(PrintType::Omikron, "{}", &cv.to_json().to_string());
log_in!(PrintType::Omega, "{}", &cv.to_json().to_string());
// Handle waiting tasks
if let Some(task) = WAITING_TASKS.remove(&msg_id) {
if (task.1)(self.clone(), cv.clone()) {
// continue in the read_loop
continue;
}
} else {
// Handle generic task
let generic_task_option = GENERIC_TASK.lock().await;
if let Some(generic_task) = generic_task_option.as_ref() {
if generic_task(self.clone(), cv.clone()) {
// continue in the read_loop
}
}
}
}
#[allow(non_snake_case)]
Some(Ok(Message::Close(_))) | None => break,
Some(Ok(Message::Close(_))) | None => continue,
Some(Err(_)) => break,
_ => {}
}