212 lines
6.3 KiB
Rust
212 lines
6.3 KiB
Rust
#![cfg_attr(
|
|
all(feature = "luajit", target_os = "macos", target_arch = "x86_64"),
|
|
feature(link_args)
|
|
)]
|
|
|
|
#[cfg_attr(
|
|
all(feature = "luajit", target_os = "macos", target_arch = "x86_64"),
|
|
link_args = "-pagezero_size 10000 -image_base 100000000",
|
|
allow(unused_attributes)
|
|
)]
|
|
extern "system" {}
|
|
|
|
use criterion::{criterion_group, criterion_main, BatchSize, Criterion};
|
|
use mlua::prelude::*;
|
|
|
|
fn create_table(c: &mut Criterion) {
|
|
c.bench_function("create table", |b| {
|
|
b.iter_batched_ref(
|
|
|| Lua::new(),
|
|
|lua| {
|
|
lua.create_table().unwrap();
|
|
},
|
|
BatchSize::SmallInput,
|
|
);
|
|
});
|
|
}
|
|
|
|
fn create_array(c: &mut Criterion) {
|
|
c.bench_function("create array 10", |b| {
|
|
b.iter_batched_ref(
|
|
|| Lua::new(),
|
|
|lua| {
|
|
let table = lua.create_table().unwrap();
|
|
for i in 1..11 {
|
|
table.set(i, i).unwrap();
|
|
}
|
|
},
|
|
BatchSize::SmallInput,
|
|
);
|
|
});
|
|
}
|
|
|
|
fn create_string_table(c: &mut Criterion) {
|
|
c.bench_function("create string table 10", |b| {
|
|
b.iter_batched_ref(
|
|
|| Lua::new(),
|
|
|lua| {
|
|
let table = lua.create_table().unwrap();
|
|
for &s in &["1", "2", "3", "4", "5", "6", "7", "8", "9", "10"] {
|
|
let s = lua.create_string(s).unwrap();
|
|
table.set(s.clone(), s).unwrap();
|
|
}
|
|
},
|
|
BatchSize::SmallInput,
|
|
);
|
|
});
|
|
}
|
|
|
|
fn call_add_function(c: &mut Criterion) {
|
|
c.bench_function("call add function 3 10", |b| {
|
|
b.iter_batched_ref(
|
|
|| {
|
|
let lua = Lua::new();
|
|
let f = {
|
|
let f: LuaFunction = lua
|
|
.load(
|
|
r#"
|
|
function(a, b, c)
|
|
return a + b + c
|
|
end
|
|
"#,
|
|
)
|
|
.eval()
|
|
.unwrap();
|
|
lua.create_registry_value(f).unwrap()
|
|
};
|
|
(lua, f)
|
|
},
|
|
|(lua, f)| {
|
|
let add_function: LuaFunction = lua.registry_value(f).unwrap();
|
|
for i in 0..10 {
|
|
let _result: i64 = add_function.call((i, i + 1, i + 2)).unwrap();
|
|
}
|
|
},
|
|
BatchSize::SmallInput,
|
|
);
|
|
});
|
|
}
|
|
|
|
fn call_add_callback(c: &mut Criterion) {
|
|
c.bench_function("call callback add 2 10", |b| {
|
|
b.iter_batched_ref(
|
|
|| {
|
|
let lua = Lua::new();
|
|
let f = {
|
|
let c: LuaFunction = lua
|
|
.create_function(|_, (a, b, c): (i64, i64, i64)| Ok(a + b + c))
|
|
.unwrap();
|
|
lua.globals().set("callback", c).unwrap();
|
|
let f: LuaFunction = lua
|
|
.load(
|
|
r#"
|
|
function()
|
|
for i = 1,10 do
|
|
callback(i, i, i)
|
|
end
|
|
end
|
|
"#,
|
|
)
|
|
.eval()
|
|
.unwrap();
|
|
lua.create_registry_value(f).unwrap()
|
|
};
|
|
(lua, f)
|
|
},
|
|
|(lua, f)| {
|
|
let entry_function: LuaFunction = lua.registry_value(f).unwrap();
|
|
entry_function.call::<_, ()>(()).unwrap();
|
|
},
|
|
BatchSize::SmallInput,
|
|
);
|
|
});
|
|
}
|
|
|
|
fn call_append_callback(c: &mut Criterion) {
|
|
c.bench_function("call callback append 10", |b| {
|
|
b.iter_batched_ref(
|
|
|| {
|
|
let lua = Lua::new();
|
|
let f = {
|
|
let c: LuaFunction = lua
|
|
.create_function(|_, (a, b): (LuaString, LuaString)| {
|
|
Ok(format!("{}{}", a.to_str()?, b.to_str()?))
|
|
})
|
|
.unwrap();
|
|
lua.globals().set("callback", c).unwrap();
|
|
let f: LuaFunction = lua
|
|
.load(
|
|
r#"
|
|
function()
|
|
for _ = 1,10 do
|
|
callback("a", "b")
|
|
end
|
|
end
|
|
"#,
|
|
)
|
|
.eval()
|
|
.unwrap();
|
|
lua.create_registry_value(f).unwrap()
|
|
};
|
|
(lua, f)
|
|
},
|
|
|(lua, f)| {
|
|
let entry_function: LuaFunction = lua.registry_value(f).unwrap();
|
|
entry_function.call::<_, ()>(()).unwrap();
|
|
},
|
|
BatchSize::SmallInput,
|
|
);
|
|
});
|
|
}
|
|
|
|
fn create_registry_values(c: &mut Criterion) {
|
|
c.bench_function("create registry 10", |b| {
|
|
b.iter_batched_ref(
|
|
|| Lua::new(),
|
|
|lua| {
|
|
for _ in 0..10 {
|
|
lua.create_registry_value(lua.pack(true).unwrap()).unwrap();
|
|
}
|
|
lua.expire_registry_values();
|
|
},
|
|
BatchSize::SmallInput,
|
|
);
|
|
});
|
|
}
|
|
|
|
fn create_userdata(c: &mut Criterion) {
|
|
struct UserData(i64);
|
|
impl LuaUserData for UserData {}
|
|
|
|
c.bench_function("create userdata 10", |b| {
|
|
b.iter_batched_ref(
|
|
|| Lua::new(),
|
|
|lua| {
|
|
let table: LuaTable = lua.create_table().unwrap();
|
|
for i in 1..11 {
|
|
table.set(i, UserData(i)).unwrap();
|
|
}
|
|
},
|
|
BatchSize::SmallInput,
|
|
);
|
|
});
|
|
}
|
|
|
|
criterion_group! {
|
|
name = benches;
|
|
config = Criterion::default()
|
|
.sample_size(200)
|
|
.noise_threshold(0.02);
|
|
targets =
|
|
create_table,
|
|
create_array,
|
|
create_string_table,
|
|
call_add_function,
|
|
call_add_callback,
|
|
call_append_callback,
|
|
create_registry_values,
|
|
create_userdata,
|
|
}
|
|
|
|
criterion_main!(benches);
|