| | |
| | | "encoding/json" |
| | | "errors" |
| | | "fmt" |
| | | "log" |
| | | "net" |
| | | "ruleprocess/cache" |
| | | "ruleprocess/logger" |
| | | "strings" |
| | | "time" |
| | | |
| | |
| | | |
| | | // 直接从规则的标签数据里拿符合规则的人脸结果 |
| | | if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.Arg)) > 0 { |
| | | log.Println("往ES插人脸数据") |
| | | logger.Info("往ES插人脸数据") |
| | | for _, face := range msg.RuleResult["face"].([]ruleserver.Arg) { |
| | | // 上传大图 |
| | | // 解压缩并上传图片 |
| | |
| | | // 人脸检测,没有相似的底库人员 |
| | | localConfig, err := cache.GetServerInfo() |
| | | if err != nil { |
| | | log.Println("查询本机信息失败!") |
| | | logger.Info("查询本机信息失败!") |
| | | } |
| | | serverIp, err := GetLocalIP() |
| | | // 解压缩并上传图片 |
| | |
| | | // 查询cameraName |
| | | camera, err := cache.GetCameraById(msg.Cid) |
| | | if err != nil { |
| | | log.Println("查询摄像机信息失败") |
| | | logger.Info("查询摄像机信息失败") |
| | | } |
| | | i := protomsg.Image{} |
| | | err = proto.Unmarshal(bdata, &i) |
| | | log.Println("-------------------------------------------看下宽和高", i.Width, i.Height) |
| | | logger.Info("-------------------------------------------看下宽和高", i.Width, i.Height) |
| | | 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)) |
| | | resp, err := util.PostFormBufferData1(weedfsUrl, bytes, uuid.NewV4().String()) |
| | | if err != nil { |
| | | log.Println("上传小图出错") |
| | | logger.Info("上传小图出错") |
| | | } |
| | | sex := "" |
| | | if face.ThftRes.Gender == 1 { |
| | |
| | | requstbody, err := json.Marshal(pervideo) |
| | | |
| | | if err != nil { |
| | | log.Println("json parse error ", err) |
| | | logger.Info("json parse error ", err) |
| | | return |
| | | |
| | | } |
| | | err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) |
| | | log.Println(err) |
| | | logger.Info(err.Error()) |
| | | } else { |
| | | log.Println("跟底库的相似数据---------:", face.Liker) |
| | | logger.Info("跟底库的相似数据---------:", face.Liker) |
| | | } |
| | | } |
| | | } |
| | | if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]ruleserver.Result)) > 0 { |
| | | log.Println("往ES插yolo数据") |
| | | logger.Info("往ES插yolo数据") |
| | | var sdkNames string = "" |
| | | alarmRules := []AlarmRule{} |
| | | for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) { |
| | |
| | | //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String()) |
| | | resp, err = util.DrawPolygonOnImage(msg.Cid, i,msg.RuleResult["yolo"].([]ruleserver.Result)) |
| | | if err != nil { |
| | | log.Println("画框或上传图片服务器出错", err) |
| | | logger.Info("画框或上传图片服务器出错", err) |
| | | } else { |
| | | log.Println("已报警并上传改帧图片到服务器") |
| | | logger.Info("已报警并上传改帧图片到服务器") |
| | | } |
| | | } else { |
| | | isAlarm = 0 |
| | | // 不是报警数据不存 |
| | | return |
| | | } |
| | | // log.Println("图片上传返回值:", resp) |
| | | // logger.Println("图片上传返回值:", resp) |
| | | // 查询本机信息 |
| | | localConfig, err := cache.GetServerInfo() |
| | | if err != nil { |
| | | log.Println("查询本机信息失败!") |
| | | logger.Info("查询本机信息失败!") |
| | | } |
| | | // 查询cameraName |
| | | camera, err := cache.GetCameraById(msg.Cid) |
| | | if err != nil { |
| | | log.Println("查询摄像机信息失败") |
| | | logger.Info("查询摄像机信息失败") |
| | | } |
| | | serverIp, err := GetLocalIP() |
| | | peraction := Personaction{ |
| | |
| | | requstbody, err := json.Marshal(peraction) |
| | | |
| | | if err != nil { |
| | | log.Println("json parse error ", err) |
| | | logger.Info("json parse error ", err) |
| | | return |
| | | |
| | | } |
| | | err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) |
| | | if err != nil { |
| | | log.Println("往ES插入数据失败", err) |
| | | logger.Info("往ES插入数据失败", err) |
| | | } else { |
| | | log.Println("__________________________________________往ES插入yolo数据成功") |
| | | logger.Info("__________________________________________往ES插入yolo数据成功") |
| | | } |
| | | } |
| | | } |