panlei
2019-11-15 40ff82ab78140e77e7b9352fb461ac604fcd3a96
insertdata/insertDataToEs.go
@@ -516,124 +516,122 @@
      // 先判断一下数据带的规则标签是否有可以插入的
      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),tar.AreaJson,tar.Location,tar.Score)
            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)
            // 获取目标缓存图片
            url := []string{}
            //InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
            // 把缓存的数据上传后得到地址存进去
            // 解压缩并上传图片
            url1 := uploadImg(tar.CacheData)
            url = append(url,url1)
            // 添加报警规则组
            sdkNames := ""
            alarmRules := []AlarmRule{}
            sdkNames = sdkNames + yoloResult.SdkName
            alarm := ChangeToString(yoloResult.DefenceState, yoloResult.AlarmLevel)
            linkInfo := ""
            if yoloResult.IsLink {
               linkInfo = "联动任务"
            }
            alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState, linkInfo})
            // 联动图像的处理
            linkTagInfos := []*LinkInfo{}
            for _, yoloResult := range msg.RuleResult["target"].([]structure.Result) {
               if (yoloResult.Others.TimeLabel == "01" || yoloResult.Others.TimeLabel == "10") && yoloResult.Others.LinkCache != nil && len(yoloResult.Others.LinkCache) > 1 {
                  for _, msg2 := range yoloResult.Others.LinkCache {
                     // 把msg2的数据装配成一个PerVideoPicture结构体
                     if msg2.Cid != msg.Cid {
                        linkTagInfos = append(linkTagInfos, msg2PersonVideo(msg2))
                     }
         // 装载目标信息`
         //logger.Info("报警目标个数:",len(yoloResult.AlarmObj),tar.AreaJson,tar.Location,tar.Score)
         //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)
         // 获取目标缓存图片
         url := []string{}
         //InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
         // 把缓存的数据上传后得到地址存进去
         // 解压缩并上传图片
         url1 := uploadImg(yoloResult.AlarmObj[0].CacheData)
         url = append(url,url1)
         // 添加报警规则组
         sdkNames := ""
         alarmRules := []AlarmRule{}
         sdkNames = sdkNames + yoloResult.SdkName
         alarm := ChangeToString(yoloResult.DefenceState, yoloResult.AlarmLevel)
         linkInfo := ""
         if yoloResult.IsLink {
            linkInfo = "联动任务"
         }
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState, linkInfo})
         // 联动图像的处理
         linkTagInfos := []*LinkInfo{}
         for _, yoloResult := range msg.RuleResult["target"].([]structure.Result) {
            if (yoloResult.Others.TimeLabel == "01" || yoloResult.Others.TimeLabel == "10") && yoloResult.Others.LinkCache != nil && len(yoloResult.Others.LinkCache) > 1 {
               for _, msg2 := range yoloResult.Others.LinkCache {
                  // 把msg2的数据装配成一个PerVideoPicture结构体
                  if msg2.Cid != msg.Cid {
                     linkTagInfos = append(linkTagInfos, msg2PersonVideo(msg2))
                  }
               }
            }
            linkTag := ""
            if len(linkTagInfos) > 0 {
               linkTag = "联动任务"
            }
            isAlarm := 0
            resp := make(map[string]interface{})
            // 解压缩并上传本帧数据图片
            bdata1, err := util.UnCompress(msg.Data)
         }
         linkTag := ""
         if len(linkTagInfos) > 0 {
            linkTag = "联动任务"
         }
         isAlarm := 0
         resp := make(map[string]interface{})
         // 解压缩并上传本帧数据图片
         bdata1, err := util.UnCompress(msg.Data)
         if err != nil {
            panic("解压缩图片时出现错误")
         }
         i := protomsg.Image{}
         err = proto.Unmarshal(bdata1, &i)
         if len(alarmRules) > 0 {
            isAlarm = 1
            //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
            resp, err = util.DrawPolygonOnImageForYolo(msg.Cid, i, msg.RuleResult["target"].([]structure.Result), weedfsUrl)
            if err != nil {
               panic("解压缩图片时出现错误")
            }
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata1, &i)
            if len(alarmRules) > 0 {
               isAlarm = 1
               //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
               resp, err = util.DrawPolygonOnImageForYolo(msg.Cid, i, msg.RuleResult["target"].([]structure.Result), weedfsUrl)
               if err != nil {
                  logger.Error("画框或上传图片服务器出错", err)
                  return
               } else {
                  logger.Info("上传的图片信息:", resp)
               }
               logger.Error("画框或上传图片服务器出错", err)
               return
            } else {
               isAlarm = 0
               // 不是报警数据不存
               logger.Info("上传的图片信息:", resp)
            }
         } else {
            isAlarm = 0
            // 不是报警数据不存
            return
         }
         if resp["fileUrl"] != nil {
            url = append(url, resp["fileUrl"].(string))
            //esDataId := uuid.NewV4().String()
            peraction := Personaction{
               msg.Push.PushId,
               msg.Cid,
               msg.Push.Cam.Name,
               msg.Push.Cam.Addr,
               msg.Tasklab.Taskid,
               msg.Tasklab.Taskname,
               sdkNames,
               "",
               alarmRules,
               msg.Push.ServerId,
               msg.Push.ServerName,
               msg.Push.LocalIp,
               "",
               url,
               i.Timestamp,
               "",
               isAlarm,
               0,
               0,
               0,
               nil,
               linkTag,
               linkTagInfos,
            }
            requstbody, err := json.Marshal(peraction)
            if len(linkTagInfos)>0 {
               logger.Info("联动任务1111")
            }
            if err != nil {
               logger.Info("json parse error ", err)
               return
            }
            if resp["fileUrl"] != nil {
               url = append(url, resp["fileUrl"].(string))
               //esDataId := uuid.NewV4().String()
               peraction := Personaction{
                  tar.Uuid,
                  msg.Cid,
                  msg.Push.Cam.Name,
                  msg.Push.Cam.Addr,
                  msg.Tasklab.Taskid,
                  msg.Tasklab.Taskname,
                  sdkNames,
                  "",
                  alarmRules,
                  msg.Push.ServerId,
                  msg.Push.ServerName,
                  msg.Push.LocalIp,
                  "",
                  url,
                  i.Timestamp,
                  "",
                  isAlarm,
                  0,
                  0,
                  0,
                  targetInfos,
                  linkTag,
                  linkTagInfos,
            resp1, err2 := EsReq("POST", personAction, requstbody)
            if err2 != nil {
               logger.Error("往ES插入数据失败", err)
            } else {
               logger.Debug("插入es返回的数据信息是(target):", resp1)
               // 发出录像信号
               ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: msg.Push.PushId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, VideoUrl: msg.Push.VideoUrl, ImgId: i.Id, SdkIds: []string{"-1"}, Type: 2})
               for index,link := range linkTagInfos {
                  logger.Info("联动任务的录像信号:",index)
                  ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: msg.Push.PushId, CameraId: link.CameraId, TaskId: msg.Tasklab.Taskid, VideoUrl: msg.Push.VideoUrl, ImgId: i.Id, SdkIds: []string{strconv.Itoa(index)}, Type: 2})
               }
               requstbody, err := json.Marshal(peraction)
               if len(linkTagInfos)>0 {
                  logger.Info("联动任务1111")
               }
               if err != nil {
                  logger.Info("json parse error ", err)
                  return
               }
               resp1, err2 := EsReq("POST", personAction, requstbody)
               if err2 != nil {
                  logger.Error("往ES插入数据失败", err)
               } else {
                  logger.Debug("插入es返回的数据信息是(target):", resp1)
                  // 发出录像信号
                  ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: msg.Push.PushId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, VideoUrl: msg.Push.VideoUrl, ImgId: i.Id, SdkIds: []string{"-1"}, Type: 2})
                  for index,link := range linkTagInfos {
                     logger.Info("联动任务的录像信号:",index)
                     ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: msg.Push.PushId, CameraId: link.CameraId, TaskId: msg.Tasklab.Taskid, VideoUrl: msg.Push.VideoUrl, ImgId: i.Id, SdkIds: []string{strconv.Itoa(index)}, Type: 2})
                  }
                  logger.Warn("__________________________________________往ES插入target数据成功")
                  //os.Exit(1)
               }
               logger.Warn("__________________________________________往ES插入target数据成功")
               //os.Exit(1)
            }
         }
      }