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)

結語

固然這裏只簡單介紹了幾個基本的用法,項目還有一些不支持的地方,好比:package.loadlib.更多的地方等待讀者本身去探索,後面將會提供源代碼分析的文章.