加一个req，持续时间两帧图片画框问题

panlei

2019-07-30 44441d5439df4e81d641b115c94b079ace31a5c6

 insertdata/insertDataToEs.go

@@ -104,9 +104,10 @@
}

type AlarmRule struct {
   GroupId    string `json:"groupId"`
   AlarmLevel string `json:"alarmLevel"`
   RuleText   string `json:"ruleText"`
   GroupId      string `json:"groupId"`
   AlarmLevel   string `json:"alarmLevel"`
   RuleText     string `json:"ruleText"`
   DefenceState bool   `json:"defenceState"`
}

// 往ES插数据
@@ -119,17 +120,19 @@
   logger.Debug("插入数据前看看报警标志位：", timeLabel)
   if timeLabel == "01" { // 无定时器状态要插入的报警数据
      InsertFace(msg)
      InsertYolo(msg)
      flag := ruleserver.BodyIsSame(msg.SdkMessage)
      if !flag {
         InsertYolo(msg)
      }
   }
   if timeLabel == "10" { // 定时器状态要插入的首帧报警数据。连带着定时器开启时的那帧
      InsertFace(msg)
      InsertYolo(msg)

   }
   if timeLabel == "12" { // 并非报警数据，只是状态改变的数据
      //ChangeStatusFace(msg)
      ChangeStatusYolo(msg)
   }
   //if timeLabel == "12" { // 并非报警数据，只是状态改变的数据
   //   //ChangeStatusFace(msg)
   //   ChangeStatusYolo(msg)
   //}
}

// 往es中插入人脸数据
@@ -145,8 +148,8 @@
               panic("解压缩图片时出现错误")
            }
            alarmRules := []AlarmRule{}
            alarm := ChangeToString(faceResult.DefenceState,faceResult.AlarmLevel)
            alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText})
            alarm := ChangeToString(faceResult.DefenceState, faceResult.AlarmLevel)
            alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText,faceResult.DefenceState})
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            bigPhotoUrl := make(map[string]interface{})
@@ -222,7 +225,7 @@
            } else {
               logger.Info("插入es返回的信息：", resp1)
               // 发出录像信号
               ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:1})
               ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 1})
            }
         }
      }
@@ -299,7 +302,7 @@
   } else {
      logger.Info("插入es返回的信息：", resp1)
      // 发出录像信号
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:1 })
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 1})
   }
   //if msg.RuleResult["cacheData"] != nil {
   //   InsertFace(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
@@ -314,8 +317,8 @@
      alarmRules := []AlarmRule{}
      for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
         sdkNames = sdkNames + yoloResult.SdkName
         alarm := ChangeToString(yoloResult.DefenceState,yoloResult.AlarmLevel)
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText})
         alarm := ChangeToString(yoloResult.DefenceState, yoloResult.AlarmLevel)
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText,yoloResult.DefenceState})
      }
      isAlarm := 0
      resp := make(map[string]interface{})
@@ -353,9 +356,8 @@
      }
      serverIp, err := GetLocalIP()
      url := []string{}
      url = append(url, strings.Split(resp["fileUrl"].(string), "/")[1])
      if msg.RuleResult["cacheData"] != nil {
         InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
         //InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
         // 把缓存的数据上传后得到地址存进去
         // 解压缩并上传图片
         msg1 := msg.RuleResult["cacheData"].(ruleserver.ResultMsg)
@@ -365,7 +367,7 @@
         }
         i := protomsg.Image{}
         err = proto.Unmarshal(bdata, &i)
         resp1, err1 := util.DrawPolygonOnImage(msg.Cid, i, msg.RuleResult["yolo"].([]ruleserver.Result), weedfsUrl)
         resp1, err1 := util.DrawPolygonOnImage(msg1.Cid, i, msg1.RuleResult["yolo"].([]ruleserver.Result), weedfsUrl)
         if err1 != nil {
            logger.Error("缓存数据画框或上传图片服务器出错", err)
         } else {
@@ -373,6 +375,7 @@
         }
         url = append(url, strings.Split(resp1["fileUrl"].(string), "/")[1])
      }
      url = append(url, strings.Split(resp["fileUrl"].(string), "/")[1])
      esDataId := uuid.NewV4().String()
      peraction := Personaction{
         esDataId,
@@ -409,7 +412,7 @@
      } else {
         logger.Debug("插入es返回的数据信息是：", resp1)
         // 发出录像信号
         ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:2 })
         ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 2})
         logger.Warn("__________________________________________往ES插入yolo数据成功")
         //os.Exit(1)
      }
@@ -480,7 +483,7 @@
   } else {
      logger.Info("插入es返回的信息：", resp1)
      // 发出录像信号
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:2 })
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 2})
      logger.Warn("__________________________________________往ES插入yolo数据成功")
      //os.Exit(1)
   }
@@ -515,7 +518,7 @@
}

// 把报警等级转化成汉字
func ChangeToString(defenceState bool,i int32) string {
func ChangeToString(defenceState bool, i int32) string {
   alarm := ""
   if defenceState {
      if i == 1 {