panlei
2019-10-28 5312ebfc1a3dce594e3fe100743652de0634f9c1
insertdata/insertDataToEs.go
@@ -109,6 +109,7 @@
   IsAckAlarm      int         `json:"isAckAlarm"`
   IsCollect       int         `json:"isCollect"`
   IsDelete        int         `json:"isDelete"`
   TargetInfo      []Target
}
type AlarmRule struct {
@@ -364,6 +365,7 @@
         logger.Info("往ES插yolo数据")
         var sdkNames string = ""
         alarmRules := []AlarmRule{}
         var targetInfos []Target
         url := []string{}
         for _, yoloResult := range msg.RuleResult["yolo"].([]structure.Result) {
            if yoloResult.Others.TimeLabel == "01" || yoloResult.Others.TimeLabel == "10" {
@@ -398,6 +400,25 @@
                     if resp1["fileUrl"] != nil {
                        url = append(url, resp1["fileUrl"].(string))
                     }
                  }
               }
               // 装配目标信息数据
               for _,target := range yoloResult.Location  {
                  // 去重添加
                  logger.Info("装配前的原始数据是:",target)
                  var flag = true
                  for _,selectTarget := range targetInfos  {
                     if strconv.FormatUint(target.TargetId, 10) == selectTarget.TargetId {
                        flag = false
                        break
                     }
                  }
                  if flag {
                     var target1 = new(Target)
                     target1.TargetId = strconv.FormatUint(target.TargetId, 10)
                     target1.TargetScore = target.TargetScore
                     target1.TargetLocation = Points{TopLeft:Point{target.X,target.Y},BottomRight:Point{target.X+target.Width,target.Y+target.Height}}
                     targetInfos = append(targetInfos,*target1)
                  }
               }
            }
@@ -455,7 +476,9 @@
         if err != nil {
            logger.Error("查询摄像机信息失败")
         }
         logger.Info("摄像机信息:",camera)
         serverIp, err := GetLocalIP()
         logger.Info("%%%%%%%%%%%%%%%%%yolo的target信息:",targetInfos)
         if resp["fileUrl"] != nil {
            url = append(url, resp["fileUrl"].(string))
            esDataId := uuid.NewV4().String()
@@ -485,6 +508,7 @@
               0,
               0,
               0,
               targetInfos,
            }
            requstbody, err := json.Marshal(peraction)