panlei
2019-10-10 b9aa64b4dc3ddd6d0a6ec726f44aab137079122a
人体添加目标信息字段
1个文件已修改
22 ■■■■■ 已修改文件
insertdata/insertDataToEs.go 22 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
insertdata/insertDataToEs.go
@@ -109,6 +109,7 @@
    IsAckAlarm      int         `json:"isAckAlarm"`
    IsCollect       int         `json:"isCollect"`
    IsDelete        int         `json:"isDelete"`
    TargetInfo        []Target
}
type AlarmRule struct {
@@ -364,6 +365,7 @@
            logger.Info("往ES插yolo数据")
            var sdkNames string = ""
            alarmRules := []AlarmRule{}
            var targetInfos []Target
            url := []string{}
            for _, yoloResult := range msg.RuleResult["yolo"].([]structure.Result) {
                if yoloResult.Others.TimeLabel == "01" || yoloResult.Others.TimeLabel == "10" {
@@ -398,6 +400,24 @@
                            if resp1["fileUrl"] != nil {
                                url = append(url, resp1["fileUrl"].(string))
                            }
                        }
                    }
                    // 装配目标信息数据
                    for _,target := range yoloResult.Location  {
                        // 去重添加
                        var flag = true
                        for _,selectTarget := range targetInfos  {
                            if strconv.FormatUint(target.TargetId, 10) == selectTarget.TargetId {
                                flag = false
                                break
                            }
                        }
                        if flag {
                            var target1 = new(Target)
                            target1.TargetId = strconv.FormatUint(target.TargetId, 10)
                            target1.TargetScore = target.TargetScore
                            target1.TargetLocation = Points{TopLeft:Point{target.X,target.Y},BottomRight:Point{target.X+target.Width,target.Y+target.Height}}
                            targetInfos = append(targetInfos,*target1)
                        }
                    }
                }
@@ -455,6 +475,7 @@
            if err != nil {
                logger.Error("查询摄像机信息失败")
            }
            serverIp, err := GetLocalIP()
            if resp["fileUrl"] != nil {
                url = append(url, resp["fileUrl"].(string))
@@ -485,6 +506,7 @@
                    0,
                    0,
                    0,
                    targetInfos,
                }
                requstbody, err := json.Marshal(peraction)