---

panlei

2019-08-01 db9d26e3e4bd889a1bde64651d6fc39c32ae5391

 insertdata/insertDataToEs.go

@@ -7,6 +7,7 @@
   "fmt"
   "io/ioutil"
   "net"
   "os"
   "ruleprocess/cache"
   "ruleprocess/logger"
   "strings"
@@ -111,27 +112,34 @@
}

// 往ES插数据
//func InsertToEs(msg ruleserver.ResultMsg) {
//   var timeLabel string
//   // 直接从规则的标签数据里拿符合规则的人脸结果
//   if msg.RuleResult["timeLabel"] != nil {
//      timeLabel = msg.RuleResult["timeLabel"].(string)
//   }
//   logger.Debug("插入数据前看看报警标志位：", timeLabel)
//   if timeLabel == "01" { // 无定时器状态要插入的报警数据
//      InsertFace(msg)
//      flag := ruleserver.BodyIsSame(msg.SdkMessage)
//      if !flag {
//         InsertYolo(msg)
//      }
//   }
//   if timeLabel == "10" { // 定时器状态要插入的首帧报警数据。连带着定时器开启时的那帧
//      InsertFace(msg)
//      InsertYolo(msg)
//   }
//   //if timeLabel == "12" { // 并非报警数据，只是状态改变的数据
//   //   //ChangeStatusFace(msg)
//   //   ChangeStatusYolo(msg)
//   //}
//}
func InsertToEs(msg ruleserver.ResultMsg) {
   var timeLabel string
   // 直接从规则的标签数据里拿符合规则的人脸结果
   if msg.RuleResult["timeLabel"] != nil {
      timeLabel = msg.RuleResult["timeLabel"].(string)
   }
   logger.Debug("插入数据前看看报警标志位：", timeLabel)
   if timeLabel == "01" { // 无定时器状态要插入的报警数据
      InsertFace(msg)
      flag := ruleserver.BodyIsSame(msg.SdkMessage)
      if !flag {
         InsertYolo(msg)
      }
   }
   if timeLabel == "10" { // 定时器状态要插入的首帧报警数据。连带着定时器开启时的那帧
      InsertFace(msg)
      InsertYolo(msg)
   }
   //if timeLabel == "12" { // 并非报警数据，只是状态改变的数据
   //   //ChangeStatusFace(msg)
   //   ChangeStatusYolo(msg)
   InsertFace(msg)
   //flag := ruleserver.BodyIsSame(msg.SdkMessage)
   //if !flag {
   InsertYolo(msg)
   //}
}

@@ -149,7 +157,7 @@
            }
            alarmRules := []AlarmRule{}
            alarm := ChangeToString(faceResult.DefenceState, faceResult.AlarmLevel)
            alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText,faceResult.DefenceState})
            alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText, faceResult.DefenceState})
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            bigPhotoUrl := make(map[string]interface{})
@@ -314,8 +322,9 @@
   if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]ruleserver.Result)) > 0 {
      // 先判断一下数据带的规则标签是否有可以插入的
      flag := false
      for _,res := range msg.RuleResult["yolo"].([]ruleserver.Result) {
         if res.Others["timeLabel"].(string) == "01" || res.Others["timeLabel"].(string) == "10" {
      for _, res := range msg.RuleResult["yolo"].([]ruleserver.Result) {
         logger.Info("定时器打的数字标签：",res.Others.TimeLabel)
         if res.Others.TimeLabel == "01" || res.Others.TimeLabel == "10" {
            flag = true
         }
      }
@@ -325,18 +334,19 @@
         alarmRules := []AlarmRule{}
         url := []string{}
         for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
            if yoloResult.Others["timeLabel"].(string) == "01" || yoloResult.Others["timeLabel"].(string) == "10" {
            if yoloResult.Others.TimeLabel == "01" || yoloResult.Others.TimeLabel == "10" {
               // 拼出sdkname
               logger.Info("应该进来才对的")
               sdkNames = sdkNames + yoloResult.SdkName
               alarm := ChangeToString(yoloResult.DefenceState, yoloResult.AlarmLevel)
               alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText,yoloResult.DefenceState})
               alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState})
               // 上传缓存数据的图片拿到url
               if yoloResult.Others["cacheData"] != nil {
               if yoloResult.Others.CacheData != nil {
                  //InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
                  // 把缓存的数据上传后得到地址存进去
                  // 解压缩并上传图片
                  msgs :=yoloResult.Others["cacheData"].([]ruleserver.ResultMsg)
                  for _,msg1 := range msgs {
                  msgs := yoloResult.Others.CacheData
                  for _, msg1 := range msgs {
                     bdata, err := util.UnCompress(msg1.Data)
                     if err != nil {
                        panic("解压缩图片时出现错误")
@@ -356,18 +366,21 @@
         }
         linkFlag := false
         for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
            if yoloResult.Others["timeLabel"].(string) == "01" || yoloResult.Others["timeLabel"].(string) == "10" && yoloResult.Others["linkCache"] != nil{
               for _,msg2 := range yoloResult.Others["linkCache"].([]*ruleserver.ResultMsg) {
                  logger.Warn("插入联动数据")
                  InsertYolo(*msg2)
            if (yoloResult.Others.TimeLabel == "01" || yoloResult.Others.TimeLabel == "10") && yoloResult.Others.LinkCache != nil && len(yoloResult.Others.LinkCache) > 1{
               for _, msg2 := range yoloResult.Others.LinkCache {
                  logger.Warn("插入联动数据","此帧数据的id为",msg2.Cid)
                  InsertYolo(msg2)
                  linkFlag = true
               }
            }
         }
         if linkFlag {
            // 联动数据中包含本帧数据，插入联动数据后就不需要插入本帧数据了
            logger.Warn("成功插入两个联动图片")
            os.Exit(1)
            return
         }
         //logger.Info("--------走到这儿就不一样")
         isAlarm := 0
         resp := make(map[string]interface{})
         // 解压缩并上传图片
@@ -446,7 +459,7 @@
            //os.Exit(1)
         }
      } else {
         logger.Debug("timeLabel条件都不符合！" )
         logger.Debug("timeLabel条件都不符合！")
      }
   }
}