From c91b363644ac6fab27fae10215d4993f133df619 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期一, 16 九月 2019 17:06:45 +0800 Subject: [PATCH] 读公共的配置文件 --- labelFilter/readyDataForLabel.go | 41 ++++++++++++++++++++++++++--------------- 1 files changed, 26 insertions(+), 15 deletions(-) diff --git a/labelFilter/readyDataForLabel.go b/labelFilter/readyDataForLabel.go index b24aa1f..2650c4e 100644 --- a/labelFilter/readyDataForLabel.go +++ b/labelFilter/readyDataForLabel.go @@ -4,6 +4,7 @@ "ruleprocess/cache" "ruleprocess/logger" "ruleprocess/ruleserver" + "time" ) type Label struct { @@ -17,22 +18,32 @@ } // 鎶婃暟鎹閰嶅埌label func (label *Label)DataFormatToLabel(result ruleserver.ResultMsg) { - label.CameraId = result.Cid - camera, err := cache.GetCameraById(result.Cid) - if err != nil { - logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭嚭閿�") + logger.Debug("face缁撴灉锛�",result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0) + logger.Debug("yolo缁撴灉锛�",result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0) + if (result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0) || (result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0) { + logger.Info("---------------鏍囩杩囨护鍣ㄨ祴鍊�") + label.CameraId = result.Cid + camera, err := cache.GetCameraById(result.Cid) + if err != nil { + logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭嚭閿�") + } + label.CameraName = camera.Name + label.CameraAddr = camera.Addr + label.TaskId = result.Tasklab.Taskid + label.Time = time.Now().Format("2006-01-02 15:04:05") + if result.RuleResult["face"] != nil && len(result.RuleResult["face"].([]ruleserver.FaceResult)) > 0 { + for _,faceGroup := range result.RuleResult["face"].([]ruleserver.FaceResult) { + label.Persons = append(label.Persons,faceGroup) + label.DefenceLevel = append(label.DefenceLevel,faceGroup.AlarmLevel) + } + } + if result.RuleResult["yolo"] != nil && len(result.RuleResult["yolo"].([]ruleserver.Result)) > 0 { + for _,yoloGroup := range result.RuleResult["yolo"].([]ruleserver.Result) { + label.DefenceLevel = append(label.DefenceLevel,yoloGroup.AlarmLevel) + } + } + logger.Info("------------鏍囩杩囨护鍣ㄨ祴鍊煎畬姣�") } - 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) - } - for _,yoloGroup := range result.RuleResult["yolo"].([]ruleserver.Result) { - label.DefenceLevel = append(label.DefenceLevel,yoloGroup.AlarmLevel) - } - Judge(label,result) } // 浠庣紦瀛樹腑鏌ュ嚭鎵�鏈夌殑瑙勫垯鏁版嵁 -- Gitblit v1.8.0