派生自 ruleprocess
insertdata/insertDataToEs.go
@@ -9,7 +9,7 @@ "net" "ruleprocess/cache" "ruleprocess/logger" "strings" "strconv" "time" "basic.com/pubsub/protomsg.git" @@ -111,6 +111,7 @@ RuleText string `json:"ruleText"` DefenceState bool `json:"defenceState"` IsLink bool `json:"isLink"` LinkInfo string `json:"linkInfo"` } // 一个face对多个规则组的归置人脸的结构体 @@ -181,7 +182,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,faceResult.IsLink}) alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText, faceResult.DefenceState,faceResult.IsLink,""}) } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) @@ -204,8 +205,9 @@ if err != nil { logger.Error("上传小图出错") } logger.Info("================小图地址:", resp["fileUrl"].(string)) //logger.Info("================小图地址:", resp["fileUrl"].(string)) sex := "" logger.Info(sex) if face.ThftRes.Gender == 1 { sex = "男" } else { @@ -213,31 +215,34 @@ } race := getRaceString(face.ThftRes.Race) ageDescription := getDescription(face.ThftRes.Age) logger.Info(ageDescription) esDataId := uuid.NewV4().String() linksId := "" if linkId != "" { linksId = linkId } logger.Info("人脸的id:",strconv.FormatUint(face.Id, 10)) logger.Info("人脸的分值：",face.Score) pervideo := PerVideoPicture{ esDataId, msg.Cid, camera.Addr, i.Timestamp, strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1], bigPhotoUrl["fileUrl"].(string), msg.Tasklab.Taskid, msg.Tasklab.Taskname, "人脸", "", alarmRules, time.Now().Format("2006-01-02 15:04:05"), // 只检测，没有比对时间 sex, strconv.FormatUint(face.Id, 10), //暂改为人脸id face.ThftRes.Age, ageDescription, fmt.Sprintf("%.2f",face.Score), // 暂改为分值 race, 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, @@ -293,7 +298,19 @@ if arg.Id == face.Id { flag = true face.rules = append(face.rules,faceResult.Result) face.Liker = append(face.Liker,arg.Liker...) // 相似者去重归并 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 { @@ -325,7 +342,11 @@ //logger.Info("应该进来才对的") sdkNames = sdkNames + yoloResult.SdkName alarm := ChangeToString(yoloResult.DefenceState, yoloResult.AlarmLevel) alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState,yoloResult.IsLink}) 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)) @@ -345,7 +366,7 @@ } else { logger.Info("上传的图片信息：", resp1) } url = append(url, strings.Split(resp1["fileUrl"].(string), "/")[1]) url = append(url, resp1["fileUrl"].(string)) } } } @@ -383,6 +404,7 @@ resp, err = util.DrawPolygonOnImage(msg.Cid, i, msg.RuleResult["yolo"].([]ruleserver.Result), weedfsUrl) if err != nil { logger.Error("画框或上传图片服务器出错", err) return } else { logger.Info("上传的图片信息：", resp) } @@ -403,52 +425,53 @@ logger.Error("查询摄像机信息失败") } serverIp, err := GetLocalIP() if resp["fileUrl"] != nil { url = append(url, resp["fileUrl"].(string)) esDataId := uuid.NewV4().String() linksId := "" if linkId != "" { linksId = linkId } peraction := Personaction{ esDataId, msg.Cid, camera.Name, camera.Addr, msg.Tasklab.Taskid, msg.Tasklab.Taskname, sdkNames, "", alarmRules, localConfig.ServerId, localConfig.ServerName, serverIp, "", url, i.Timestamp, linksId, "", isAlarm, 0, 0, 0, } requstbody, err := json.Marshal(peraction) url = append(url, strings.Split(resp["fileUrl"].(string), "/")[1]) esDataId := uuid.NewV4().String() linksId := "" if linkId != "" { linksId = linkId } peraction := Personaction{ esDataId, msg.Cid, camera.Name, camera.Addr, msg.Tasklab.Taskid, msg.Tasklab.Taskname, sdkNames, "", alarmRules, localConfig.ServerId, localConfig.ServerName, serverIp, "", url, i.Timestamp, linksId, "", isAlarm, 0, 0, 0, } requstbody, err := json.Marshal(peraction) if err != nil { logger.Info("json parse error ", err) return if err != nil { logger.Info("json parse error ", err) return } resp1, err2 := EsReq("POST", personAction, requstbody) if err2 != nil { logger.Error("往ES插入数据失败", err) } 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}) logger.Warn("__________________________________________往ES插入yolo数据成功") //os.Exit(1) } resp1, err2 := EsReq("POST", personAction, requstbody) if err2 != nil { logger.Error("往ES插入数据失败", err) } 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}) logger.Warn("__________________________________________往ES插入yolo数据成功") //os.Exit(1) } } } else { logger.Debug("timeLabel条件都不符合！") @@ -499,7 +522,7 @@ localConfig.ServerName, serverIp, "", []string{strings.Split(resp["fileUrl"].(string), "/")[1]}, []string{resp["fileUrl"].(string)}, i.Timestamp, "", "", @@ -557,7 +580,7 @@ msg.Cid, camera.Addr, i.Timestamp, strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1], bigPhotoUrl["fileUrl"].(string), msg.Tasklab.Taskid, msg.Tasklab.Taskname, "人脸",
