823c2deaca
Provide a method for automatic cleanup of expired RegistryKey values, so that manually cleaning up registry values is optional.
98 lines
2.6 KiB
Rust
98 lines
2.6 KiB
Rust
use std::fmt;
|
|
use std::os::raw::{c_int, c_void};
|
|
use std::sync::{Arc, Mutex};
|
|
|
|
use ffi;
|
|
use error::Result;
|
|
use value::MultiValue;
|
|
use lua::Lua;
|
|
|
|
/// Type of Lua integer numbers.
|
|
pub type Integer = ffi::lua_Integer;
|
|
/// Type of Lua floating point numbers.
|
|
pub type Number = ffi::lua_Number;
|
|
|
|
/// A "light" userdata value. Equivalent to an unmanaged raw pointer.
|
|
#[derive(Debug, Copy, Clone, Eq, PartialEq)]
|
|
pub struct LightUserData(pub *mut c_void);
|
|
|
|
/// An auto generated key into the Lua registry.
|
|
///
|
|
/// This is a handle into a value stored inside the Lua registry, similar to the normal handle types
|
|
/// like `Table` or `Function`. The difference is that this handle does not require holding a
|
|
/// reference to a parent `Lua` instance, and thus is managed differently. Though it is more
|
|
/// difficult to use than the normal handle types, it is Send + Sync + 'static, which means that it
|
|
/// can be used in many situations where it would be impossible to store a regular handle value.
|
|
pub struct RegistryKey {
|
|
pub(crate) registry_id: c_int,
|
|
pub(crate) drop_list: Arc<Mutex<Vec<c_int>>>,
|
|
}
|
|
|
|
impl Drop for RegistryKey {
|
|
fn drop(&mut self) {
|
|
if self.registry_id != ffi::LUA_REFNIL {
|
|
self.drop_list.lock().unwrap().push(self.registry_id);
|
|
}
|
|
}
|
|
}
|
|
|
|
pub(crate) type Callback<'lua> =
|
|
Box<FnMut(&'lua Lua, MultiValue<'lua>) -> Result<MultiValue<'lua>> + 'lua>;
|
|
|
|
pub(crate) struct LuaRef<'lua> {
|
|
pub lua: &'lua Lua,
|
|
pub registry_id: c_int,
|
|
}
|
|
|
|
impl<'lua> fmt::Debug for LuaRef<'lua> {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(f, "LuaRef({})", self.registry_id)
|
|
}
|
|
}
|
|
|
|
impl<'lua> Clone for LuaRef<'lua> {
|
|
fn clone(&self) -> Self {
|
|
unsafe {
|
|
self.lua.push_ref(self.lua.state, self);
|
|
self.lua.pop_ref(self.lua.state)
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'lua> Drop for LuaRef<'lua> {
|
|
fn drop(&mut self) {
|
|
unsafe {
|
|
ffi::luaL_unref(self.lua.state, ffi::LUA_REGISTRYINDEX, self.registry_id);
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::LightUserData;
|
|
use function::Function;
|
|
use lua::Lua;
|
|
|
|
use std::os::raw::c_void;
|
|
|
|
#[test]
|
|
fn test_lightuserdata() {
|
|
let lua = Lua::new();
|
|
let globals = lua.globals();
|
|
lua.exec::<()>(
|
|
r#"
|
|
function id(a)
|
|
return a
|
|
end
|
|
"#,
|
|
None,
|
|
).unwrap();
|
|
let res = globals
|
|
.get::<_, Function>("id")
|
|
.unwrap()
|
|
.call::<_, LightUserData>(LightUserData(42 as *mut c_void))
|
|
.unwrap();
|
|
assert_eq!(res, LightUserData(42 as *mut c_void));
|
|
}
|
|
}
|