lune-org/lune
1
1
Fork 0
mirror of https://github.com/lune-org/lune.git synced 2025-01-05 19:09:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use static coroutine close function instead of always fetching global

Browse source
This commit is contained in:
Filip Tibell 2023-01-23 17:34:43 -05:00
parent dc8206d9fd
commit b1e7b2dd77
No known key found for this signature in database
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/lib/globals/task.rs
View file

@ -10,6 +10,11 @@ use smol::{channel::Sender, LocalExecutor, Timer};
use crate::{utils::table_builder::TableBuilder, LuneMessage}; use crate::{utils::table_builder::TableBuilder, LuneMessage};
pub fn create(lua: &Lua) -> LuaResult<()> { pub fn create(lua: &Lua) -> LuaResult<()> {
// HACK: There is no way to call coroutine.close directly from the mlua
// create, so we need to fetch the function and store it in the registry
let coroutine: LuaTable = lua.globals().raw_get("coroutine")?;
let close: LuaFunction = coroutine.raw_get("close")?;
lua.set_named_registry_value("coroutine.close", close)?;
lua.globals().raw_set( lua.globals().raw_set(
"task", "task",
TableBuilder::new(lua)? TableBuilder::new(lua)?
@ -77,9 +82,8 @@ async fn run_registered_task(
} }
async fn task_cancel<'a>(lua: &'a Lua, thread: LuaThread<'a>) -> LuaResult<()> { async fn task_cancel<'a>(lua: &'a Lua, thread: LuaThread<'a>) -> LuaResult<()> {
let coroutine: LuaTable = lua.globals().raw_get("coroutine")?; let close: LuaFunction = lua.named_registry_value("coroutine.close")?;
let close: LuaFunction = coroutine.raw_get("close")?; close.call_async::<_, LuaMultiValue>(thread).await?;
close.call_async(thread).await?;
Ok(()) Ok(())
} }