20 个版本

使用旧的 Rust 2015

0.3.4 2024 年 4 月 10 日
0.3.2 2024 年 1 月 18 日
0.3.1 2023 年 2 月 6 日
0.2.4 2022 年 12 月 8 日
0.1.0 2016 年 3 月 31 日

#34FFI


5 crates 中使用

MIT/Apache

620KB
16K SLoC

C 15K SLoC // 0.1% comments Rust 1.5K SLoC // 0.0% comments

td_rlua

这个库是 Lua 5.3 的高级绑定。您无法访问 Lua 栈,您只能读写变量(包括回调)并执行 Lua 代码。

Build Status

如何安装它?

将以下内容添加到您的项目中的 Cargo.toml 文件

[dependencies]
td_rlua = "0.1.0"

如何使用它?

extern crate td_rlua;
use td_rlua::Lua;

Lua 结构是此库的主要元素。它表示一个可以执行 Lua 代码的上下文。

let mut lua = Lua::new();     // mutable is mandatory

读写变量

lua.set("x", 2);
let _: () = lua.exec_string("x = x + 1").unwrap();
let x: i32 = lua.query("x").unwrap();
assert_eq!(x, 3);

可以使用 setquery 函数读取和写入 Lua 上下文的全局变量。该 query 函数返回一个 Option<T> 并复制值。

可以读取和写入的基本类型包括:i8i16i32u8u16u32f32f64boolString。可以写入但不能读取 &str

如果您愿意,也可以通过实现 LuaPushLuaRead 特性来添加其他类型。

执行 Lua

let x: u32 = lua.exec_string("return 6 * 2;").unwrap();    // equals 12

exec_string 函数接受一个 &str 并返回一个 Option<T>,其中 T: LuaRead

编写函数

为了编写一个函数,你必须将其包裹在 td_rlua::functionX 里面,其中 X 是参数的数量。这是目前Rust推断系统的限制。

fn add(a: i32, b: i32) -> i32 {
    a + b
}

lua.set("add", td_rlua::function2(add));
let _: () = lua.exec_string("c = add(2, 4)").unwrap();   // calls the `add` function above
let c: i32 = lua.query("c").unwrap();
assert_eq!(c, 6);

在Lua中,函数与普通变量完全相同。

你可以编写普通函数以及闭包。

lua.set("mul", td_rlua::function2(|a: i32, b: i32| a * b));

请注意,Lua上下文的生存期必须等于或短于闭包的生存期。这是在编译时强制执行的。

let mut a = 5i;

{
    let mut lua = Lua::new();

    lua.set("inc", || a += 1);    // borrows 'a'
    for i in (0 .. 15) {
        let _: () = lua.exec_string("inc()").unwrap();
    }
} // unborrows `a`

assert_eq!(a, 20)
错误处理
extern "C" fn error_handle(lua: *mut c_lua::lua_State) -> libc::c_int {
    let err = unsafe { c_lua::lua_tostring(lua, -1) };
    let err = unsafe { CStr::from_ptr(err) };
    let err = String::from_utf8(err.to_bytes().to_vec()).unwrap();
    println!("error:{}", err);
    0
}
lua.register("error_handle", error_handle);

exec_string中的默认值会调用pcall,并将错误函数设置为_G["error_handle"],这样你可以将'error_handle'函数重置为自定义的。

操作Lua表

可以通过读取一个LuaTable对象来操作Lua表。这可以通过读取一个LuaTable对象轻松实现。

let _:() = lua.exec_string("a = { 9, 8, 7 }").unwrap();
let mut table : LuaTable = lua.query("a").unwrap();

let x: i32 = table.query(2).unwrap();
assert_eq!(x, 8);

table.set(3, "hello");
let y: String = table.query(3).unwrap();
assert_eq!(y, "hello");

let z: i32 = table.query(1).unwrap();
assert_eq!(z, 9);

然后你可以使用.iter()函数遍历表。请注意,迭代器返回的值是一个Option<(键,值)>,当键或值无法转换为请求的类型时,Option为空。当处理此类情况时,标准Iterator trait提供的filter_map函数非常有用。

let _:() = lua.exec_string("a = { 9, 8, 7 }").unwrap();
let mut table : LuaTable = lua.query("a").unwrap();
for _ in 0 .. 10 {
    let table_content: Vec<Option<(u32, u32)>> = table.iter().collect();
    assert_eq!(table_content, vec![ Some((1,9)), Some((2,8)), Some((3,7)) ]);
}

用户数据

当你将函数暴露给Lua时,你可能希望读取或写入更复杂的对象。这被称为用户数据

要这样做,你应该为你的类型实现LuaPush。这通常是通过将调用重定向到userdata::push_userdata来完成的。如果你使用userdata::push_userdata,它将操作对象的引用,如果使用userdata::push_lightuserdata,则用户数据的生命周期由Rust管理,因此不会进行复制。

#[derive(Clone, Debug)]
struct Foo {
    a : i32,
};

impl<'a> td_rlua::LuaPush for &'a mut  Foo {
    fn push_to_lua(self, lua: *mut c_lua::lua_State) -> i32 {
        td_rlua::userdata::push_userdata(self, lua, |_|{})
    }
}
impl<'a> td_rlua::LuaRead for &'a mut  Foo {
    fn lua_read_at_position(lua: *mut c_lua::lua_State, index: i32) -> Option<&'a mut Foo> {
        td_rlua::userdata::read_userdata(lua, index)
    }
}

let xx  = &mut Foo {
    a : 10,
};
lua.set("a", xx);
let get: &mut Foo = lua.query("a").unwrap();
assert!(get.a == 10);
get.a = 100;

let get: &mut Foo = lua.query("a").unwrap();
assert!(get.a == 100);

使用轻量级用户数据可以改变

impl<'a> td_rlua::LuaPush for &'a mut  Foo {
    fn push_to_lua(self, lua: *mut c_lua::lua_State) -> i32 {
        td_rlua::userdata::push_lightuserdata(self, lua, |_|{})
    }
}

自定义Lua调用用户数据需要实现NewStruct

#[derive(Clone, Debug)]
struct TestLuaSturct {
    index : i32,
}

impl NewStruct for TestLuaSturct {
    fn new() -> TestLuaSturct {
        println!("new !!!!!!!!!!!!!!");
        TestLuaSturct {
            index : 19,
        }
    }

    fn name() -> &'static str {
        "TestLuaSturct"
    }
}

impl<'a> LuaRead for &'a mut TestLuaSturct {
    fn lua_read_at_position(lua: *mut c_lua::lua_State, index: i32) -> Option<&'a mut TestLuaSturct> {
        td_rlua::userdata::read_userdata(lua, index)
    }
}

现在我们可以自定义函数

let mut lua = Lua::new();
lua.openlibs();
fn one_arg(obj : &mut TestLuaSturct) -> i32 { obj.index = 10; 5 };
fn two_arg(obj : &mut TestLuaSturct, index : i32) { obj.index = index;};

let mut value = td_rlua::LuaStruct::<TestLuaSturct>::new(lua.state());
value.create().def("one_arg", td_rlua::function1(one_arg)).def("two_arg", td_rlua::function2(two_arg));

let _ : Option<()> = lua.exec_string("x = TestLuaSturct()");
let val : Option<i32> = lua.exec_string("return x:one_arg()");
assert_eq!(val, Some(5));
let obj : Option<&mut TestLuaSturct> = lua.query("x");
assert_eq!(obj.unwrap().index, 10);
let val : Option<i32> = lua.exec_string("x:two_arg(121)");
assert_eq!(val, None);
let obj : Option<&mut TestLuaSturct> = lua.query("x");
assert_eq!(obj.unwrap().index, 121);

let obj : Option<&mut TestLuaSturct> = lua.exec_string("return TestLuaSturct()");
assert_eq!(obj.unwrap().index, 19);

热修复

在运行时,如果我们需要更改某些逻辑,我们需要重新启动进程,这可能会导致一些内存数据丢失,因此有时我们需要更新逻辑,同时保留内存数据,因此我们需要热修复。

let mut lua = Lua::new();
lua.openlibs();
lua.enable_hotfix();
let _ = lua.exec_func2("hotfix", r"
    local value = {3, 4}
    function get_a()
        value[2] = 3
        return value[1]
    end

    function get_b()
        return value[2]
    end
    ", "hotfix");

参考

项目编写参考hlua,如果你使用lua5.2,你可以使用它。

贡献

欢迎贡献!

依赖