mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-23 10:58:35 +08:00
规则链添加RPC请求,及响应处理
This commit is contained in:
48
pkg/rule_engine/nodes/action_rpc_request_node.go
Normal file
48
pkg/rule_engine/nodes/action_rpc_request_node.go
Normal file
@@ -0,0 +1,48 @@
|
||||
package nodes
|
||||
|
||||
import (
|
||||
"pandax/pkg/global"
|
||||
"pandax/pkg/mqtt"
|
||||
"pandax/pkg/rule_engine/message"
|
||||
)
|
||||
|
||||
type rpcRequestNode struct {
|
||||
bareNode
|
||||
Timeout int `json:"timeout"`
|
||||
Payload mqtt.RpcPayload `json:"payload"`
|
||||
}
|
||||
|
||||
type rpcRequestNodeFactory struct{}
|
||||
|
||||
func (f rpcRequestNodeFactory) Name() string { return "RpcRequestNode" }
|
||||
func (f rpcRequestNodeFactory) Category() string { return NODE_CATEGORY_ACTION }
|
||||
func (f rpcRequestNodeFactory) Labels() []string { return []string{"Success", "Failure"} }
|
||||
func (f rpcRequestNodeFactory) Create(id string, meta Metadata) (Node, error) {
|
||||
node := &rpcRequestNode{
|
||||
bareNode: newBareNode(f.Name(), id, meta, f.Labels()),
|
||||
}
|
||||
return decodePath(meta, node)
|
||||
}
|
||||
|
||||
func (n *rpcRequestNode) Handle(msg message.Message) error {
|
||||
successLableNode := n.GetLinkedNode("Success")
|
||||
failureLableNode := n.GetLinkedNode("Failure")
|
||||
|
||||
var rpc = &mqtt.RpcRequest{Client: global.MqttClient, Mode: "double", Timeout: n.Timeout}
|
||||
rpc.GetRequestId()
|
||||
respPayload, err := rpc.RequestCmd(n.Payload)
|
||||
if err != nil {
|
||||
if failureLableNode != nil {
|
||||
return failureLableNode.Handle(msg)
|
||||
} else {
|
||||
return err
|
||||
}
|
||||
}
|
||||
msgM := msg.GetMsg()
|
||||
msgM["payload"] = respPayload
|
||||
msg.SetMsg(msgM)
|
||||
if successLableNode != nil {
|
||||
return successLableNode.Handle(msg)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user