89 lines
1.6 KiB
Go
89 lines
1.6 KiB
Go
// Package lua ...
|
|
//
|
|
// Description : lua ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2024-11-14 17:46
|
|
package lua
|
|
|
|
import (
|
|
luaCompile "github.com/yuin/gopher-lua"
|
|
"sync"
|
|
)
|
|
|
|
var (
|
|
VMInstance *VM
|
|
)
|
|
|
|
func InitVM(poolSize int) {
|
|
if poolSize <= 0 {
|
|
poolSize = 32
|
|
}
|
|
VMInstance = &VM{
|
|
l: &sync.RWMutex{},
|
|
pool: make([]*luaCompile.LState, 0),
|
|
}
|
|
}
|
|
|
|
type VM struct {
|
|
l *sync.RWMutex
|
|
pool []*luaCompile.LState
|
|
}
|
|
|
|
// GetVm 获取一个VM实例
|
|
func (v *VM) GetVm() *luaCompile.LState {
|
|
v.l.Lock()
|
|
defer v.l.Unlock()
|
|
if len(v.pool) == 0 {
|
|
return luaCompile.NewState()
|
|
}
|
|
l := v.pool[len(v.pool)-1]
|
|
v.pool = v.pool[:len(v.pool)-1]
|
|
return l
|
|
}
|
|
|
|
// Close 关闭指定虚拟机
|
|
func (v *VM) Close(l *luaCompile.LState) {
|
|
l.Close()
|
|
v.l.Lock()
|
|
defer v.l.Unlock()
|
|
// 实例放回实例实例池
|
|
v.pool = append(v.pool, l)
|
|
}
|
|
|
|
// Shutdown 关闭全部虚拟机
|
|
func (v *VM) Shutdown() {
|
|
for _, item := range v.pool {
|
|
item.Close()
|
|
}
|
|
}
|
|
|
|
// Run 运行脚本
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 18:08 2024/11/14
|
|
func (v *VM) Run(script string) (*luaCompile.LState, error) {
|
|
l := v.GetVm()
|
|
if err := l.DoString(script); err != nil {
|
|
// 直接归还实例
|
|
l.Close()
|
|
return nil, err
|
|
}
|
|
return l, nil
|
|
}
|
|
|
|
// GetResultAndRemove 获取最后一个返回值, 并从结果中移除
|
|
func (v *VM) GetResultAndRemove(l *luaCompile.LState) luaCompile.LValue {
|
|
val := l.Get(-1)
|
|
l.Pop(1)
|
|
return val
|
|
}
|
|
|
|
// GetResultByIndex 根据索引获取返回值
|
|
func (v *VM) GetResultByIndex(l *luaCompile.LState, index int) luaCompile.LValue {
|
|
val := l.Get(index)
|
|
return val
|
|
}
|