mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-23 02:48:34 +08:00
规则链添加RPC请求,及响应处理
This commit is contained in:
50
pkg/rule_engine/nodes/action_rpc_respond_node.go
Normal file
50
pkg/rule_engine/nodes/action_rpc_respond_node.go
Normal file
@@ -0,0 +1,50 @@
|
||||
package nodes
|
||||
|
||||
import (
|
||||
"pandax/pkg/global"
|
||||
"pandax/pkg/mqtt"
|
||||
"pandax/pkg/rule_engine/message"
|
||||
)
|
||||
|
||||
type rpcRespondNode struct {
|
||||
bareNode
|
||||
RequestId int `json:"requestId"`
|
||||
}
|
||||
|
||||
type rpcRespondFactory struct{}
|
||||
|
||||
func (f rpcRespondFactory) Name() string { return "RpcRespondNode" }
|
||||
func (f rpcRespondFactory) Category() string { return NODE_CATEGORY_ACTION }
|
||||
func (f rpcRespondFactory) Labels() []string { return []string{"Success", "Failure"} }
|
||||
func (f rpcRespondFactory) Create(id string, meta Metadata) (Node, error) {
|
||||
node := &rpcRespondNode{
|
||||
bareNode: newBareNode(f.Name(), id, meta, f.Labels()),
|
||||
}
|
||||
return decodePath(meta, node)
|
||||
}
|
||||
|
||||
func (n *rpcRespondNode) Handle(msg message.Message) error {
|
||||
successLableNode := n.GetLinkedNode("Success")
|
||||
failureLableNode := n.GetLinkedNode("Failure")
|
||||
RequestId := n.RequestId
|
||||
if RequestId == 0 {
|
||||
RequestId = int(msg.GetMetadata().GetKeyValue("requestId").(float64))
|
||||
}
|
||||
var datas = mqtt.RpcPayload{
|
||||
Method: msg.GetMsg()["method"].(string),
|
||||
Params: msg.GetMsg()["params"],
|
||||
}
|
||||
rpc := &mqtt.RpcRequest{Client: global.MqttClient, RequestId: RequestId}
|
||||
err := rpc.RespondTpc(datas)
|
||||
if err != nil {
|
||||
if failureLableNode != nil {
|
||||
return failureLableNode.Handle(msg)
|
||||
} else {
|
||||
return err
|
||||
}
|
||||
}
|
||||
if successLableNode != nil {
|
||||
return successLableNode.Handle(msg)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user