CATEGORY / Development

C/C++ 中遍历 Lua table 完整版

Permanent Link: http://wutiam.net/notes/191

在 C/C++ 中遍历一个 Lua table用 lua_next 来实现,这个大家都知道。然而,我却看到很多文章在示范 lua_next 时都只是点到为止,或绝口不提如何获取 key 值,或直接定义该 table 的 key 都是非匿名的,从而简单粗暴地使用 lua_tostring 来获取值。

仔细看看,Lua manual 里对 lua_next 的说明中最后有一句很重要的话:

While traversing a table, do not call lua_tolstring directly on a key, unless you know that the key is actually a string. Recall that lua_tolstring changes the value at the given index; this confuses the next call to lua_next.

遍历 table 的过程中不要直接对处于 -2 位置的 key 做 lua_tostring 操作(还记得这个函数说了它会原地修改栈上的值的吧),除非你确信现在这个 key 就是字符串类型,否则下一次执行 lua_next 就等着意外吧。

好吧,来个完整版,其实不就是把 key 拷贝一份到栈上,然后对这份拷贝进行取值么[1]:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#include <lauxlib.h>
#include <lua.h>
 
void traverse_table(lua_State *L, int index)
{
    lua_pushnil(L); 
    // 现在的栈:-1 => nil; index => table
    while (lua_next(L, index))
    {
        // 现在的栈:-1 => value; -2 => key; index => table
        // 拷贝一份 key 到栈顶,然后对它做 lua_tostring 就不会改变原始的 key 值了
        lua_pushvalue(L, -2);
        // 现在的栈:-1 => key; -2 => value; -3 => key; index => table
        const char* key = lua_tostring(L, -1);
        const char* value = lua_tostring(L, -2);
 
        printf("%s => %s\n", key, value);
 
        // 弹出 value 和拷贝的 key,留下原始的 key 作为下一次 lua_next 的参数
        lua_pop(L, 2);
        // 现在的栈:-1 => key; index => table
    }
    // 现在的栈:index => table (最后 lua_next 返回 0 的时候它已经把上一次留下的 key 给弹出了)
    // 所以栈已经恢复到进入这个函数时的状态
}
 
int main()
{
    lua_State *L = luaL_newstate();
    luaL_openlibs(L);
 
    // 假设现在栈上已经有了一个 talbe,内容为 {one=1,[2]='two',three=3},位置在 top
 
    traverse_table(L, -1);
 
    return 0;
}

[1] Iterate through Lua Table

1 Comments / Trackbacks / Pingbacks

Leave a Reply

:) :wink: 8-O :lol: :-D 8) :-| :mrgreen: :oops: :-o :-? :( :twisted: :cry: more »