iot init

pkg/rule_engine/nodes/action_clear_alarm_node.go

@@ -1,7 +1,11 @@
 package nodes
 
 import (
+	"encoding/json"
 	"github.com/sirupsen/logrus"
+	"log"
+	"pandax/apps/device/services"
+	"pandax/pkg/global"
 	"pandax/pkg/rule_engine/message"
 )
 
@@ -30,13 +34,26 @@ func (n *clearAlarmNode) Handle(msg message.Message) error {
 	cleared := n.GetLinkedNode("Cleared")
 	failure := n.GetLinkedNode("Failure")
 
-	scriptEngine := NewScriptEngine(msg, "Details", n.Script)
-	details, err := scriptEngine.ScriptAlarmDetails()
-	if err != nil {
-		return failure.Handle(msg)
+	alarm := services.DeviceAlarmModelDao.FindOneByType(msg.GetMetadata().GetKeyValue("deviceId").(string), n.AlarmType, "0")
+	if alarm.DeviceId != "" {
+		log.Println("清除告警")
+		alarm.State = global.CLEARED
+		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 cleared != nil {
+				return cleared.Handle(msg)
+			}
+		}
+	} else {
+		if failure != nil {
+			return failure.Handle(msg)
+		}
 	}
-	// TODO 编写创建告警信息
-	logrus.Info(details)
-	cleared.Handle(msg)
 	return nil
 }

pkg/rule_engine/nodes/action_create_alarm_node.go

@@ -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
 }

pkg/rule_engine/nodes/action_log_node.go

@@ -1,8 +1,9 @@
 package nodes
 
 import (
-	"pandax/apps/visual/entity"
-	"pandax/apps/visual/services"
+	"pandax/apps/rule/entity"
+	"pandax/apps/rule/services"
+	"pandax/pkg/global"
 	"pandax/pkg/rule_engine/message"
 )
 
@@ -36,13 +37,14 @@ func (n *logNode) Handle(msg message.Message) error {
 			return err
 		}
 	}
-	services.VisualRuleChainMsgLogModelDao.Insert(entity.VisualRuleChainMsgLog{
+	services.RuleChainMsgLogModelDao.Insert(entity.RuleChainMsgLog{
 		MessageId:  msg.GetId(),
 		MsgType:    msg.GetType(),
 		DeviceName: msg.GetMetadata().GetValues()["deviceName"].(string),
 		Ts:         msg.GetTs(),
 		Content:    logMessage,
 	})
+	global.Log.Info(logMessage)
 	if err != nil {
 		if failureLableNode != nil {
 			return failureLableNode.Handle(msg)

pkg/rule_engine/nodes/action_save_attributes_node.go

@@ -1,10 +1,12 @@
 package nodes
 
 import (
+	"github.com/sirupsen/logrus"
+	"pandax/pkg/global"
 	"pandax/pkg/rule_engine/message"
 )
 
-type SaveAttributesNode struct {
+type saveAttributesNode struct {
 	bareNode
 }
 
@@ -14,41 +16,33 @@ func (f saveAttributesNodeFactory) Name() string     { return "SaveAttributesNod
 func (f saveAttributesNodeFactory) Category() string { return NODE_CATEGORY_ACTION }
 func (f saveAttributesNodeFactory) Labels() []string { return []string{"Success", "Failure"} }
 func (f saveAttributesNodeFactory) Create(id string, meta Metadata) (Node, error) {
-	node := &SaveAttributesNode{
+	node := &saveAttributesNode{
 		bareNode: newBareNode(f.Name(), id, meta, f.Labels()),
 	}
 	return decodePath(meta, node)
 }
 
-func (n *SaveAttributesNode) Handle(msg message.Message) error {
+func (n *saveAttributesNode) Handle(msg message.Message) error {
+	logrus.Infof("%s handle message '%s'", n.Name(), msg.GetType())
 	successLabelNode := n.GetLinkedNode("Success")
 	failureLabelNode := n.GetLinkedNode("Failure")
-	if msg.GetType() != message.EventAttributesType {
+	if msg.GetType() != message.AttributesMes {
 		if failureLabelNode != nil {
 			return failureLabelNode.Handle(msg)
 		} else {
 			return nil
 		}
 	}
-	//deviceName := msg.GetMetadata().GetValues()["deviceName"].(string)
-	//namespace := msg.GetMetadata().GetValues()["namespace"].(string)
-	//marshal, err := json.Marshal(msg.GetMsg())
-
-	//if err != nil {
-	//	if failureLabelNode != nil {
-	//		return failureLabelNode.Handle(msg)
-	//	} else {
-	//		return nil
-	//	}
-	//}
-
-	// todo 添加设备上报参数
-
+	//deviceId := msg.GetMetadata().GetValues()["deviceId"].(string)
+	deviceName := msg.GetMetadata().GetValues()["deviceName"].(string)
+	err := global.TdDb.InsertDevice(deviceName+"_attributes", msg.GetMsg())
+	if err != nil {
+		if failureLabelNode != nil {
+			return failureLabelNode.Handle(msg)
+		}
+	}
 	if successLabelNode != nil {
 		return successLabelNode.Handle(msg)
-	} else {
-		return nil
 	}
-
 	return nil
 }

pkg/rule_engine/nodes/action_save_timeseries_node.go

@@ -1,10 +1,13 @@
 package nodes
 
 import (
+	"github.com/sirupsen/logrus"
+	"log"
+	"pandax/pkg/global"
 	"pandax/pkg/rule_engine/message"
 )
 
-type SaveTimeSeriesNode struct {
+type saveTimeSeriesNode struct {
 	bareNode
 }
 
@@ -14,29 +17,35 @@ func (f saveTimeSeriesNodeFactory) Name() string     { return "SaveTimeSeriesNod
 func (f saveTimeSeriesNodeFactory) Category() string { return NODE_CATEGORY_ACTION }
 func (f saveTimeSeriesNodeFactory) Labels() []string { return []string{"Success", "Failure"} }
 func (f saveTimeSeriesNodeFactory) Create(id string, meta Metadata) (Node, error) {
-	return nil, nil
+	node := &saveTimeSeriesNode{
+		bareNode: newBareNode(f.Name(), id, meta, f.Labels()),
+	}
+	return decodePath(meta, node)
 }
 
-func (n *SaveTimeSeriesNode) Handle(msg message.Message) error {
-	successLableNode := n.GetLinkedNode("Success")
-	failureLableNode := n.GetLinkedNode("Failure")
-	if msg.GetType() != message.EventTelemetryType {
-		if failureLableNode != nil {
-			return failureLableNode.Handle(msg)
+func (n *saveTimeSeriesNode) Handle(msg message.Message) error {
+	logrus.Infof("%s handle message '%s'", n.Name(), msg.GetType())
+	successLabelNode := n.GetLinkedNode("Success")
+	failureLabelNode := n.GetLinkedNode("Failure")
+	if msg.GetType() != message.TelemetryMes {
+		if failureLabelNode != nil {
+			return failureLabelNode.Handle(msg)
 		} else {
 			return nil
 		}
 	}
-	/*deviceName := msg.GetMetadata().GetValues()["deviceName"].(string)
-	namespace := msg.GetMetadata().GetValues()["namespace"].(string)
-	marshal, err := json.Marshal(msg.GetMsg())*/
-
-	// todo 添加设备上报遥测
-
-	if successLableNode != nil {
-		return successLableNode.Handle(msg)
-	} else {
-		return nil
+	//deviceId := msg.GetMetadata().GetValues()["deviceId"].(string)
+	deviceName := msg.GetMetadata().GetValues()["deviceName"].(string)
+	log.Println("telemetry", msg.GetMsg())
+	err := global.TdDb.InsertDevice(deviceName+"_telemetry", msg.GetMsg())
+	log.Println(err)
+	if err != nil {
+		if failureLabelNode != nil {
+			return failureLabelNode.Handle(msg)
+		}
 	}
-
+	if successLabelNode != nil {
+		return successLabelNode.Handle(msg)
+	}
+	return nil
 }

pkg/rule_engine/nodes/external_rule_chain_node.go

@@ -1,8 +1,7 @@
 package nodes
 
 import (
-	"pandax/apps/visual/services"
-	"pandax/pkg/rule_engine/manifest"
+	"pandax/apps/rule/services"
 	"pandax/pkg/rule_engine/message"
 
 	"errors"
@@ -30,11 +29,13 @@ func (f externalRuleChainNodeFactory) Create(id string, meta Metadata) (Node, er
 
 func (n *externalRuleChainNode) Handle(msg message.Message) error {
 	logrus.Infof("%s handle message '%s'", n.Name(), msg.GetType())
-	data := services.VisualRuleChainModelDao.FindOne(n.RuleId)
+	data := services.RuleChainModelDao.FindOne(n.RuleId)
 	if data == nil {
 		return errors.New(fmt.Sprintf("节点 %s ,获取规则链失败", n.Name()))
 	}
-	m, err := manifest.New([]byte(data.RuleDataJson))
+
+	/*code, _ := json.Marshal(data.RuleDataJson.LfData.DataCode)
+	m, err := manifest.New(code)
 	if err != nil {
 		logrus.WithError(err).Errorf("invalidi manifest file")
 		return err
@@ -45,6 +46,6 @@ func (n *externalRuleChainNode) Handle(msg message.Message) error {
 	}
 	if node, found := nodes[m.FirstRuleNodeId]; found {
 		go node.Handle(msg)
-	}
+	}*/
 	return nil
 }

pkg/rule_engine/nodes/external_send_email_node.go

@@ -34,11 +34,6 @@ func (f externalSendEmailNodeFactory) Create(id string, meta Metadata) (Node, er
 
 	node := &externalSendEmailNode{
 		bareNode: newBareNode(f.Name(), id, meta, f.Labels()),
-		/*Host:     email.Host,
-		Port:     email.Port,
-		Nickname: email.Nickname,
-		Secret:   email.Secret,
-		IsSSL:    email.IsSSL,*/
 	}
 	return decodePath(meta, node)
 }

pkg/rule_engine/nodes/factory.go

@@ -14,9 +14,6 @@ const (
 	NODE_CATEGORY_FLOWS      = "flows"
 )
 
-// Factory is node's factory to create node based on metadata
-// factory also manage node's metadta description which can be used by other
-// service to present node in web
 type Factory interface {
 	Name() string
 	Category() string
@@ -33,8 +30,6 @@ var (
 	allCategories     []map[string]interface{}            = make([]map[string]interface{}, 0)
 )
 
-// RegisterFactory add a new node factory and classify its category for
-// metadata description
 func RegisterFactory(f Factory) {
 	allNodeFactories[f.Name()] = f
 

pkg/rule_engine/nodes/filter_message_type_switch_node.go

@@ -14,13 +14,15 @@ func (f messageTypeSwitchNodeFactory) Name() string     { return "MessageTypeSwi
 func (f messageTypeSwitchNodeFactory) Category() string { return NODE_CATEGORY_FILTER }
 func (f messageTypeSwitchNodeFactory) Labels() []string {
 	return []string{
-		message.EventAttributesType,
-		message.EventAlarmType,
-		message.EventTelemetryType,
-		message.EventUpEventType,
-		message.EventConnectType,
-		message.EventDisConnectType,
-		"Other",
+		message.RowMes,
+		message.AttributesMes,
+		message.TelemetryMes,
+		message.RpcRequestMes,
+		message.AlarmMes,
+		message.UpEventMes,
+		message.ConnectMes,
+		message.ConnectMes,
+		message.DisConnectMes,
 	}
 }
 func (f messageTypeSwitchNodeFactory) Create(id string, meta Metadata) (Node, error) {
@@ -40,10 +42,5 @@ func (n *messageTypeSwitchNode) Handle(msg message.Message) error {
 			return node.Handle(msg)
 		}
 	}
-	// 自定义类型 或 未识别类型
-	if node := n.GetLinkedNode("Other"); node != nil {
-		return node.Handle(msg)
-	}
-	// not found
 	return nil
 }

pkg/rule_engine/nodes/filter_switch_node.go

@@ -16,13 +16,15 @@ func (f switchFilterNodeFactory) Name() string     { return "SwitchNode" }
 func (f switchFilterNodeFactory) Category() string { return NODE_CATEGORY_FILTER }
 func (f switchFilterNodeFactory) Labels() []string {
 	return []string{
-		"Failure", "True", "False",
-		message.EventAttributesType,
-		message.EventAlarmType,
-		message.EventTelemetryType,
-		message.EventUpEventType,
-		message.EventConnectType,
-		message.EventDisConnectType,
+		"True", "False",
+		message.RowMes,
+		message.AttributesMes,
+		message.TelemetryMes,
+		message.RpcRequestMes,
+		message.AlarmMes,
+		message.UpEventMes,
+		message.ConnectMes,
+		message.DisConnectMes,
 	}
 }
 func (f switchFilterNodeFactory) Create(id string, meta Metadata) (Node, error) {

pkg/rule_engine/nodes/init.go

@@ -7,7 +7,6 @@ func init() {
 	RegisterFactory(scriptFilterNodeFactory{})
 	RegisterFactory(messageTypeFilterNodeFactory{})
 	RegisterFactory(messageTypeSwitchNodeFactory{})
-	RegisterFactory(deviceTypeSwitchNodeFactory{})
 
 	RegisterFactory(transformDeleteKeyNodeFactory{})
 	RegisterFactory(transformRenameKeyNodeFactory{})
@@ -15,7 +14,6 @@ func init() {
 
 	RegisterFactory(createAlarmNodeFactory{})
 	RegisterFactory(clearAlarmNodeFactory{})
-	RegisterFactory(messageGeneratorNodeFactory{})
 	RegisterFactory(logNodeFactory{})
 	RegisterFactory(saveAttributesNodeFactory{})
 	RegisterFactory(saveTimeSeriesNodeFactory{})

pkg/rule_engine/nodes/node.go

@@ -2,11 +2,10 @@ package nodes
 
 import (
 	"errors"
+	"github.com/sirupsen/logrus"
+	"log"
 	"pandax/pkg/rule_engine/manifest"
 	"pandax/pkg/rule_engine/message"
-	"strings"
-
-	"github.com/sirupsen/logrus"
 )
 
 type Node interface {
@@ -72,6 +71,7 @@ func GetNodes(m *manifest.Manifest) (map[string]Node, error) {
 		metadata := NewMetadataWithValues(n.Properties)
 		node, err := NewNode(n.Type, n.Id, metadata)
 		if err != nil {
+			log.Println(err)
 			logrus.Errorf("new node '%s' failure", n.Id)
 			continue
 		}
@@ -93,15 +93,14 @@ func GetNodes(m *manifest.Manifest) (map[string]Node, error) {
 			continue
 		}
 		//可以有多个类型
-		//可以有多个类型
-		types := make([]string, 0)
-		if _, ok := edge.Properties["type"]; !ok {
+		types := make([]interface{}, 0)
+		if _, ok := edge.Properties["lineType"]; !ok {
 			types = append(types, "True")
 		} else {
-			types = strings.Split(edge.Properties["type"].(string), "/")
+			types = edge.Properties["lineType"].([]interface{})
 		}
 		for _, ty := range types {
-			originalNode.AddLinkedNode(ty, targetNode)
+			originalNode.AddLinkedNode(ty.(string), targetNode)
 		}
 
 	}

pkg/rule_engine/nodes/script_engine.go

@@ -12,7 +12,6 @@ type ScriptEngine interface {
 	ScriptOnSwitch() ([]string, error)
 	ScriptOnFilter() (bool, error)
 	ScriptToString() (string, error)
-	ScriptAlarmDetails() (map[string]interface{}, error)
 	ScriptGenerate() (map[string]interface{}, error)
 }
 
@@ -105,24 +104,6 @@ func (bse *baseScriptEngine) ScriptToString() (string, error) {
 	return data, nil
 }
 
-func (bse *baseScriptEngine) ScriptAlarmDetails() (map[string]interface{}, error) {
-	msg := bse.Msg
-	vm := goja.New()
-	_, err := vm.RunString(bse.Script)
-	if err != nil {
-		logrus.Info("JS代码有问题")
-		return nil, err
-	}
-	var fn func(map[string]interface{}, map[string]interface{}, string) map[string]interface{}
-	err = vm.ExportTo(vm.Get(bse.Fun), &fn)
-	if err != nil {
-		logrus.Info("Js函数映射到 Go 函数失败！")
-		return nil, err
-	}
-	datas := fn(msg.GetMsg(), msg.GetMetadata().GetValues(), msg.GetType())
-	return datas, nil
-}
-
 func (bse *baseScriptEngine) ScriptGenerate() (map[string]interface{}, error) {
 	msg := bse.Msg
 	vm := goja.New()

pkg/rule_engine/nodes/transform_script_node.go

@@ -12,7 +12,7 @@ type transformScriptNode struct {
 
 type transformScriptNodeFactory struct{}
 
-func (f transformScriptNodeFactory) Name() string     { return "ScriptNode" }
+func (f transformScriptNodeFactory) Name() string     { return "ScriptKeyNode" }
 func (f transformScriptNodeFactory) Category() string { return NODE_CATEGORY_TRANSFORM }
 func (f transformScriptNodeFactory) Labels() []string { return []string{"Success", "Failure"} }
 func (f transformScriptNodeFactory) Create(id string, meta Metadata) (Node, error) {