From 5ae97eebd0dfcaad32e92f9df7b7f73caad922bf Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期四, 07 十一月 2019 18:20:43 +0800
Subject: [PATCH] merge module
---
insertdata/insertDataToEs.go | 46 +++++++---------------------------------------
1 files changed, 7 insertions(+), 39 deletions(-)
diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go
index 704741d..a38b48d 100644
--- a/insertdata/insertDataToEs.go
+++ b/insertdata/insertDataToEs.go
@@ -1,7 +1,6 @@
package insertdata
import (
- "basic.com/valib/logger.git"
"encoding/base64"
"encoding/json"
"errors"
@@ -16,6 +15,7 @@
"github.com/satori/go.uuid"
"ruleprocess/ruleserver"
"ruleprocess/util"
+ "ruleprocess/structure"
)
var weedfsUrl, videoPersonUrl, personAction string
@@ -127,30 +127,6 @@
rules []structure.Result
}
-// 寰�ES鎻掓暟鎹�
-//func InsertToEs(msg ruleserver.ResultMsg) {
-// var timeLabel string
-// // 鐩存帴浠庤鍒欑殑鏍囩鏁版嵁閲屾嬁绗﹀悎瑙勫垯鐨勪汉鑴哥粨鏋�
-// if msg.RuleResult["timeLabel"] != nil {
-// timeLabel = msg.RuleResult["timeLabel"].(string)
-// }
-// logger.Debug("鎻掑叆鏁版嵁鍓嶇湅鐪嬫姤璀︽爣蹇椾綅锛�", timeLabel)
-// if timeLabel == "01" { // 鏃犲畾鏃跺櫒鐘舵�佽鎻掑叆鐨勬姤璀︽暟鎹�
-// InsertFace(msg)
-// flag := ruleserver.BodyIsSame(msg.SdkMessage)
-// if !flag {
-// InsertYolo(msg)
-// }
-// }
-// if timeLabel == "10" { // 瀹氭椂鍣ㄧ姸鎬佽鎻掑叆鐨勯甯ф姤璀︽暟鎹�傝繛甯︾潃瀹氭椂鍣ㄥ紑鍚椂鐨勯偅甯�
-// InsertFace(msg)
-// InsertYolo(msg)
-// }
-// //if timeLabel == "12" { // 骞堕潪鎶ヨ鏁版嵁锛屽彧鏄姸鎬佹敼鍙樼殑鏁版嵁
-// // //ChangeStatusFace(msg)
-// // ChangeStatusYolo(msg)
-// //}
-//}
func InsertToEs(msg structure.ResultMsg) {
defer func() {
if err := recover(); err != nil {
@@ -186,7 +162,7 @@
panic("瑙ｅ帇缂╁浘鐗囨椂鍑虹幇閿欒")
}
alarmRules := []AlarmRule{}
- logger.Warn("浜鸿劯id涓猴細", face.Id, "浜鸿劯鐨勮鍒欓暱搴︿负锛�", len(face.rules))
+ //logger.Warn("浜鸿劯id涓猴細", face.Id, "浜鸿劯鐨勮鍒欓暱搴︿负锛�", len(face.rules))
//os.Exit(1)
for _, faceResult := range face.rules {
alarm := ChangeToString(faceResult.DefenceState, faceResult.AlarmLevel)
@@ -228,7 +204,6 @@
target.TargetScore = face.Score
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}}
//logger.Info("浜鸿劯鐨刬d:",strconv.FormatUint(face.Id, 10))
- logger.Info("%%%%%%%%%%%%%%%%%%%%%鐩爣淇℃伅锛�",*target)
pervideo := PerVideoPicture{
msg.Push.PushId,
msg.Cid,
@@ -296,12 +271,12 @@
func hebingFace(faces []*FaceAndRules, faceResult structure.FaceResult) []*FaceAndRules {
for _, arg := range faceResult.Args {
// 鎷垮埌姣忎竴寮犱汉鑴�
- logger.Info("褰掔疆浜鸿劯鏃剁浉浼艰�呯殑鏁伴噺锛�", len(arg.Liker))
+ //logger.Info("褰掔疆浜鸿劯鏃剁浉浼艰�呯殑鏁伴噺锛�", len(arg.Liker))
flag := false
for _, face := range faces {
- for _, lik := range face.Liker {
- logger.Warn("--------鍚堝苟浜鸿劯鏃剁浉浼艰�咃細", lik.PersonId, lik.TableName)
- }
+ //for _, lik := range face.Liker {
+ // //logger.Warn("--------鍚堝苟浜鸿劯鏃剁浉浼艰�咃細", lik.PersonId, lik.TableName)
+ //}
if arg.Id == face.Id {
flag = true
face.rules = append(face.rules, faceResult.Result)
@@ -368,12 +343,6 @@
}
i := protomsg.Image{}
err = proto.Unmarshal(bdata, &i)
- if msg1.RuleResult != nil {
- logger.Debug("鎸佺画鏃堕棿绗竴寮犵敾妗嗭細")
- for _,res := range msg1.RuleResult["yolo"].([]structure.Result) {
- logger.Debug("鎸佺画鏃堕棿绗竴寮犺鍒欙細",res)
- }
- }
resp1, err1 := util.DrawPolygonOnImageForYolo(msg1.Cid, i, msg1.RuleResult["yolo"].([]structure.Result), weedfsUrl)
if err1 != nil {
logger.Error("缂撳瓨鏁版嵁鐢绘鎴栦笂浼犲浘鐗囨湇鍔″櫒鍑洪敊", err)
@@ -388,7 +357,7 @@
// 瑁呴厤鐩爣淇℃伅鏁版嵁
for _,target := range yoloResult.Location {
// 鍘婚噸娣诲姞
- logger.Info("瑁呴厤鍓嶇殑鍘熷鏁版嵁鏄細",target)
+ //logger.Info("瑁呴厤鍓嶇殑鍘熷鏁版嵁鏄細",target)
var flag = true
for _,selectTarget := range targetInfos {
if strconv.FormatUint(target.TargetId, 10) == selectTarget.TargetId {
@@ -448,7 +417,6 @@
// 涓嶆槸鎶ヨ鏁版嵁涓嶅瓨
return
}
-
if resp["fileUrl"] != nil {
url = append(url, resp["fileUrl"].(string))
//esDataId := uuid.NewV4().String()
--
Gitblit v1.8.0