lune-org/mlua-luau-scheduler
1
0
Fork 0
mirror of https://github.com/lune-org/mlua-luau-scheduler.git synced 2025-04-03 01:50:57 +01:00
Code Issues Projects Releases Packages Wiki Activity

Fix unwrap on cancelled threads

Browse source
This commit is contained in:
Filip Tibell 2024-02-01 14:57:31 +01:00
parent 1aba727ce6
commit c2cf9da5cd
No known key found for this signature in database
2 changed files with 23 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
lib/runtime.rs
View file

@ -326,12 +326,11 @@ impl<'lua> Runtime<'lua> {
} else {
None
};
// Spawn it on the executor and store the result when done
local_exec
.spawn(async move {
if id_tracked {
// Run until yield and check if we got a final result
let res = run_until_yield(thread.clone(), args).await;
// Create our future which will run the thread and store its final result
let fut = async move {
if id_tracked {
// Run until yield and check if we got a final result
if let Some(res) = run_until_yield(thread.clone(), args).await {
if let Err(e) = res.as_ref() {
self.error_callback.call(e);
}
@ -339,15 +338,18 @@ impl<'lua> Runtime<'lua> {
let thread_res = ThreadResult::new(res, self.lua);
result_map_inner.unwrap().insert(id, thread_res);
}
} else {
// Just run until yield
let res = run_until_yield(thread, args).await;
}
} else {
// Just run until yield
if let Some(res) = run_until_yield(thread, args).await {
if let Err(e) = res.as_ref() {
self.error_callback.call(e);
}
}
})
.detach();
}
};
// Spawn it on the executor
local_exec.spawn(fut).detach();
}
};

13
lib/util.rs
View file

@ -4,12 +4,14 @@ use mlua::prelude::*;
/**
Runs a Lua thread until it manually yields (using coroutine.yield), errors, or completes.
Returns the values yielded by the thread, or the error that caused it to stop.
May return `None` if the thread was cancelled.
Otherwise returns the values yielded by the thread, or the error that caused it to stop.
*/
pub(crate) async fn run_until_yield<'lua>(
thread: LuaThread<'lua>,
args: LuaMultiValue<'lua>,
) -> LuaResult<LuaMultiValue<'lua>> {
) -> Option<LuaResult<LuaMultiValue<'lua>>> {
let mut stream = thread.into_async(args);
/*
NOTE: It is very important that we drop the thread/stream as
@ -17,8 +19,13 @@ pub(crate) async fn run_until_yield<'lua>(
and detached tasks will not drop until the executor does
https://github.com/smol-rs/smol/issues/294
We also do not unwrap here since returning `None` is expected behavior for cancellation.
Even though we are converting into a stream, and then immediately running it,
the future may still be cancelled before it is polled, which gives us None.
*/
stream.next().await.unwrap()
stream.next().await
}
/**