mirror of
https://github.com/lune-org/mlua-luau-scheduler.git
synced 2025-04-04 10:30:56 +01:00
053b85e0c1
* Minimize dependencies, no longer depending on smol directly, only async-excecutor and its utility crates * Error callback is no longer thread safe, but faster * Improved documentation and panic messages for internal workings of runtime * Depend on mlua exact version needed and serialize feature * Change crate name
47 lines
1.2 KiB
Rust
47 lines
1.2 KiB
Rust
use std::time::Duration;
|
|
|
|
use async_io::{block_on, Timer};
|
|
|
|
use mlua::prelude::*;
|
|
use mlua_luau_runtime::*;
|
|
|
|
const MAIN_SCRIPT: &str = include_str!("./lua/lots_of_threads.luau");
|
|
|
|
const ONE_NANOSECOND: Duration = Duration::from_nanos(1);
|
|
|
|
pub fn main() -> LuaResult<()> {
|
|
// Set up persistent lua environment, note that we enable thread reuse for
|
|
// mlua's internal async handling since we will be spawning lots of threads
|
|
let lua = Lua::new_with(
|
|
LuaStdLib::ALL,
|
|
LuaOptions::new()
|
|
.catch_rust_panics(false)
|
|
.thread_pool_size(10_000),
|
|
)?;
|
|
let rt = Runtime::new(&lua)?;
|
|
|
|
lua.globals().set("spawn", rt.create_spawn_function()?)?;
|
|
lua.globals().set(
|
|
"sleep",
|
|
lua.create_async_function(|_, ()| async move {
|
|
// Obviously we can't sleep for a single nanosecond since
|
|
// this uses OS scheduling under the hood, but we can try
|
|
Timer::after(ONE_NANOSECOND).await;
|
|
Ok(())
|
|
})?,
|
|
)?;
|
|
|
|
// Load the main script into the runtime
|
|
let main = lua.load(MAIN_SCRIPT);
|
|
rt.spawn_thread(main, ())?;
|
|
|
|
// Run until completion
|
|
block_on(rt.run());
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_lots_of_threads() -> LuaResult<()> {
|
|
main()
|
|
}
|