Luajit - 为什么在限制标准库时会出现性能下降

问题描述

我目前正在编写一个小的 c 程序，它在一个紧密的循环中重复调用 lua 函数 - 我正在使用 luajit 来加快速度。我的要求之一是限制脚本可用的标准库。根据 this answer 的说法，使用 lua_call 调用 luaopen_* 是实现此目的的方法，但是与使用 luaL_openlibs 相比，这会导致性能大幅下降。

一个小测试用例：

test.c

// gcc -Wall -O3 -o test test.c -lm $(pkg-config --cflags --libs luajit)
#include <luajit.h>
#include <lualib.h>
#include <luaconf.h>
#include <lauxlib.h>
lua_State *LUA = NULL;

int lua_setup(char *filename) {
    LUA = luaL_newstate();
    // lua_pushcfunction(LUA,luaopen_math);
    // lua_pushstring(LUA,LUA_MATHLIBNAME);
    // lua_call(LUA,1,0);
    luaL_openlibs(LUA);
    luaL_loadfile(LUA,filename);
    lua_pcall(LUA,0);
    return 1;
}

int main() {
    lua_setup("test.lua");
    for (int i = 0; i < 10000; i++) {
        lua_getglobal(LUA,"test");
        lua_call(LUA,1);
        float ret = lua_tonumber(LUA,-1);
        lua_pop(LUA,1);
    }
    lua_close(LUA);
}

test.lua

function test()
    r = 0
    for i=0,10000 do
        r = r + math.sin(i);
        r = r % 2;
    end
    return r
end

结果：

> time ./test 

real    0m1.696s
user    0m1.650s
sys     0m0.000s

在用注释掉的行替换 luaL_openlibs 之后

> time ./test 

real    0m6.409s
user    0m6.239s
sys     0m0.004s

谁能告诉我为什么会这样？谢谢。

解决方法

好吧，原来我是个白痴。 Luajit 包含一个需要加载的 lua 库。添加以下内容解决了问题。

lua_pushcfunction(LUA,luaopen_jit);
lua_pushstring(LUA,LUA_JITLIBNAME);
lua_call(LUA,1,0);

luajit luajit luajit