派生自 ruleprocess
insertdata/insertDataToEs.go
@@ -1,12 +1,13 @@ package insertdata import ( "ruleprocess/logger" "encoding/json" "errors" "fmt" "io/ioutil" "net" "ruleprocess/cache" "ruleprocess/logger" "strings" "time" @@ -15,9 +16,24 @@ "github.com/satori/go.uuid" "ruleprocess/ruleserver" "ruleprocess/util" "github.com/go-yaml/yaml" ) var weedfsUrl string type conf struct { PhotoUrl string `yaml:"photoUrl"` } func init() { data, err := ioutil.ReadFile("./config/conf.yml") if err != nil { fmt.Println("读取配置文件出错--",err) logger.Error("读取配置文件出错--",err) } c := conf{} //把yaml形式的字符串解析成struct类型 yaml.Unmarshal(data, &c) weedfsUrl = c.PhotoUrl } var weedfsUrl = "http://192.168.1.182:6333/submit" // 人脸的数据结构 type PerVideoPicture struct { Id string `json:"id"` @@ -84,6 +100,7 @@ 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) { @@ -102,7 +119,7 @@ // 人脸检测，没有相似的底库人员 localConfig, err := cache.GetServerInfo() if err != nil { logger.Info("查询本机信息失败！") logger.Error("查询本机信息失败！") } serverIp, err := GetLocalIP() // 解压缩并上传图片 @@ -113,16 +130,17 @@ // 查询cameraName camera, err := cache.GetCameraById(msg.Cid) if err != nil { logger.Info("查询摄像机信息失败") logger.Error("查询摄像机信息失败") } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) logger.Info("-------------------------------------------看下宽和高", i.Width, i.Height) //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("上传小图出错") logger.Error("上传小图出错") } logger.Info("================小图地址:",resp["fileUrl"].(string)) sex := "" if face.ThftRes.Gender == 1 { sex = "男" @@ -149,7 +167,7 @@ race, face.ThftRes.Smile, face.ThftRes.Beauty, "", string(face.Feature), []string{strings.Split(resp["fileUrl"].(string), "/")[1]}, "暂无集群", localConfig.ServerId, @@ -169,7 +187,10 @@ return } err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) err1 := EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) if err1 != nil { logger.Error("上传ES出错！---",err1) } //logger.Info(err.Error()) } else { // 人脸提取 @@ -186,16 +207,17 @@ // 查询cameraName camera, err := cache.GetCameraById(msg.Cid) if err != nil { logger.Info("查询摄像机信息失败") logger.Error("查询摄像机信息失败") } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) logger.Info("-------------------------------------------看下宽和高", i.Width, i.Height) //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("上传小图出错") logger.Error("上传小图出错") } logger.Info("================小图地址:",resp["fileUrl"].(string)) sex := "" if face.ThftRes.Gender == 1 { sex = "男" @@ -237,7 +259,7 @@ requstbody, err := json.Marshal(pervideo) if err != nil { logger.Info("json parse error ", err) logger.Error("json parse error ", err) return } @@ -272,7 +294,7 @@ //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.Info("画框或上传图片服务器出错", err) logger.Error("画框或上传图片服务器出错", err) } else { logger.Info("已报警并上传改帧图片到服务器") } @@ -285,12 +307,12 @@ // 查询本机信息 localConfig, err := cache.GetServerInfo() if err != nil { logger.Info("查询本机信息失败！") logger.Error("查询本机信息失败！") } // 查询cameraName camera, err := cache.GetCameraById(msg.Cid) if err != nil { logger.Info("查询摄像机信息失败") logger.Error("查询摄像机信息失败") } serverIp, err := GetLocalIP() peraction := Personaction{ @@ -324,7 +346,7 @@ } err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) if err != nil { logger.Info("往ES插入数据失败", err) logger.Error("往ES插入数据失败", err) } else { logger.Info("__________________________________________往ES插入yolo数据成功") }
