From 6cc33b9bcbf1db10bb1ca3683d6fe80af1abf6b8 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期五, 05 七月 2019 13:48:26 +0800 Subject: [PATCH] 人脸提取的上传 --- insertdata/insertDataToEs.go | 96 +++++++++++++++++++++++++++++++++++++++--------- 1 files changed, 78 insertions(+), 18 deletions(-) diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go index 4fe05b4..026925f 100644 --- a/insertdata/insertDataToEs.go +++ b/insertdata/insertDataToEs.go @@ -3,6 +3,7 @@ import ( "encoding/json" "errors" + "fmt" "net" "ruleprocess/cache" "ruleprocess/logger" @@ -46,21 +47,7 @@ IsAckAlarm int `json:"isAckAlarm"` IsCollect int `json:"isCollect"` IsDelete int `json:"isDelete"` - BaseInfo []Base `json:"baseInfo"` -} - -type Base struct { - TableId string `json:"tableId"` - TableName string `json:"tableName"` - CompareScore float64 `json:"compareScore"` - PersonId string `json:"personId"` - PersonName string `json:"personName"` - PersonPicUrl string `json:"personPicUrl"` - PhoneNum string `json:"phoneNum"` - Sex string `json:"sex"` - IdCard string `json:"idCard"` - MonitorLevel string `json:"monitorLevel"` - Content string `json:"content"` + BaseInfo []*protomsg.Baseinfo `json:"baseInfo"` } // yolo琛屼负鐨勬暟鎹粨鏋� @@ -162,7 +149,7 @@ race, face.ThftRes.Smile, face.ThftRes.Beauty, - "涓嶆槸姣忎釜浜鸿劯绠楁硶閮芥湁", + "", []string{strings.Split(resp["fileUrl"].(string), "/")[1]}, "鏆傛棤闆嗙兢", localConfig.ServerId, @@ -173,7 +160,7 @@ 0, 0, 0, - []Base{}, + []*protomsg.Baseinfo{}, } requstbody, err := json.Marshal(pervideo) @@ -183,9 +170,82 @@ } err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) - logger.Info(err.Error()) + //logger.Info(err.Error()) } else { + // 浜鸿劯鎻愬彇 + localConfig, err := cache.GetServerInfo() + if err != nil { + logger.Info("鏌ヨ鏈満淇℃伅澶辫触锛�") + } + serverIp, err := GetLocalIP() + // 瑙e帇缂╁苟涓婁紶鍥剧墖 + bdata, err := util.UnCompress(msg.Data) + if err != nil { + panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒") + } + // 鏌ヨcameraName + camera, err := cache.GetCameraById(msg.Cid) + if err != nil { + logger.Info("鏌ヨ鎽勫儚鏈轰俊鎭け璐�") + } + i := protomsg.Image{} + err = proto.Unmarshal(bdata, &i) + 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 { + logger.Info("涓婁紶灏忓浘鍑洪敊") + } + sex := "" + if face.ThftRes.Gender == 1 { + sex = "鐢�" + } else { + sex = "濂�" + } + race := getRaceString(face.ThftRes.Race) + ageDescription := getDescription(face.ThftRes.Age) + pervideo := PerVideoPicture{ + uuid.NewV4().String(), + msg.Cid, + camera.Addr, + time.Now().Format("2006-01-02 15:04:05"), + strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1], + msg.Tasklab.Taskid, + msg.Tasklab.Taskname, + "浜鸿劯", + "", + time.Now().Format("2006-01-02 15:04:05"), // 鍙娴嬶紝娌℃湁姣斿鏃堕棿 + sex, + face.ThftRes.Age, + ageDescription, + race, + face.ThftRes.Smile, + face.ThftRes.Beauty, + "", + []string{strings.Split(resp["fileUrl"].(string), "/")[1]}, + "鏆傛棤闆嗙兢", + localConfig.ServerId, + localConfig.ServerName, + serverIp, + "", + 1, + 0, + 0, + 0, + face.Liker, + } + requstbody, err := json.Marshal(pervideo) + + if err != nil { + logger.Info("json parse error ", err) + return + + } + err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) logger.Info("璺熷簳搴撶殑鐩镐技鏁版嵁---------锛�", face.Liker) + + fmt.Println("------------------------------------------鍝堝搱鍝堝搱锛屽簳搴撴湁浜�") + } } } -- Gitblit v1.8.0