工作流

pkg/flow_engine/nodes/script_engine.go

@@ -0,0 +1,84 @@
+package nodes
+
+import (
+	"github.com/dop251/goja"
+	"github.com/sirupsen/logrus"
+	"pandax/pkg/flow_engine/message"
+)
+
+type ScriptEngine interface {
+	ScriptOnMessage(msg message.Message, script string) (message.Message, error)
+	//used by filter_switch_node
+	ScriptOnSwitch(msg message.Message, script string) ([]string, error)
+	//used by filter_script_node
+	ScriptOnFilter(msg message.Message, script string) (bool, error)
+	ScriptToString(msg message.Message, script string) (string, error)
+}
+
+type baseScriptEngine struct {
+}
+
+func NewScriptEngine() ScriptEngine {
+	return &baseScriptEngine{}
+}
+
+func (bse *baseScriptEngine) ScriptOnMessage(msg message.Message, script string) (message.Message, error) {
+	vm := goja.New()
+	_, err := vm.RunString(script)
+	if err != nil {
+		logrus.Info("JS代码有问题")
+		return nil, err
+	}
+	var fn func(map[string]interface{}, map[string]interface{}, string) map[string]interface{}
+	err = vm.ExportTo(vm.Get("Transform"), &fn)
+	if err != nil {
+		logrus.Info("Js函数映射到 Go 函数失败！")
+		return nil, err
+	}
+	datas := fn(msg.GetMsg(), msg.GetMetadata().GetValues(), msg.GetType())
+	msg.SetMsg(datas["msg"].(map[string]interface{}))
+	msg.SetMetadata(message.NewDefaultMetadata(datas["metadata"].(map[string]interface{})))
+	msg.SetType(datas["msgType"].(string))
+	return msg, nil
+
+	return nil, nil
+}
+
+func (bse *baseScriptEngine) ScriptOnSwitch(msg message.Message, script string) ([]string, error) {
+	vm := goja.New()
+	_, err := vm.RunString(script)
+	if err != nil {
+		logrus.Info("JS代码有问题")
+		return nil, err
+	}
+	var fn func(map[string]interface{}, map[string]interface{}, string) []string
+	err = vm.ExportTo(vm.Get("Switch"), &fn)
+	if err != nil {
+		logrus.Info("Js函数映射到 Go 函数失败！")
+		return nil, err
+	}
+	datas := fn(msg.GetMsg(), msg.GetMetadata().GetValues(), msg.GetType())
+	return datas, nil
+}
+
+func (bse *baseScriptEngine) ScriptOnFilter(msg message.Message, script string) (bool, error) {
+	vm := goja.New()
+	_, err := vm.RunString(script)
+	if err != nil {
+		logrus.Info("JS代码有问题")
+		return false, err
+	}
+	var fn func(map[string]interface{}, map[string]interface{}, string) bool
+	err = vm.ExportTo(vm.Get("Filter"), &fn)
+	if err != nil {
+		logrus.Info("Js函数映射到 Go 函数失败！")
+		return false, err
+	}
+	datas := fn(msg.GetMsg(), msg.GetMetadata().GetValues(), msg.GetType())
+	return datas, nil
+}
+
+func (bse *baseScriptEngine) ScriptToString(msg message.Message, script string) (string, error) {
+
+	return "", nil
+}