public/PandaX
1
0
Fork 0
mirror of https://gitee.com/XM-GO/PandaX.git synced 2026-04-23 10:58:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

规则链

Browse Source
This commit is contained in:
XM-GO
2023-04-18 16:29:26 +08:00
parent db5a45590a
commit 5dfa341083
46 changed files with 980 additions and 381 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
pkg/rule_engine/nodes/action_generator_node.go
View File

@@ -1,22 +1,23 @@
package nodes
import (
"fmt"
"github.com/sirupsen/logrus"
"pandax/pkg/rule_engine/message"
"time"
)
type messageGeneratorNode struct {
bareNode
DetailBuilderScript string `json:"detail_builder_script" yaml:"detail_builder_script"`
FrequenceInSecond int32 `json:"frequency" yaml:"frequency"`
Script string `json:"script" yaml:"script"`
PeriodSecond int64 `json:"periodSecond" yaml:"periodSecond"` //周期
MessageCount int64 `json:"messageCount" yaml:"messageCount"`
}
type messageGeneratorNodeFactory struct{}
func (f messageGeneratorNodeFactory) Name() string { return "GeneratorNode" }
func (f messageGeneratorNodeFactory) Name() string { return "MessageGeneratorNode" }
func (f messageGeneratorNodeFactory) Category() string { return NODE_CATEGORY_ACTION }
func (f messageGeneratorNodeFactory) Labels() []string { return []string{"Created", "Updated"} }
func (f messageGeneratorNodeFactory) Labels() []string { return []string{"Success", "Failure"} }
func (f messageGeneratorNodeFactory) Create(id string, meta Metadata) (Node, error) {
node := &messageGeneratorNode{
bareNode: newBareNode(f.Name(), id, meta, f.Labels()),
@@ -27,11 +28,36 @@ func (f messageGeneratorNodeFactory) Create(id string, meta Metadata) (Node, err
func (n *messageGeneratorNode) Handle(msg message.Message) error {
logrus.Infof("%s handle message '%s'", n.Name(), msg.GetType())
createdLabelNode := n.GetLinkedNode("Created")
updatedLabelNode := n.GetLinkedNode("Updated")
if createdLabelNode == nil || updatedLabelNode == nil {
return fmt.Errorf("no valid label linked node in %s", n.Name())
}
successLabelNode := n.GetLinkedNode("Success")
failureLabelNode := n.GetLinkedNode("Failure")
ticker := time.NewTicker(time.Duration(n.PeriodSecond) * time.Second)
count := 0
go func() {
for {
<-ticker.C
count++
if int64(count) == n.MessageCount {
ticker.Stop()
return
}
scriptEngine := NewScriptEngine(msg, "Generate", n.Script)
generate, err := scriptEngine.ScriptGenerate()
if err != nil {
if failureLabelNode != nil {
go failureLabelNode.Handle(msg)
}
return
}
msg.SetMsg(generate["msg"].(map[string]interface{}))
msg.SetType(generate["msgType"].(string))
msg.SetMetadata(message.NewDefaultMetadata(generate["metadata"].(map[string]interface{})))
if successLabelNode != nil {
go successLabelNode.Handle(msg)
}
}
}()
return nil
}