联动任务缓存数据(结构重整)

panlei

2019-07-31 b96e8aadb0f3c216031dc8f6f5f2dad0786b390a

 insertdata/insertDataToEs.go

@@ -312,13 +312,61 @@
// 往es中插入yolo数据
func InsertYolo(msg ruleserver.ResultMsg) {
   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" {
            flag = true
         }
      }
      if flag {
      logger.Info("往ES插yolo数据")
      var sdkNames string = ""
      alarmRules := []AlarmRule{}
         url := []string{}
      for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
            if yoloResult.Others["timeLabel"].(string) == "01" || yoloResult.Others["timeLabel"].(string) == "10" {
               // 拼出sdkname
         sdkNames = sdkNames + yoloResult.SdkName
         alarm := ChangeToString(yoloResult.DefenceState, yoloResult.AlarmLevel)
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText,yoloResult.DefenceState})
               // 上传缓存数据的图片拿到url
               if yoloResult.Others["cacheData"] != nil {
                  //InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
                  // 把缓存的数据上传后得到地址存进去
                  // 解压缩并上传图片
                  msgs :=yoloResult.Others["cacheData"].([]ruleserver.ResultMsg)
                  for _,msg1 := range msgs {
                     bdata, err := util.UnCompress(msg1.Data)
                     if err != nil {
                        panic("解压缩图片时出现错误")
                     }
                     i := protomsg.Image{}
                     err = proto.Unmarshal(bdata, &i)
                     resp1, err1 := util.DrawPolygonOnImage(msg1.Cid, i, msg1.RuleResult["yolo"].([]ruleserver.Result), weedfsUrl)
                     if err1 != nil {
                        logger.Error("缓存数据画框或上传图片服务器出错", err)
                     } else {
                        logger.Info("上传的图片信息：", resp1)
                     }
                     url = append(url, strings.Split(resp1["fileUrl"].(string), "/")[1])
                  }
               }
            }
         }
         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)
                  linkFlag = true
               }
            }
         }
         if linkFlag {
            // 联动数据中包含本帧数据，插入联动数据后就不需要插入本帧数据了
            return
      }
      isAlarm := 0
      resp := make(map[string]interface{})
@@ -355,28 +403,7 @@
         logger.Error("查询摄像机信息失败")
      }
      serverIp, err := GetLocalIP()
      url := []string{}
      if msg.RuleResult["cacheData"] != nil {
         //InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
         // 把缓存的数据上传后得到地址存进去
         // 解压缩并上传图片
         msgs := msg.RuleResult["cacheData"].([]ruleserver.ResultMsg)
         for _,msg1 := range msgs {
            bdata, err := util.UnCompress(msg1.Data)
            if err != nil {
               panic("解压缩图片时出现错误")
            }
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            resp1, err1 := util.DrawPolygonOnImage(msg1.Cid, i, msg1.RuleResult["yolo"].([]ruleserver.Result), weedfsUrl)
            if err1 != nil {
               logger.Error("缓存数据画框或上传图片服务器出错", err)
            } else {
               logger.Info("上传的图片信息：", resp1)
            }
            url = append(url, strings.Split(resp1["fileUrl"].(string), "/")[1])
         }
      }

      url = append(url, strings.Split(resp["fileUrl"].(string), "/")[1])
      esDataId := uuid.NewV4().String()
      peraction := Personaction{
@@ -418,6 +445,9 @@
         logger.Warn("__________________________________________往ES插入yolo数据成功")
         //os.Exit(1)
      }
      } else {
         logger.Debug("timeLabel条件都不符合！" )
      }
   }
}
func ChangeStatusYolo(msg ruleserver.ResultMsg) {