派生自 ruleprocess
insertdata/insertDataToEs.go
@@ -1,6 +1,7 @@ package insertdata import ( "encoding/base64" "encoding/json" "errors" "fmt" @@ -103,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插数据 @@ -125,10 +127,10 @@ InsertYolo(msg) } if timeLabel == "12" { // 并非报警数据，只是状态改变的数据 //ChangeStatusFace(msg) ChangeStatusYolo(msg) } //if timeLabel == "12" { // 并非报警数据，只是状态改变的数据 // //ChangeStatusFace(msg) // ChangeStatusYolo(msg) //} } // 往es中插入人脸数据 @@ -144,8 +146,8 @@ panic("解压缩图片时出现错误") } alarmRules := []AlarmRule{} alarm := ChangeToString(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{}) @@ -176,9 +178,9 @@ } race := getRaceString(face.ThftRes.Race) ageDescription := getDescription(face.ThftRes.Age) esDataId := uuid.NewV4().String() pervideo := PerVideoPicture{ uuid.NewV4().String(), esDataId, msg.Cid, camera.Addr, i.Timestamp, @@ -195,7 +197,7 @@ race, face.ThftRes.Smile, face.ThftRes.Beauty, string(face.Feature), base64.StdEncoding.EncodeToString(face.Feature), []string{strings.Split(resp["fileUrl"].(string), "/")[1]}, "暂无集群", localConfig.ServerId, @@ -221,7 +223,7 @@ } else { logger.Info("插入es返回的信息：", resp1) // 发出录像信号 ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}}) ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: esDataId, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type: 1}) } } } @@ -252,8 +254,9 @@ logger.Error("查询摄像机信息失败") } esDataId := uuid.NewV4().String() pervideo := PerVideoPicture{ uuid.NewV4().String(), esDataId, msg.Cid, camera.Addr, i.Timestamp, @@ -297,7 +300,7 @@ } else { logger.Info("插入es返回的信息：", resp1) // 发出录像信号 ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}}) 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)) @@ -312,8 +315,8 @@ alarmRules := []AlarmRule{} for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) { sdkNames = sdkNames + yoloResult.SdkName alarm := ChangeToString(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{}) @@ -371,8 +374,9 @@ } url = append(url, strings.Split(resp1["fileUrl"].(string), "/")[1]) } esDataId := uuid.NewV4().String() peraction := Personaction{ uuid.NewV4().String(), esDataId, msg.Cid, camera.Name, camera.Addr, @@ -406,7 +410,7 @@ } else { logger.Debug("插入es返回的数据信息是：", resp1) // 发出录像信号 ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}}) 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) } @@ -441,8 +445,9 @@ logger.Error("查询摄像机信息失败") } serverIp, err := GetLocalIP() esDataId := uuid.NewV4().String() peraction := Personaction{ uuid.NewV4().String(), esDataId, msg.Cid, camera.Name, camera.Addr, @@ -476,7 +481,7 @@ } else { logger.Info("插入es返回的信息：", resp1) // 发出录像信号 ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}}) 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) } @@ -511,23 +516,28 @@ } // 把报警等级转化成汉字 func ChangeToString(i int32) string { func ChangeToString(defenceState bool, i int32) string { alarm := "" if i == 1 { alarm = "一级" if defenceState { if i == 1 { alarm = "一级" } if i == 2 { alarm = "二级" } if i == 3 { alarm = "三级" } if i == 4 { alarm = "四级" } if i == 5 { alarm = "五级" } } else { alarm = "撤防" } if i == 2 { alarm = "二级" } if i == 3 { alarm = "三级" } if i == 4 { alarm = "四级" } if i == 5 { alarm = "五级" } return alarm }
