---

panlei

2019-08-02 c37ce161ee662e0f7f7f640e2fe2e553e1510575

 insertdata/insertDataToEs.go

@@ -165,7 +165,9 @@
func InsertFace(msg ruleserver.ResultMsg, linkId string) {
   if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.FaceResult)) > 0 {
      logger.Info("往ES插人脸数据")
      faces := PutFace(msg)
      faces := []*FaceAndRules{}
      PutFace(faces,msg)
      logger.Info("整理后的数据：",faces)
      if faces != nil {
         for _,face := range faces {
            // 上传大图
@@ -175,7 +177,7 @@
               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)
@@ -269,18 +271,15 @@
}

// 归置人脸
func PutFace(msg ruleserver.ResultMsg) []FaceAndRules{
func PutFace(faces []*FaceAndRules,msg ruleserver.ResultMsg){
   if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.FaceResult)) > 0 {
      logger.Info("整理人脸数据")
      faces := []FaceAndRules{}
      for _, faceResult := range msg.RuleResult["face"].([]ruleserver.FaceResult) {
         hebingFace(faces, faceResult)
      }
      return faces
   }
   return nil
}
func hebingFace(faces []FaceAndRules, faceResult ruleserver.FaceResult) {
func hebingFace(faces []*FaceAndRules, faceResult ruleserver.FaceResult){
   for _, arg := range faceResult.Args {
      // 拿到每一张人脸
      flag := false
@@ -290,10 +289,11 @@
            face.rules = append(face.rules,faceResult.Result)
         }
      }
      if flag {
         faces = append(faces, FaceAndRules{arg, []ruleserver.Result{faceResult.Result}})
      if !flag {
         faces = append(faces, &FaceAndRules{arg, []ruleserver.Result{faceResult.Result}})
      }
   }
   logger.Info("整理中的数据",faces)
}

// 往es中插入yolo数据