---

panlei

2019-08-02 955dea537e8bd9c0be481796c3328c10354abfad

 insertdata/insertDataToEs.go

@@ -7,6 +7,7 @@
   "fmt"
   "io/ioutil"
   "net"
   "os"
   "ruleprocess/cache"
   "ruleprocess/logger"
   "strings"
@@ -148,6 +149,10 @@
   fk := ruleserver.TrackOrNot(msg.RuleResult)
   if fk {
      InsertYolo(msg, "")
      //if msg.Cid == "61de081a-7ed9-4970-8432-41d642c35456" {
      //   logger.Warn("捕捉188摄像机的持续时间任务.....")
      //   os.Exit(1)
      //}
   } else {
      flag := ruleserver.BodyIsSame(msg.SdkMessage)
      if !flag {
@@ -161,6 +166,7 @@
   if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.FaceResult)) > 0 {
      logger.Info("往ES插人脸数据")
      faces := PutFace(msg)
      logger.Info("整理后的数据：",faces)
      if faces != nil {
         for _,face := range faces {
            // 上传大图
@@ -170,7 +176,8 @@
               panic("解压缩图片时出现错误")
            }
            alarmRules := []AlarmRule{}
            logger.Warn("人脸的规则长度为：",len(face.rules))
            logger.Warn("人脸id为：",face.Id,"人脸的规则长度为：",len(face.rules))
            os.Exit(1)
            for _,faceResult := range face.rules {
               alarm := ChangeToString(faceResult.DefenceState, faceResult.AlarmLevel)
               alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText, faceResult.DefenceState})
@@ -271,8 +278,9 @@
         hebingFace(faces, faceResult)
      }
      return faces
   } else {
      return nil
   }
   return nil
}
func hebingFace(faces []FaceAndRules, faceResult ruleserver.FaceResult) {
   for _, arg := range faceResult.Args {
@@ -284,10 +292,11 @@
            face.rules = append(face.rules,faceResult.Result)
         }
      }
      if flag {
      if !flag {
         faces = append(faces, FaceAndRules{arg, []ruleserver.Result{faceResult.Result}})
      }
   }
   logger.Info("整理中的数据",faces)
}

// 往es中插入yolo数据