将Go函数导入Lua中供lua调用

有一个需求,就是使用go语言来实现一些基本模块,使用lua来实现基本逻辑,于是就有了在lua中调用go函数的需求。

go的资料非常少,好在go可以嵌入c语言,使得这个实现变得有可行性。最终使用了lua中的full userdata来实现这个东西。将go中的函数全部封装成userdata,供lua调用。这样是可行的,由于userdata可以设置metatable,metatable内可以设置一个原方法__call,那么如下的调用:

func(1, 3) 

就可以变成

func.metatable.__call(func, 1, 3)

我们只要在c中实现这么一个元方法,装入特定的metatable中,将此metatable设为函数userdata的metatable,那么就可以实现这个需求了。

在lua脚本中的所有调用,均会调用到go中的c代码部分,在c函数中,将此userdata解析,确定到某个go函数,那么在go中再调用go语言部分,那么lua-》go的桥梁就已经打通了。

那么,我们可以在go中定义一个结构

type LuaGo_State struct {
    ....BasePart
    gofunctions map[int]interface{}
    baseSeq int
}

在BasePart中应该已经实现了lua的一些基本操作,比如创建什么的,我就不贴了。baseSeq是导出的gofunction的序号,用于回调时确定回调的go函数。gofunctions则记录着某个序号对应的gofunction。

然后,我们需要在lua_state中记录该lua_state中对应的LuaGo_State。

//    开启gofunction注册支持
func (this *LuaGo_State) LuaGo_OpenInvokeCompenent() {
    //    初始化导出map
    C.luago_open(this.handle, unsafe.Pointer(this))
    this.gofunctions = make(map[int]interface{})
    this.baseSeq = 1
}

于是我们要在c代码中压入LuaGo_State。

接下来就是注册某个Go函数了,每个go函数对应一个LuaGo_State里面的序列,然后在map中做下记录,最后在c代码中进行注册。

//    压入gofunction
func (this *LuaGo_State) LuaGo_PushGoFunction(_name string, _func LuaGo_Function) int {
    funcSeq := this.luago_getGoFunctionSeq()

    this.gofunctions[funcSeq] = _func

    //    注册到lua中
    cfuncname := C.CString(_name)
    C.luago_pushGoFunction(this.handle, cfuncname, C.int(funcSeq))
    C.free(unsafe.Pointer(cfuncname))

    return funcSeq
}

在c代码中,思路就是为这个go函数创建一个userdata,userdata的值就是它的序号,然后设置好上述的metatable,记录在lua_state的全局变量里。那么在lua中进行函数调用时候,就会调用我们c中的原方法。

//    元方法
static int luago_metamethod_call(struct lua_State* L)
{
    //    stack userdata, parameters...
    if(lua_isuserdata(L, 1))
    {
        //    check valid
        int* pUserData = (int*)luaL_checkudata(L, 1, g_szGoMetaTable);
        if(0 != pUserData)
        {
            int nGoFunctionSeq = *pUserData;

            // get the golua_state
            void* pLuaGoState = luago_getGoLuaState(L);
            //    call go callback
            return (int)luago_call(pLuaGoState, nGoFunctionSeq);
        }
        else
        {
            luaL_argcheck(L, pUserData != 0, 1, "GoFunction expected");
        }
    }

    luago_error(L, "trying to call a non-callable object");
    return 0;
}

这个也很简单,将userdata的序列取出,调用go的回调函数,在go中通过这个序列找出对应的函数,调用这个函数即可。

github:luago

共 0 条回复
暂时没有人回复哦,赶紧抢沙发
发表新回复

作者

sryan
today is a good day