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

iot init

Browse Source
This commit is contained in:
XM-GO
2023-08-22 15:17:14 +08:00
parent 85f4f328f4
commit 4344771547
143 changed files with 13004 additions and 6957 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
pkg/rule_engine/nodes/action_create_alarm_node.go
View File

@@ -1,18 +1,20 @@
package nodes
import (
"encoding/json"
"github.com/kakuilan/kgo"
"github.com/sirupsen/logrus"
"pandax/apps/device/entity"
"pandax/apps/device/services"
"pandax/pkg/global"
"pandax/pkg/rule_engine/message"
"time"
)
type createAlarmNode struct {
bareNode
Script string `json:"script" yaml:"script"`
AlarmType string `json:"alarmType" yaml:"alarmType"`
AlarmSeverity int64 `json:"alarmSeverity" yaml:"alarmSeverity"`
Propagate string `json:"propagate" yaml:"propagate"`
/* AlarmStartTime string `json:"alarmStartTime" yaml:"alarmStartTime"`
AlarmEndTime string `json:"alarmEndTime" yaml:"alarmEndTime"`*/
AlarmSeverity string `json:"alarmSeverity" yaml:"alarmSeverity"`
}
type createAlarmNodeFactory struct{}
@@ -30,20 +32,46 @@ func (f createAlarmNodeFactory) Create(id string, meta Metadata) (Node, error) {
func (n *createAlarmNode) Handle(msg message.Message) error {
logrus.Infof("%s handle message '%s'", n.Name(), msg.GetType())
node1 := n.GetLinkedNode("Created")
//node2 := n.GetLinkedNode("Updated")
node3 := n.GetLinkedNode("Failure")
created := n.GetLinkedNode("Created")
updated := n.GetLinkedNode("Updated")
failure := n.GetLinkedNode("Failure")
scriptEngine := NewScriptEngine(msg, "Details", n.Script)
details, err := scriptEngine.ScriptAlarmDetails()
if err != nil {
if node3 != nil {
return node3.Handle(msg)
alarm := services.DeviceAlarmModelDao.FindOneByType(msg.GetMetadata().GetKeyValue("deviceId").(string), n.AlarmType, "0")
if alarm.DeviceId != "" {
marshal, _ := json.Marshal(msg.GetMsg())
alarm.Details = string(marshal)
err := services.DeviceAlarmModelDao.Update(*alarm)
if err != nil {
if failure != nil {
return failure.Handle(msg)
}
} else {
if updated != nil {
return updated.Handle(msg)
}
}
} else {
alarm = &entity.DeviceAlarm{}
alarm.Id = kgo.KStr.Uniqid("a")
alarm.DeviceId = msg.GetMetadata().GetKeyValue("deviceId").(string)
alarm.ProductId = msg.GetMetadata().GetKeyValue("productId").(string)
alarm.Name = msg.GetMetadata().GetKeyValue("deviceName").(string)
alarm.Level = n.AlarmSeverity
alarm.State = global.ALARMING
alarm.Type = n.AlarmType
alarm.Time = time.Now()
marshal, _ := json.Marshal(msg.GetMsg())
alarm.Details = string(marshal)
err := services.DeviceAlarmModelDao.Insert(*alarm)
if err != nil {
if failure != nil {
return failure.Handle(msg)
}
} else {
if created != nil {
return created.Handle(msg)
}
}
}
// TODO 创建告警
logrus.Info(details)
node1.Handle(msg)
return nil
}