panlei
2019-11-16 0d95e05cf47aa7b24a57a673db46d5607016016c
insertdata/insertDataToEs.go
@@ -516,22 +516,31 @@
      // 先判断一下数据带的规则标签是否有可以插入的
      logger.Info("插入定时目标信息:",len(msg.RuleResult["target"].([]structure.Result)))
      for _, yoloResult := range msg.RuleResult["target"].([]structure.Result) {
         for _, tar := range yoloResult.AlarmObj {
            // 装载目标信息
            logger.Info("报警目标个数:",len(yoloResult.AlarmObj))
            var target = new(Target)
            target.TargetId = strconv.FormatUint(tar.Id, 10)
            target.TargetScore = tar.Score
            target.TargetLocation = Points{TopLeft: Point{tar.Location.X, tar.Location.Y}, BottomRight: Point{tar.Location.X + tar.Location.Width, tar.Location.Y + tar.Location.Height}}
         // 装载目标信息`
         // 如果有首次报警的则本帧数据可以插入
         insertFlag := false
         alarmNum := 0
         for _,obj := range yoloResult.AlarmObj {
            if obj.TimeLable == "10" {
               insertFlag = true
               alarmNum++
            }
         }
         if insertFlag {
            logger.Info("报警目标个数:",alarmNum)
            //var target = new(Target)
            //target.TargetId = strconv.FormatUint(tar.Id, 10)
            //target.TargetScore = tar.Score
            //target.TargetLocation = Points{TopLeft: Point{tar.Location.X, tar.Location.Y}, BottomRight: Point{tar.Location.X + tar.Location.Width, tar.Location.Y + tar.Location.Height}}
            //logger.Info("人脸的id:",strconv.FormatUint(face.Id, 10))
            var targetInfos []Target
            targetInfos = append(targetInfos, *target)
            //var targetInfos []Target
            //targetInfos = append(targetInfos, *target)
            // 获取目标缓存图片
            url := []string{}
            //InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
            // 把缓存的数据上传后得到地址存进去
            // 解压缩并上传图片
            url1 := uploadImg(tar.CacheData)
            url1 := uploadImg(yoloResult.AlarmObj[0].CacheData)
            url = append(url,url1)
            // 添加报警规则组
            sdkNames := ""
@@ -587,7 +596,7 @@
               url = append(url, resp["fileUrl"].(string))
               //esDataId := uuid.NewV4().String()
               peraction := Personaction{
                  tar.Uuid,
                  msg.Push.PushId,
                  msg.Cid,
                  msg.Push.Cam.Name,
                  msg.Push.Cam.Addr,
@@ -607,7 +616,7 @@
                  0,
                  0,
                  0,
                  targetInfos,
                  nil,
                  linkTag,
                  linkTagInfos,
               }
@@ -635,8 +644,8 @@
                  //os.Exit(1)
               }
            }
         } else {
            logger.Info("没有首次报警的目标,不插入")
         }
      }
   }
@@ -702,7 +711,7 @@
   }
   i := protomsg.Image{}
   err = proto.Unmarshal(bdata, &i)
   resp1, err1 := util.DrawPolygonOnImageForYolo(msg.Cid, i, msg.RuleResult["target"].([]structure.Result), weedfsUrl)
   resp1, err1 := util.DrawPolygonOnImageForTarget(msg.Cid, i, msg.RuleResult["target"].([]structure.Result), weedfsUrl)
   if err1 != nil {
      logger.Error("缓存数据画框或上传图片服务器出错", err)
   } else {