Golang中使用lua進行擴展

前言

最近在項目中需要使用lua進行擴展,發現github上有一個用golang編寫的lua虛擬機,名字叫做gopher-lua.使用后發現還不錯,借此分享給大家.

數據類型

lua中的數據類型與golang中的數據類型對應關系作者已經在文檔中說明,值得注意的是類型是以L開頭的,類型的名稱是以LT開頭的.
golang中的數據轉換為lua中的數據就必須轉換為L開頭的類型:

str := "hello"
num := 10
L.LString(str)
L.LNumber(float64(num))

lua中的數據轉換為golang中的數據,項目提供了ToInt,CheckString之類的函數來進行轉換,但是這都是必須提前知道類型的,如果不知道就必須進行類型判斷:

value := L.Get(1)
switch value.Type() {
case lua.LTString:
case lua.LTTable:
....
}

這里還可以使用gopher-luar來方便的進行類型轉換.

golang和lua互相調用函數

golang中的函數必須轉換為func(L *lua.State) int這種形式才能注入lua中,返回參數的int代表了返回參數的個數.

func hello(L *lua.State) int {
     //將返回參數壓入棧中
     L.Push(lua.LString("hello"))
     //返回參數為1個
     return 1
}
//注入lua中
L.SetGlobal("hello", L.NewFunction(hello))

在golang中調用lua函數,lua腳本中需先定義這個函數,然后調用CallByParam進行調用:

//先獲取lua中定義的函數
fn := L.GetGlobal("hello")
if err := L.CallByParam(lua.P{
    Fn: fn, NRet: 1, Protect: true, }, lua.LNumber(10)); err != nil {
    panic(err)
}
//這里獲取函數返回值
ret := L.Get(-1)

Table

關於lua中的table是一個很強大的東西,項目對table也提供了很多方法的支持比如獲取一個字段,添加一個字段.這里推薦使用gluamapper,可以將tabl轉換為golang中的結構體或者map[string]interface{}類型,這里使用了作者提供的例子:

type Role struct {
    Name string
}

type Person struct {
    Name      string
    Age       int
    WorkPlace string
    Role      []*Role
}

L := lua.NewState()
if err := L.DoString(` person = { name = "Michel", age = "31", -- weakly input work_place = "San Jose", role = { { name = "Administrator" }, { name = "Operator" } } } `); err != nil {
    panic(err)
}
var person Person
if err := gluamapper.Map(L.GetGlobal("person").(*lua.LTable), &person); err != nil {
    panic(err)
}
fmt.Printf("%s %d", person.Name, person.Age)

模塊的加載與使用

項目中提供了lua基本模塊,調用OpenLibs就可以加載這些模塊,其中包括io,math,os,debug等.如果想自己加載可以使用SkipOpenLibs參數跳過.
如果想開發自己的庫,文檔中也做出了說明:

func Loader(L *lua.LState) int {
    //注冊模塊中的導出函數
    mod := L.SetFuncs(L.NewTable(), exports)
    L.Push(mod)
    return 1
}

var exports = map[string]lua.LGFunction{
    "myfunc": myfunc,
}

func myfunc(L *lua.LState) int {
    return 0
}
//這里就可以加載mymodule模塊
L.PreloadModule("mymodule", mymodule.Loader)