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-08-25 17:41:59 +08:00
parent fac575ac7b
commit e436f3b7ad
37 changed files with 254 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
pkg/rule_engine/nodes/transform_rename_key_node.go
View File

@@ -26,28 +26,27 @@ func (f transformRenameKeyNodeFactory) Create(id string, meta Metadata) (Node, e
return decodePath(meta, node)
}
func (n *transformRenameKeyNode) Handle(msg message.Message) error {
logrus.Infof("%s handle message '%s'", n.Name(), msg.GetType())
func (n *transformRenameKeyNode) Handle(msg *message.Message) error {
logrus.Infof("%s handle message '%s'", n.Name(), msg.MsgType)
successLabelNode := n.GetLinkedNode("Success")
failureLabelNode := n.GetLinkedNode("Failure")
if n.FormType == "msg" {
data := msg.GetMsg()
data := msg.Msg
for _, key := range n.Keys {
if _, found := data[key.OldName]; found {
data[key.NewName] = data[key.OldName]
delete(data, key.OldName)
msg.SetMsg(data)
msg.Msg = data
}
}
} else if n.FormType == "metadata" {
data := msg.GetMetadata()
data := msg.Metadata
for _, key := range n.Keys {
if data.GetKeyValue(key.OldName) != nil {
values := data.GetValues()
values[key.NewName] = values[key.OldName]
delete(values, key.OldName)
msg.SetMetadata(message.NewDefaultMetadata(values))
if data.GetValue(key.OldName) != nil {
data[key.NewName] = data[key.OldName]
delete(data, key.OldName)
msg.Metadata = data
}
}
} else {