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

[优化] 规则链优化

Browse Source
This commit is contained in:
PandaX
2024-02-25 13:56:14 +08:00
parent 6f7132a009
commit 4eca198425
9 changed files with 112 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/rule_engine/engine_data.go
View File

@@ -1,6 +1,7 @@
package rule_engine
import (
"errors"
"pandax/pkg/rule_engine/message"
"pandax/pkg/rule_engine/nodes"
)
@@ -22,7 +23,7 @@ func ClearDebugData(ruleId, nodeId string) {
}
}
func GetDebugDataPage(page, pageSize int, ruleId, nodeId string) (int64, []message.DebugData) {
func GetDebugDataPage(page, pageSize int, ruleId, nodeId string) (int64, []message.DebugData, error) {
if page < 1 {
page = 1
}
@@ -35,8 +36,8 @@ func GetDebugDataPage(page, pageSize int, ruleId, nodeId string) (int64, []messa
if end >= total {
end = total
}
return int64(total), nodeData.Items[offset:end]
return int64(total), nodeData.Items[offset:end], nil
}
}
return 0, nil
return 0, nil, errors.New("规则不存在")
}

4
pkg/rule_engine/nodes/external_rule_chain_node.go
View File

@@ -29,8 +29,8 @@ func (f externalRuleChainNodeFactory) Create(id string, meta Properties) (Node,
func (n *externalRuleChainNode) Handle(msg *message.Message) error {
logrus.Infof("%s handle message '%s'", n.Name(), msg.MsgType)
data := services.RuleChainModelDao.FindOne(n.RuleId)
if data == nil {
_, err := services.RuleChainModelDao.FindOne(n.RuleId)
if err != nil {
return errors.New(fmt.Sprintf("节点 %s ,获取规则链失败", n.Name()))
}