From ca177f11067d4886e615a2179e943ed6acc687b4 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期一, 15 七月 2019 10:54:07 +0800 Subject: [PATCH] 把es插入路径换成配置文件里的 --- insertdata/insertDataToEs.go | 145 +++++++++++++++++++++++++----------------------- config/conf.yml | 4 + 2 files changed, 78 insertions(+), 71 deletions(-) diff --git a/config/conf.yml b/config/conf.yml index 2bb51ae..1e2c4df 100644 --- a/config/conf.yml +++ b/config/conf.yml @@ -1 +1,3 @@ -photoUrl: "http://192.168.1.182:6333/submit" \ No newline at end of file +photoUrl: "http://192.168.1.182:6333/submit" +videopersons: "http://192.168.1.182:9200/videopersons/perVideoPicture" +personaction: "http://192.168.1.182:9200/personaction/perVideoAction" \ No newline at end of file diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go index f967650..60fea44 100644 --- a/insertdata/insertDataToEs.go +++ b/insertdata/insertDataToEs.go @@ -19,10 +19,12 @@ "ruleprocess/util" ) -var weedfsUrl string +var weedfsUrl,videoPersonUrl,personAction string type conf struct { - PhotoUrl string `yaml:"photoUrl"` + PhotoUrl string `yaml:"photoUrl"` + Videopersons string `yaml:"videopersons"` + Personaction string `yaml:"personaction"` } func init() { @@ -35,6 +37,8 @@ //鎶妝aml褰㈠紡鐨勫瓧绗︿覆瑙f瀽鎴恠truct绫诲瀷 yaml.Unmarshal(data, &c) weedfsUrl = c.PhotoUrl + videoPersonUrl = c.Videopersons + personAction = c.Personaction } // 浜鸿劯鐨勬暟鎹粨鏋� @@ -202,7 +206,7 @@ logger.Info("json parse error ", err) return } - err1 := EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) + err1 := EsReq("POST", videoPersonUrl, requstbody) if err1 != nil { logger.Error("涓婁紶ES鍑洪敊锛�---", err1) } @@ -211,73 +215,73 @@ } func ChangeStatusFace(msg ruleserver.ResultMsg) { logger.Info("寰�ES鎻掑叆浜鸿劯闈炴姤璀︿絾鏄姸鎬佽浆鎹㈡暟鎹�") - // 涓婁紶澶у浘 - // 瑙e帇缂╁苟涓婁紶鍥剧墖 - bdata, err := util.UnCompress(msg.Data) - if err != nil { - panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒") - } - i := protomsg.Image{} - err = proto.Unmarshal(bdata, &i) - bigPhotoUrl := make(map[string]interface{}) - bigPhotoUrl, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String()) - logger.Debug("========澶у浘璺緞锛�", bigPhotoUrl) - // 浜鸿劯妫�娴嬶紝娌℃湁鐩镐技鐨勫簳搴撲汉鍛� - localConfig, err := cache.GetServerInfo() - if err != nil { - logger.Error("鏌ヨ鏈満淇℃伅澶辫触锛�") - } - serverIp, err := GetLocalIP() - // 鏌ヨcameraName - camera, err := cache.GetCameraById(msg.Cid) - if err != nil { - logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭け璐�") - } + // 涓婁紶澶у浘 + // 瑙e帇缂╁苟涓婁紶鍥剧墖 + bdata, err := util.UnCompress(msg.Data) + if err != nil { + panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒") + } + i := protomsg.Image{} + err = proto.Unmarshal(bdata, &i) + bigPhotoUrl := make(map[string]interface{}) + bigPhotoUrl, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String()) + logger.Debug("========澶у浘璺緞锛�", bigPhotoUrl) + // 浜鸿劯妫�娴嬶紝娌℃湁鐩镐技鐨勫簳搴撲汉鍛� + localConfig, err := cache.GetServerInfo() + if err != nil { + logger.Error("鏌ヨ鏈満淇℃伅澶辫触锛�") + } + serverIp, err := GetLocalIP() + // 鏌ヨcameraName + camera, err := cache.GetCameraById(msg.Cid) + if err != nil { + logger.Error("鏌ヨ鎽勫儚鏈轰俊鎭け璐�") + } - pervideo := PerVideoPicture{ - uuid.NewV4().String(), - msg.Cid, - camera.Addr, - i.Timestamp, - strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1], - msg.Tasklab.Taskid, - msg.Tasklab.Taskname, - "浜鸿劯", - "鐘舵�佽浆鎹㈡暟鎹紝闈炴姤璀︽暟鎹�", - time.Now().Format("2006-01-02 15:04:05"), // 鍙娴嬶紝娌℃湁姣斿鏃堕棿 - "", - 0, - "", - "", - 0, - 0, - "", - []string{""}, - "鏆傛棤闆嗙兢", - localConfig.ServerId, - localConfig.ServerName, - serverIp, - "", - 0, - 1, - 0, - 0, - 0, - []*protomsg.Baseinfo{}, - } - requstbody, err := json.Marshal(pervideo) + pervideo := PerVideoPicture{ + uuid.NewV4().String(), + msg.Cid, + camera.Addr, + i.Timestamp, + strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1], + msg.Tasklab.Taskid, + msg.Tasklab.Taskname, + "浜鸿劯", + "鐘舵�佽浆鎹㈡暟鎹紝闈炴姤璀︽暟鎹�", + time.Now().Format("2006-01-02 15:04:05"), // 鍙娴嬶紝娌℃湁姣斿鏃堕棿 + "", + 0, + "", + "", + 0, + 0, + "", + []string{""}, + "鏆傛棤闆嗙兢", + localConfig.ServerId, + localConfig.ServerName, + serverIp, + "", + 0, + 1, + 0, + 0, + 0, + []*protomsg.Baseinfo{}, + } + requstbody, err := json.Marshal(pervideo) - if err != nil { - logger.Info("json parse error ", err) - return - } - err1 := EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) - if err1 != nil { - logger.Error("涓婁紶ES鍑洪敊锛�---", err1) - } - //if msg.RuleResult["cacheData"] != nil { - // InsertFace(msg.RuleResult["cacheData"].(ruleserver.ResultMsg)) - //} + if err != nil { + logger.Info("json parse error ", err) + return + } + err1 := EsReq("POST", videoPersonUrl, requstbody) + if err1 != nil { + logger.Error("涓婁紶ES鍑洪敊锛�---", err1) + } + //if msg.RuleResult["cacheData"] != nil { + // InsertFace(msg.RuleResult["cacheData"].(ruleserver.ResultMsg)) + //} } // 寰�es涓彃鍏olo鏁版嵁 @@ -355,7 +359,7 @@ return } - err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) + err = EsReq("POST", personAction, requstbody) if err != nil { logger.Error("寰�ES鎻掑叆鏁版嵁澶辫触", err) } else { @@ -425,7 +429,7 @@ return } - err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) + err = EsReq("POST", personAction, requstbody) if err != nil { logger.Error("寰�ES鎻掑叆鏁版嵁澶辫触", err) } else { @@ -433,6 +437,7 @@ //os.Exit(1) } } + // 鑾峰彇鏈満ip func GetLocalIP() (ipv4 string, err error) { var ( -- Gitblit v1.8.0