Lua实现PHP的include函数

 最近在参与一个基于C++/LUA的项目, 用LUA实现单元测试时需要在测试代码内对单元测试库文件进行加载, 目的是希望做到在任何测试文件内都可以运行测试, 而不是必须在特定的入口执行. 但是使用LUA内置require函数加载会失败, 因为其是lua实现module功能的一个函数所以会有一些和模块相关的特性, 在定制文件加载目录时会受到一些局限(比如想加载上层目录的模块时), 后来在看lunit代码时找到了一个实现思路:

local function loadtestcase(filename)
if not is_string(filename) then
return error("lunit.main: invalid argument")
end
local chunk, err = loadfile(filename)
if err then
return error(err)
else
chunk()
end
end

关键就是loadfile函数, 通过手册了解, 原来此函数会加载一个lua文件然后将其保存为chunk(语句组), 以下是手册对语句组的介绍.

Lua 的一个执行单元被称作 chunk。 一个 chunk 就是一串语句段，它们会被循序的执行。 每个语句段可以以一个分号结束：

chunk ::= {stat [`;]}

这儿不允许有空的语句段，所以 ';;' 是非法的。

lua 把一个 chunk 当作一个拥有不定参数的匿名函数 （参见 ）处理。 正是这样，chunk 内可以定义局部变量，接收参数，并且返回值。

chunk 可以被保存在一个文件中，也可以保存在宿主程序的一个字符串中。 当一个 chunk 被执行，首先它会被预编译成虚拟机中的指令序列， 然后被虚拟机解释运行这些指令。

chunk 也可以被预编译成二进制形式；细节参考程序 luac。 用源码形式提供的程序和被编译过的二进制形式的程序是可以相互替换的； Lua 会自动识别文件类型并做正确的处理。

所以loadfile加载回来的语句组其实就是是个匿名函数, 我们只需执行此语句组则目标文件就会加载成功, 就像PHP的include函数一样, 化简的使用办法:

loadfile( filename )()

现在可以试试加载你需要的文件了.

loadfile( "../foo_one.lua" )()
loadfile( "bar/foo_two.lua" )()
loadfile( "lua_three.lua" )()

参考资料:

云风的中文lua手册 -

lua原版手册 -