panlei
2019-08-22 5f4aac0b2d65037876f9cf6ca76ffeba7f98220d
insertdata/insertDataToEs.go
@@ -9,7 +9,6 @@
   "net"
   "ruleprocess/cache"
   "ruleprocess/logger"
   "strings"
   "time"
   "basic.com/pubsub/protomsg.git"
@@ -110,6 +109,8 @@
   AlarmLevel   string `json:"alarmLevel"`
   RuleText     string `json:"ruleText"`
   DefenceState bool   `json:"defenceState"`
   IsLink       bool   `json:"isLink"`
   LinkInfo    string `json:"linkInfo"`
}
// 一个face对多个规则组的归置人脸的结构体
@@ -180,7 +181,7 @@
            //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})
               alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText, faceResult.DefenceState,faceResult.IsLink,""})
            }
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
@@ -203,7 +204,7 @@
            if err != nil {
               logger.Error("上传小图出错")
            }
            logger.Info("================小图地址:", resp["fileUrl"].(string))
            //logger.Info("================小图地址:", resp["fileUrl"].(string))
            sex := ""
            if face.ThftRes.Gender == 1 {
               sex = "男"
@@ -222,7 +223,7 @@
               msg.Cid,
               camera.Addr,
               i.Timestamp,
               strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1],
               bigPhotoUrl["fileUrl"].(string),
               msg.Tasklab.Taskid,
               msg.Tasklab.Taskname,
               "人脸",
@@ -236,7 +237,7 @@
               face.ThftRes.Smile,
               face.ThftRes.Beauty,
               base64.StdEncoding.EncodeToString(face.Feature),
               []string{strings.Split(resp["fileUrl"].(string), "/")[1]},
               []string{resp["fileUrl"].(string)},
               "暂无集群",
               localConfig.ServerId,
               localConfig.ServerName,
@@ -283,15 +284,28 @@
func hebingFace(faces []*FaceAndRules, faceResult ruleserver.FaceResult) []*FaceAndRules{
   for _, arg := range faceResult.Args {
      // 拿到每一张人脸
      logger.Info("归置人脸时相似者的数量:",len(arg.Liker))
      flag := false
      for _, face := range faces {
         logger.Info("归置人脸时相似者的数量:",len(face.Liker))
         for _,lik := range face.Liker {
            logger.Warn("--------合并人脸时相似者:",lik.PersonId,lik.TableName)
         }
         if arg.Id == face.Id {
            flag = true
            face.rules = append(face.rules,faceResult.Result)
            // 相似者去重归并
            for _,liker := range arg.Liker {
               flag1 := true
               for _,liker1 := range face.Liker {
                  if liker.PersonId == liker1.PersonId {
                     flag1 = false
                  }
               }
               if flag1 {
                  face.Liker = append(face.Liker,liker)
               }
            }
            //face.Liker = append(face.Liker,arg.Liker...)
         }
      }
      if !flag {
@@ -323,7 +337,11 @@
               //logger.Info("应该进来才对的")
               sdkNames = sdkNames + yoloResult.SdkName
               alarm := ChangeToString(yoloResult.DefenceState, yoloResult.AlarmLevel)
               alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState})
               linkInfo := ""
               if yoloResult.IsLink {
                  linkInfo = "联动任务"
               }
               alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState,yoloResult.IsLink,linkInfo})
               // 上传缓存数据的图片拿到url
               if yoloResult.Others.CacheData != nil {
                  //InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
@@ -343,7 +361,7 @@
                     } else {
                        logger.Info("上传的图片信息:", resp1)
                     }
                     url = append(url, strings.Split(resp1["fileUrl"].(string), "/")[1])
                     url = append(url, resp1["fileUrl"].(string))
                  }
               }
            }
@@ -402,7 +420,7 @@
         }
         serverIp, err := GetLocalIP()
         url = append(url, strings.Split(resp["fileUrl"].(string), "/")[1])
         url = append(url, resp["fileUrl"].(string))
         esDataId := uuid.NewV4().String()
         linksId := ""
         if linkId != "" {
@@ -497,7 +515,7 @@
      localConfig.ServerName,
      serverIp,
      "",
      []string{strings.Split(resp["fileUrl"].(string), "/")[1]},
      []string{resp["fileUrl"].(string)},
      i.Timestamp,
      "",
      "",
@@ -555,7 +573,7 @@
      msg.Cid,
      camera.Addr,
      i.Timestamp,
      strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1],
      bigPhotoUrl["fileUrl"].(string),
      msg.Tasklab.Taskid,
      msg.Tasklab.Taskname,
      "人脸",