---

panlei

2019-12-17 3ff1af612ee690063dceb5aa9a6e9791e6d9a943

---

 algorithm/middleware/insertDataToEs.go

@@ -137,110 +137,109 @@
            var imgMaxUrl []string = []string{}
            var picTime string = ""
            for _, face := range faces {
                // 上传大图
                // 解压缩并上传图片
                bdata, err := util.UnCompress(msg.Data)
                if err != nil {
                    panic("解压缩图片时出现错误")
                }
                alarmRules := []AlarmRule{}
                //logger.Info("循环每一个目标")
                for _, faceResult := range face.rules {
                    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)
                // 先传小图，再传大图，防止脸上有线
                bytes := util.SubImg(i, int(face.Location.X), int(face.Location.Y), int(face.Location.X+face.Location.Width), int(face.Location.Y+face.Location.Height),face.Type)
                resp, err := util.PostFormBufferData1(weedfsUrl, bytes, uuid.NewV4().String())
                if err != nil {
                    logger.Error("上传小图出错")
                }
                // 上传大图
                if len(imgMaxUrl) == 0 {
                    bigPhotoUrl := make(map[string]interface{})
                    bigPhotoUrl, err = util.DrawPolygonOnImageForFace(msg.Cid, i, msg.RuleResult["face"].([]structure.Result), weedfsUrl)
                    logger.Debug("========大图路径：", bigPhotoUrl)
                    imgMaxUrl = append(imgMaxUrl, bigPhotoUrl["fileUrl"].(string))
                    picTime = i.Timestamp
                }
                lable, lableAttach := Feature2Jsonstr(*face)
                var target = new(Target)
                target.TargetId = face.Id
                target.TargetScore = face.Score
                target.TargetType = face.Type
                target.Feature = face.Feature
                target.PicSmUrl = resp["fileUrl"].(string)
                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}}
                target.AttachTarget = SourceTarget{face.AttachArg.Id,face.AttachArg.Score,face.AttachArg.Type,face.AttachArg.Feature,"",Points{TopLeft: Point{face.AttachArg.Location.X, face.AttachArg.Location.Y}, BottomRight: Point{face.AttachArg.Location.X + face.AttachArg.Location.Width, face.AttachArg.Location.Y + face.AttachArg.Location.Height}}}
                var targetInfos []Target
                targetInfos = append(targetInfos, *target)

                sdkname := ""
                switch key {
                case "face":
                    sdkname = "人脸"
                case "plate":
                    sdkname = "车牌识别"
                case "track":
                    sdkname = "跟踪"
                }
                //logger.Info("人脸目标target:",targetInfos)
                if key == "track" && len(face.Liker) == 1{
                    logger.Info("确认过眼神是要追加的数据")
                    pervideo1 := LinkInfo{
                        face.Liker[0].TargetId,
                        msg.Cid,
                        msg.Push.Cam.Addr,
                        msg.Push.Cam.Name,
                        picTime,
                        imgMaxUrl,
                        msg.Tasklab.Taskid,
                        msg.Tasklab.Taskname,
                        sdkname,
                        "",
                        alarmRules,
                        time.Now().Format("2006-01-02 15:04:05"), // 只检测，没有比对时间
                        lable,
                        lableAttach,
                        "",
                        msg.Push.ServerId,
                        msg.Push.ServerName,
                        msg.Push.LocalIp,
                        "",
                        true,
                        false,
                        false,
                        false,
                        face.Liker,
                        targetInfos,
                    }
                    requstbody, err := json.Marshal(pervideo1)

                    if err != nil {
                        logger.Info("json parse error ", err)
                        return
                    }
                    esid := face.Liker[0].TargetId
                    logger.Info("liker信息：",*face.Liker[0])
                    returnMsg, err1 := esutil.AppendTargetInfo(esid, string(requstbody),structure.EsInfo.EsIndex.AIOcean.IndexName,serverIp,serverPort)
                    if err1 != nil {
                        logger.Error("追加数据出错！---", err1)
                    } else {
                        logger.Info("插入es返回的信息：", returnMsg)
                        // 发出录像信号
                        ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: face.Uuid, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, VideoUrl: msg.Push.VideoUrl, ImgId: i.Id, SdkIds: []string{"-1"}, Type: 1})
                    }
                if face.Uuid == "" {
                    // 比到了相似的人但在此摄像机下已经出现过，最后不能插入数据库
                    logger.Info("冗余数据，并不入库")
                } else {
                    id := face.Uuid
                    if id == "" {
                        // 比到了相似的人但在此摄像机下已经出现过，最后不能插入数据库
                        logger.Info("冗余数据，并不入库")
                    // 上传大图
                    // 解压缩并上传图片
                    bdata, err := util.UnCompress(msg.Data)
                    if err != nil {
                        panic("解压缩图片时出现错误")
                    }
                    alarmRules := []AlarmRule{}
                    //logger.Info("循环每一个目标")
                    for _, faceResult := range face.rules {
                        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)
                    // 先传小图，再传大图，防止脸上有线
                    bytes := util.SubImg(i, int(face.Location.X), int(face.Location.Y), int(face.Location.X+face.Location.Width), int(face.Location.Y+face.Location.Height),face.Type)
                    resp, err := util.PostFormBufferData1(weedfsUrl, bytes, uuid.NewV4().String())
                    if err != nil {
                        logger.Error("上传小图出错")
                    }
                    // 上传大图
                    if len(imgMaxUrl) == 0 {
                        bigPhotoUrl := make(map[string]interface{})
                        bigPhotoUrl, err = util.DrawPolygonOnImageForFace(msg.Cid, i, msg.RuleResult["face"].([]structure.Result), weedfsUrl)
                        logger.Debug("========大图路径：", bigPhotoUrl)
                        imgMaxUrl = append(imgMaxUrl, bigPhotoUrl["fileUrl"].(string))
                        picTime = i.Timestamp
                    }
                    lable, lableAttach := Feature2Jsonstr(*face)
                    var target = new(Target)
                    target.TargetId = face.Id
                    target.TargetScore = face.Score
                    target.TargetType = face.Type
                    target.Feature = face.Feature
                    target.PicSmUrl = resp["fileUrl"].(string)
                    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}}
                    target.AttachTarget = SourceTarget{face.AttachArg.Id,face.AttachArg.Score,face.AttachArg.Type,face.AttachArg.Feature,"",Points{TopLeft: Point{face.AttachArg.Location.X, face.AttachArg.Location.Y}, BottomRight: Point{face.AttachArg.Location.X + face.AttachArg.Location.Width, face.AttachArg.Location.Y + face.AttachArg.Location.Height}}}
                    var targetInfos []Target
                    targetInfos = append(targetInfos, *target)

                    sdkname := ""
                    switch key {
                    case "face":
                        sdkname = "人脸"
                    case "plate":
                        sdkname = "车牌识别"
                    case "track":
                        sdkname = "跟踪"
                    }
                    //logger.Info("人脸目标target:",targetInfos)
                    if key == "track" && len(face.Liker) == 1{
                        logger.Info("确认过眼神是要追加的数据")
                        pervideo1 := LinkInfo{
                            face.Liker[0].TargetId,
                            msg.Cid,
                            msg.Push.Cam.Addr,
                            msg.Push.Cam.Name,
                            picTime,
                            imgMaxUrl,
                            msg.Tasklab.Taskid,
                            msg.Tasklab.Taskname,
                            sdkname,
                            "",
                            alarmRules,
                            time.Now().Format("2006-01-02 15:04:05"), // 只检测，没有比对时间
                            lable,
                            lableAttach,
                            "",
                            msg.Push.ServerId,
                            msg.Push.ServerName,
                            msg.Push.LocalIp,
                            "",
                            true,
                            false,
                            false,
                            false,
                            face.Liker,
                            targetInfos,
                        }
                        requstbody, err := json.Marshal(pervideo1)

                        if err != nil {
                            logger.Info("json parse error ", err)
                            return
                        }
                        esid := face.Liker[0].TargetId
                        logger.Info("liker信息：",*face.Liker[0])
                        returnMsg, err1 := esutil.AppendTargetInfo(esid, string(requstbody),structure.EsInfo.EsIndex.AIOcean.IndexName,serverIp,serverPort)
                        if err1 != nil {
                            logger.Error("追加数据出错！---", err1)
                        } else {
                            logger.Info("插入es返回的信息：", returnMsg)
                            // 发出录像信号
                            ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: face.Uuid, CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, VideoUrl: msg.Push.VideoUrl, ImgId: i.Id, SdkIds: []string{"-1"}, Type: 1})
                        }
                    } else {
                        logger.Info("新增数据")
                        pervideo := PerVideoPicture{LinkInfo{
                            id,
                            face.Uuid,
                            msg.Cid,
                            msg.Push.Cam.Addr,
                            msg.Push.Cam.Name,
@@ -287,7 +286,6 @@
                }
            }
        }

}

// 归置人脸