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

Get rid of unwraps where possible

Browse source
This commit is contained in:
Filip Tibell 2024-01-18 18:16:49 +01:00
parent cd090d877f
commit 13d1f13c88
No known key found for this signature in database
1 changed files with 9 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/thread_runtime.rs
View file

@ -74,18 +74,15 @@ impl ThreadRuntime {
lua.globals().set("spawn", fn_spawn)?;
lua.globals().set("defer", fn_defer)?;
// HACK: Extract mlua "pending" constant value
// HACK: Extract mlua "pending" constant value and store it
let pending = lua
.create_async_function(|_, ()| async move {
yield_now().await;
Ok(())
})
.unwrap()
.into_lua_thread(lua)
.unwrap()
.resume::<_, LuaValue>(())
.unwrap();
let pending_key = lua.create_registry_value(pending).unwrap();
})?
.into_lua_thread(lua)?
.resume::<_, LuaValue>(())?;
let pending_key = lua.create_registry_value(pending)?;
Ok(ThreadRuntime {
pending_key,
@ -112,7 +109,7 @@ impl ThreadRuntime {
let stored = ThreadWithArgs::new(lua, thread, args);
self.queue.lock_blocking().push_front(stored);
self.tx.try_send(()).unwrap();
self.tx.try_send(()).unwrap(); // Unwrap is safe since this struct also holds the receiver
}
/**
@ -149,7 +146,9 @@ impl ThreadRuntime {
// before we got here, so we need to check it again
let (thread, args) = queued_thread.into_inner(lua);
if thread.status() == LuaThreadStatus::Resumable {
let pending = lua.registry_value(&self.pending_key).unwrap();
let pending = lua
.registry_value(&self.pending_key)
.expect("ran out of memory");
let mut stream = thread.into_async::<_, LuaValue>(args);
// Keep resuming the thread until we get a