From b7e65bd6d482ce9d95bd423e64c80344938d21d5 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期五, 12 七月 2019 19:58:27 +0800 Subject: [PATCH] 改造定时器 --- insertdata/insertDataToEs.go | 168 +++++++++++++++++++++++++++---------------------------- 1 files changed, 83 insertions(+), 85 deletions(-) diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go index 10cafc5..99e8323 100644 --- a/insertdata/insertDataToEs.go +++ b/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("___________________________________________杩欐槸鏈塨aseinfo鐨�") 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鎻抷olo鏁版嵁") - // 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 - // // 瑙e帇缂╁苟涓婁紶鍥剧墖 - // bdata, err := util.UnCompress(msg.Data) - // if err != nil { - // panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒") - // } - // 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鎻抷olo鏁版嵁") + 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 + // 瑙e帇缂╁苟涓婁紶鍥剧墖 + bdata, err := util.UnCompress(msg.Data) + if err != nil { + panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒") + } + 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 -- Gitblit v1.8.0