From 11082e25407ea22361c54787f84d1c1a88a93301 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期一, 04 十一月 2019 11:32:29 +0800 Subject: [PATCH] 测事件推送所用事件 --- labelFilter/readyDataForLabel.go | 31 +++++++++++++++++++++---------- 1 files changed, 21 insertions(+), 10 deletions(-) diff --git a/labelFilter/readyDataForLabel.go b/labelFilter/readyDataForLabel.go index b0b7143..6d94756 100644 --- a/labelFilter/readyDataForLabel.go +++ b/labelFilter/readyDataForLabel.go @@ -1,9 +1,10 @@ package labelFilter import ( + "basic.com/valib/logger.git" "ruleprocess/cache" - "ruleprocess/logger" - "ruleprocess/ruleserver" + "ruleprocess/structure" + "time" ) type Label struct { @@ -12,12 +13,15 @@ CameraAddr string TaskId string Time string - Persons []ruleserver.FaceResult + Persons []structure.FaceResult DefenceLevel []int32 } // 鎶婃暟鎹閰嶅埌label -func (label *Label)DataFormatToLabel(result ruleserver.ResultMsg) { - if (result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0) || (result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0) { +func (label *Label)DataFormatToLabel(result structure.ResultMsg) { + logger.Debug("face缁撴灉锛�",result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]structure.FaceResult)) > 0) + logger.Debug("yolo缁撴灉锛�",result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]structure.Result)) > 0) + if (result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]structure.FaceResult)) > 0) || (result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]structure.Result)) > 0) { + logger.Info("---------------鏍囩杩囨护鍣ㄨ祴鍊�") label.CameraId = result.Cid camera, err := cache.GetCameraById(result.Cid) if err != nil { @@ -26,13 +30,20 @@ label.CameraName = camera.Name label.CameraAddr = camera.Addr label.TaskId = result.Tasklab.Taskid - for _,faceGroup := range result.RuleResult["face"].([]ruleserver.FaceResult) { - label.Persons = append(label.Persons,faceGroup) - label.DefenceLevel = append(label.DefenceLevel,faceGroup.AlarmLevel) + label.Time = time.Now().Format("2006-01-02 15:04:05") + if result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]structure.FaceResult)) > 0 { + for _,faceGroup := range result.RuleResult["face"].([]structure.FaceResult) { + label.Persons = append(label.Persons,faceGroup) + label.DefenceLevel = append(label.DefenceLevel,faceGroup.AlarmLevel) + } } - for _,yoloGroup := range result.RuleResult["yolo"].([]ruleserver.Result) { - label.DefenceLevel = append(label.DefenceLevel,yoloGroup.AlarmLevel) + if result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]structure.Result)) > 0 { + + for _,yoloGroup := range result.RuleResult["yolo"].([]structure.Result) { + label.DefenceLevel = append(label.DefenceLevel,yoloGroup.AlarmLevel) + } } + logger.Info("------------鏍囩杩囨护鍣ㄨ祴鍊煎畬姣�") } } -- Gitblit v1.8.0