From 77e41325d05d81d65ef004cceca46d6d1a649365 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期三, 13 十一月 2019 14:45:17 +0800 Subject: [PATCH] target --- insertdata/insertDataToEs.go | 78 +++++++++++++++++++++------------------ 1 files changed, 42 insertions(+), 36 deletions(-) diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go index a4f7a60..b273033 100644 --- a/insertdata/insertDataToEs.go +++ b/insertdata/insertDataToEs.go @@ -40,6 +40,7 @@ Id string `json:"id"` CameraId string `json:"cameraId"` CameraAddr string `json:"cameraAddr"` + CameraName string `json:"cameraName"` PicDate string `json:"picDate"` PicMaxUrl string `json:"picMaxUrl"` TaskId string `json:"taskId"` @@ -67,9 +68,9 @@ IsCollect int `json:"isCollect"` IsDelete int `json:"isDelete"` BaseInfo []*structure.BaseInfo `json:"baseInfo"` - TargetInfo []Target - LinkTag string `json:"linkTag"` - LinkTagInfo []*PerVideoPicture `json:"linkTagInfo"` + TargetInfo []Target `json:"targetInfo"` + LinkTag string `json:"linkTag"` + LinkTagInfo []*PerVideoPicture `json:"linkTagInfo"` } type Target struct { @@ -90,28 +91,28 @@ // yolo琛屼负鐨勬暟鎹粨鏋� type Personaction struct { - Id string `json:"id"` - CameraId string `json:"cameraId"` - CameraName string `json:"cameraName"` - CameraAddr string `json:"cameraAddr"` - TaskId string `json:"taskId"` - TaskName string `json:"taskName"` - SdkName string `json:"sdkName"` - Content string `json:"content"` - AlarmRules []AlarmRule `json:"alarmRules"` - AnalyServerId string `json:"analyServerId"` - AnalyServerName string `json:"analyServerName"` - AnalyServerIp string `json:"analyServerIp"` - ClusterId string `json:"clusterId"` - PicSmUrl []string `json:"picSmUrl"` - PicDate string `json:"picDate"` - VideoUrl string `json:"videoUrl"` - IsAlarm int `json:"isAlarm"` - IsAckAlarm int `json:"isAckAlarm"` - IsCollect int `json:"isCollect"` - IsDelete int `json:"isDelete"` - TargetInfo []Target - LinkTag string `json:"linkTag"` + Id string `json:"id"` + CameraId string `json:"cameraId"` + CameraName string `json:"cameraName"` + CameraAddr string `json:"cameraAddr"` + TaskId string `json:"taskId"` + TaskName string `json:"taskName"` + SdkName string `json:"sdkName"` + Content string `json:"content"` + AlarmRules []AlarmRule `json:"alarmRules"` + AnalyServerId string `json:"analyServerId"` + AnalyServerName string `json:"analyServerName"` + AnalyServerIp string `json:"analyServerIp"` + ClusterId string `json:"clusterId"` + PicSmUrl []string `json:"picSmUrl"` + PicDate string `json:"picDate"` + VideoUrl string `json:"videoUrl"` + IsAlarm int `json:"isAlarm"` + IsAckAlarm int `json:"isAckAlarm"` + IsCollect int `json:"isCollect"` + IsDelete int `json:"isDelete"` + TargetInfo []Target `json:"targetInfo"` + LinkTag string `json:"linkTag"` LinkTagInfo []*PerVideoPicture `json:"linkTagInfo"` } @@ -120,7 +121,6 @@ AlarmLevel string `json:"alarmLevel"` RuleText string `json:"ruleText"` DefenceState bool `json:"defenceState"` - IsLink bool `json:"isLink"` LinkInfo string `json:"linkInfo"` } @@ -168,7 +168,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, ""}) } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) @@ -202,11 +202,12 @@ target.TargetLocation = Points{TopLeft: Point{face.Location.X, face.Location.Y}, BottomRight: Point{face.Location.X + face.Location.Width, face.Location.Y + face.Location.Height}} //logger.Info("浜鸿劯鐨刬d:",strconv.FormatUint(face.Id, 10)) var targetInfos []Target - targetInfos = append(targetInfos,*target) + targetInfos = append(targetInfos, *target) pervideo := PerVideoPicture{ face.Uuid, msg.Cid, msg.Push.Cam.Addr, + msg.Push.Cam.Name, picTime, imgMaxUrl, msg.Tasklab.Taskid, @@ -329,7 +330,7 @@ if yoloResult.IsLink { linkInfo = "鑱斿姩浠诲姟" } - alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState, yoloResult.IsLink, linkInfo}) + alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState, linkInfo}) // 涓婁紶缂撳瓨鏁版嵁鐨勫浘鐗囨嬁鍒皍rl if yoloResult.Others.CacheData != nil { //InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg)) @@ -380,9 +381,13 @@ if (yoloResult.Others.TimeLabel == "01" || yoloResult.Others.TimeLabel == "10") && yoloResult.Others.LinkCache != nil && len(yoloResult.Others.LinkCache) > 1 { for _, msg2 := range yoloResult.Others.LinkCache { // 鎶妋sg2鐨勬暟鎹閰嶆垚涓�涓狿erVideoPicture缁撴瀯浣� - linkTagInfos = append(linkTagInfos,msg2PersonVideo(msg2)) + linkTagInfos = append(linkTagInfos, msg2PersonVideo(msg2)) } } + } + linkTag := "" + if len(linkTagInfos) > 0 { + linkTag = "鑱斿姩浠诲姟" } isAlarm := 0 resp := make(map[string]interface{}) @@ -433,7 +438,7 @@ 0, 0, targetInfos, - "", + linkTag, linkTagInfos, } requstbody, err := json.Marshal(peraction) @@ -542,7 +547,7 @@ return race } -func msg2PersonVideo (msg structure.ResultMsg) *PerVideoPicture{ +func msg2PersonVideo(msg structure.ResultMsg) *PerVideoPicture { if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]structure.Result)) > 0 { // 鍏堝垽鏂竴涓嬫暟鎹甫鐨勮鍒欐爣绛炬槸鍚︽湁鍙互鎻掑叆鐨� flag := false @@ -568,7 +573,7 @@ if yoloResult.IsLink { linkInfo = "鑱斿姩浠诲姟" } - alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState, yoloResult.IsLink, linkInfo}) + alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText, yoloResult.DefenceState, linkInfo}) // 涓婁紶缂撳瓨鏁版嵁鐨勫浘鐗囨嬁鍒皍rl if yoloResult.Others.CacheData != nil { //InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg)) @@ -644,6 +649,7 @@ msg.Push.PushId + "-" + uuid.NewV4().String(), msg.Cid, msg.Push.Cam.Addr, + msg.Push.Cam.Name, i.Timestamp, "", msg.Tasklab.Taskid, @@ -652,13 +658,13 @@ "", alarmRules, time.Now().Format("2006-01-02 15:04:05"), // 鍙娴嬶紝娌℃湁姣斿鏃堕棿 - "", //鏆傛敼涓轰汉鑴竔d strconv.FormatUint(face.Id, 10) + "", //鏆傛敼涓轰汉鑴竔d strconv.FormatUint(face.Id, 10) 0, "", // 鏆傛敼涓哄垎鍊糵mt.Sprintf("%.2f",face.Score) "", 0, 0, - nil, + "", []string{resp["fileUrl"].(string)}, "", msg.Push.ServerId, @@ -686,4 +692,4 @@ } else { return nil } -} \ No newline at end of file +} -- Gitblit v1.8.0