派生自 ruleprocess
insertdata/insertDataToEs.go
@@ -101,7 +101,6 @@ func InsertToEs(msg ruleserver.ResultMsg) { // 直接从规则的标签数据里拿符合规则的人脸结果 //logger.Info("------------------------------------从配置文件读取的值：",weedfsUrl) if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.Arg)) > 0 { logger.Info("往ES插人脸数据") for _, face := range msg.RuleResult["face"].([]ruleserver.Arg) { @@ -135,7 +134,6 @@ } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) logger.Info("-------------------------------------------看下宽和高", int(face.Location.X),int(face.Location.Y),int(face.Location.X+face.Location.Width),int(face.Location.Y+face.Location.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 { @@ -196,6 +194,7 @@ //logger.Info(err.Error()) } else { // 人脸比对 logger.Warn("___________________________________________这是有baseinfo的") localConfig, err := cache.GetServerInfo() if err != nil { logger.Info("查询本机信息失败！") @@ -267,93 +266,92 @@ } err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) logger.Info("跟底库的相似数据---------：", face.Liker) fmt.Println("------------------------------------------哈哈哈哈，底库有人") logger.Info("------------------------------------------哈哈哈哈，底库有人") } } } //if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]ruleserver.Result)) > 0 { // logger.Info("往ES插yolo数据") // var sdkNames string = "" // alarmRules := []AlarmRule{} // for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) { // sdkNames = sdkNames + yoloResult.SdkName // alarm := ChangeToString(yoloResult.AlarmLevel) // alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText}) // } // isAlarm := 0 // resp := make(map[string]interface{}) // if len(alarmRules) > 0 { // isAlarm = 1 // // 解压缩并上传图片 // bdata, err := util.UnCompress(msg.Data) // if err != nil { // panic("解压缩图片时出现错误") // } // i := protomsg.Image{} // err = proto.Unmarshal(bdata, &i) // //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String()) // resp, err = util.DrawPolygonOnImage(msg.Cid, i,msg.RuleResult["yolo"].([]ruleserver.Result)) // if err != nil { // logger.Error("画框或上传图片服务器出错", err) // } else { // logger.Info("已报警并上传改帧图片到服务器") // } // } else { // isAlarm = 0 // // 不是报警数据不存 // return // } // // logger.Println("图片上传返回值：", resp) // // 查询本机信息 // localConfig, err := cache.GetServerInfo() // if err != nil { // logger.Error("查询本机信息失败！") // } // // 查询cameraName // camera, err := cache.GetCameraById(msg.Cid) // if err != nil { // logger.Error("查询摄像机信息失败") // } // serverIp, err := GetLocalIP() // peraction := Personaction{ // uuid.NewV4().String(), // msg.Cid, // camera.Name, // camera.Addr, // msg.Tasklab.Taskid, // msg.Tasklab.Taskname, // sdkNames, // "", // alarmRules, // localConfig.ServerId, // localConfig.ServerName, // serverIp, // "", // []string{strings.Split(resp["fileUrl"].(string), "/")[1]}, // time.Now().Format("2006-01-02 15:04:05"), // "", // isAlarm, // 0, // 0, // 0, // } // requstbody, err := json.Marshal(peraction) // // if err != nil { // logger.Info("json parse error ", err) // return // // } // err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) // if err != nil { // logger.Error("往ES插入数据失败", err) // } else { // logger.Info("__________________________________________往ES插入yolo数据成功") // } //} if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]ruleserver.Result)) > 0 { logger.Info("往ES插yolo数据") var sdkNames string = "" alarmRules := []AlarmRule{} for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) { sdkNames = sdkNames + yoloResult.SdkName alarm := ChangeToString(yoloResult.AlarmLevel) alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText}) } isAlarm := 0 resp := make(map[string]interface{}) if len(alarmRules) > 0 { isAlarm = 1 // 解压缩并上传图片 bdata, err := util.UnCompress(msg.Data) if err != nil { panic("解压缩图片时出现错误") } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String()) resp, err = util.DrawPolygonOnImage(msg.Cid, i,msg.RuleResult["yolo"].([]ruleserver.Result)) if err != nil { logger.Error("画框或上传图片服务器出错", err) } else { logger.Info("上传的图片信息：",resp) } } else { isAlarm = 0 // 不是报警数据不存 return } // logger.Println("图片上传返回值：", resp) // 查询本机信息 localConfig, err := cache.GetServerInfo() if err != nil { logger.Error("查询本机信息失败！") } // 查询cameraName camera, err := cache.GetCameraById(msg.Cid) if err != nil { logger.Error("查询摄像机信息失败") } serverIp, err := GetLocalIP() peraction := Personaction{ uuid.NewV4().String(), msg.Cid, camera.Name, camera.Addr, msg.Tasklab.Taskid, msg.Tasklab.Taskname, sdkNames, "", alarmRules, localConfig.ServerId, localConfig.ServerName, serverIp, "", []string{strings.Split(resp["fileUrl"].(string), "/")[1]}, time.Now().Format("2006-01-02 15:04:05"), "", isAlarm, 0, 0, 0, } requstbody, err := json.Marshal(peraction) if err != nil { logger.Info("json parse error ", err) return } err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) if err != nil { logger.Error("往ES插入数据失败", err) } else { logger.Warn("__________________________________________往ES插入yolo数据成功") //os.Exit(1) } } } // 获取本机ip
