ErrorNoInternet/ErrorNoWatcher
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(events): manipulate listeners in a task to avoid deadlocks

Browse Source
This commit is contained in:
Ryan 2025-02-23 02:49:35 -05:00
parent 168ac1bb46
commit 2ea7ec9e7e
Signed by: ErrorNoInternet
GPG Key ID: 2486BFB7B1E6A4A3
1 changed files with 26 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/lua/events.rs
View File

@ -9,9 +9,9 @@ pub async fn register_functions(lua: &Lua, globals: &Table, state: State) -> Res
"add_listener", "add_listener",
lua.create_function( lua.create_function(
move |_, (event_type, callback, id): (String, Function, Option<String>)| { move |_, (event_type, callback, id): (String, Function, Option<String>)| {
let mut l = block_on(l.lock()); let l = l.clone();
tokio::spawn(async move {
l.entry(event_type).or_default().push(( l.lock().await.entry(event_type).or_default().push((
id.unwrap_or(callback.info().name.unwrap_or(format!( id.unwrap_or(callback.info().name.unwrap_or(format!(
"anonymous @ {}", "anonymous @ {}",
SystemTime::now() SystemTime::now()
@ -21,6 +21,7 @@ pub async fn register_functions(lua: &Lua, globals: &Table, state: State) -> Res
))), ))),
callback, callback,
)); ));
});
Ok(()) Ok(())
}, },
)?, )?,
@ -30,8 +31,9 @@ pub async fn register_functions(lua: &Lua, globals: &Table, state: State) -> Res
globals.set( globals.set(
"remove_listener", "remove_listener",
lua.create_function(move |_, (event_type, target_id): (String, String)| { lua.create_function(move |_, (event_type, target_id): (String, String)| {
let mut l = block_on(l.lock()); let l = l.clone();
tokio::spawn(async move {
let mut l = l.lock().await;
let empty = if let Some(listeners) = l.get_mut(&event_type) { let empty = if let Some(listeners) = l.get_mut(&event_type) {
listeners.retain(|(id, _)| target_id != *id); listeners.retain(|(id, _)| target_id != *id);
listeners.is_empty() listeners.is_empty()
@ -41,7 +43,7 @@ pub async fn register_functions(lua: &Lua, globals: &Table, state: State) -> Res
if empty { if empty {
l.remove(&event_type); l.remove(&event_type);
} }
});
Ok(()) Ok(())
})?, })?,
)?; )?;