我在Lua中创建了自定义require
函数,但是当我多次创建该函数时,它似乎并没有取代之前创建的require
函数。
我希望能够不时替换(或覆盖(以前的require
功能。
我的代码 :
#include "lua.hpp"
int main()
{
lua_State *L = luaL_newstate();
luaL_openlibs(L);
luaopen_my(L);
lua_settop(L, 0);
for (int i = 0; i < 2; ++i)
{
luaL_dostring(L, "local require_original = requiren"
"function require(name, ...)n"
"print('Hello World')n"
"local val = table.pack(require_original(name, ...))n"
"return table.unpack(val,1,val.n)n"
"endn");
luaL_dostring(L, "package.preload['A'] = function()n"
"local a = {}n"
"print('A required')n"
"return an"
"endn");
luaL_dostring(L, "local a = require 'A'");
luaL_dostring(L, "package.preload['A'] = nil package.loaded['A'] = nil");
}
lua_close(L);
}
我得到的结果:
Hello World
A required
Hello World
Hello World
A required
我期望的结果:
Hello World
A required
Hello World
A required
我应该改变什么才能得到我期望的结果?
您可以在全局命名空间中提供特殊的"卸载"功能:-(
将第一个
将第一个
luaL_dostring
中的Lua代码替换为以下内容:
if uninstall_my_require then
uninstall_my_require()
end
local require_original = require
local function my_require(name, ...)
print('Hello World')
local val = table.pack(require_original(name, ...))
return table.unpack(val,1,val.n)
end
require = my_require
function uninstall_my_require()
if require == my_require then
require = require_original
end
uninstall_my_require = nil
end
第一次将全局
require
函数设置为新函数,第二次保存新函数并将其设置为另一个实例。 在 lua 中是这样的:
local require_original1 = require
function require(name, ...)
print('Hello World')
local val = table.pack(require_original1(name, ...))
return table.unpack(val,1,val.n)
end
local require_original2 = require -- now the function above
function require(name, ...) -- redefine yet again
print('Hello World')
-- require_original2 is your function above
local val = table.pack(require_original2(name, ...))
return table.unpack(val,1,val.n)
end
您只需跟踪一次原始 require 函数,要么使用像 Egor 的答案这样的卸载函数,要么将其更改为仅分配一次的全局函数:
require_original = require_original or require