1. 程式人生 > >Lua自己實現載入一個模組

Lua自己實現載入一個模組

載入一個模組

import() 與 require() 功能相同,但具有一定程度的自動化特性。

假設我們有如下的目錄結構:

~~~
app/
app/classes/
app/classes/MyClass.lua
app/classes/MyClassBase.lua
app/classes/data/Data1.lua
app/classes/data/Data2.lua

~~~
MyClass 中需要載入 MyClassBase 和 MyClassData。如果用 require(),MyClass 內的程式碼如下:
local MyClassBase = require("app.classes.MyClassBase")
local MyClass = class("MyClass", MyClassBase)


local Data1 = require("app.classes.data.Data1")
local Data2 = require("app.classes.data.Data2")

假如我們將 MyClass 及其相關檔案換一個目錄存放,那麼就必須修改 MyClass 中的 require() 命令,否則將找不到模組檔案。

而使用 import(),我們只需要如下寫:

local MyClassBase = import(".MyClassBase")
local MyClass = class("MyClass", MyClassBase)


local Data1 = import(".data.Data1")
local Data2 = import(".data.Data2")
       當在模組名前面有一個"." 時,import() 會從當前模組所在目錄中查詢其他模組
。因此 MyClass 及其相關檔案不管存放到什麼目錄裡,我們都不再需要修改 MyClass 中的 import() 命令。這在開發一些重複使用的功能元件時,會非常方便。
       我們可以在模組名前新增多個"." ,這樣 import() 會從更上層的目錄開始查詢模組。
       不過 import() 只有在模組級別呼叫(也就是沒有將 import() 寫在任何函式中)時,才能夠自動得到當前模組名。如果需要在函式中呼叫 import(),那麼就需要指定當前模組名:

# MyClass.lua
# 這裡的 ... 是隱藏引數,包含了當前模組的名字,所以最好將這行程式碼寫在模組的第一行
local CURRENT_MODULE_NAME = ...

local function testLoad()
    local MyClassBase = import(".MyClassBase", CURRENT_MODULE_NAME)
    # 更多程式碼
end

@param string moduleName 要載入的模組的名字
@param [string currentModuleName] 當前模組名
@return module

function import(moduleName, currentModuleName)
    local currentModuleNameParts
    local moduleFullName = moduleName
    local offset = 1

    while true do
        if string.byte(moduleName, offset) ~= 46 then -- .
            moduleFullName = string.sub(moduleName, offset)
            if currentModuleNameParts and #currentModuleNameParts > 0 then
                moduleFullName = table.concat(currentModuleNameParts, ".") .. "." .. moduleFullName
            end
            break
        end
        offset = offset + 1

        if not currentModuleNameParts then
            if not currentModuleName then
                local n,v = debug.getlocal(3, 1)
                currentModuleName = v
            end

            currentModuleNameParts = string.split(currentModuleName, ".")
        end
        table.remove(currentModuleNameParts, #currentModuleNameParts)
    end


    return require(moduleFullName)
end