派生自 ruleprocess
insertdata/insertDataToEs.go
@@ -20,34 +20,34 @@ var weedfsUrl = "http://192.168.1.182:6333/submit" // 人脸的数据结构 type PerVideoPicture struct { Id string `json:"id"` CameraId string `json:"cameraId"` CameraAddr string `json:"cameraAddr"` PicDate string `json:"picDate"` PicMaxUrl string `json:"picMaxUrl"` TaskId string `json:"taskId"` TaskName string `json:"taskName"` SdkName string `json:"sdkName"` Content string `json:"content"` LikeDate string `json:"likeDate"` Sex int32 `json:"sex"` Age int32 `json:"age"` AgeDescription string `json:"ageDescription"` Race int32 `json:"race"` SmileLevel int32 `json:"smileLevel"` BeautyLevel int32 `json:"beautyLevel"` FaceFeature string `json:"faceFeature"` PicSmUrl string `json:"picSmUrl"` VideoUrl string `json:"videoUrl"` AnalyServerId string `json:"analyServerId"` AnalyServerName string `json:"analyServerName"` AnalyServerIp string `json:"analyServerIp"` ClusterId string `json:"clusterId"` IsAlarm string `json:"isAlarm"` IsAckAlarm string `json:"isAckAlarm"` IsCollect string `json:"isCollect"` IsDelete int `json:"isDelete"` BaseInfo Base `json:"baseInfo"` Id string `json:"id"` CameraId string `json:"cameraId"` CameraAddr string `json:"cameraAddr"` PicDate string `json:"picDate"` PicMaxUrl string `json:"picMaxUrl"` TaskId string `json:"taskId"` TaskName string `json:"taskName"` SdkName string `json:"sdkName"` Content string `json:"content"` LikeDate string `json:"likeDate"` Sex int32 `json:"sex"` Age int32 `json:"age"` AgeDescription string `json:"ageDescription"` Race int32 `json:"race"` SmileLevel int32 `json:"smileLevel"` BeautyLevel int32 `json:"beautyLevel"` FaceFeature string `json:"faceFeature"` PicSmUrl []string `json:"picSmUrl"` VideoUrl string `json:"videoUrl"` AnalyServerId string `json:"analyServerId"` AnalyServerName string `json:"analyServerName"` AnalyServerIp string `json:"analyServerIp"` ClusterId string `json:"clusterId"` IsAlarm string `json:"isAlarm"` IsAckAlarm string `json:"isAckAlarm"` IsCollect string `json:"isCollect"` IsDelete int `json:"isDelete"` BaseInfo []Base `json:"baseInfo"` } type Base struct { @@ -90,7 +90,7 @@ type AlarmRule struct { GroupId string `json:"groupId"` AlarmLevel `json:"alarmLevel"` AlarmLevel string `json:"alarmLevel"` RuleText string `json:"ruleText"` } @@ -98,9 +98,9 @@ func InsertToEs(msg ruleserver.ResultMsg) { // 直接从规则的标签数据里拿符合规则的人脸结果 if msg.RuleResult["faces"] != nil && len(msg.RuleResult["faces"].([]ruleserver.Arg)) > 0 { if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.Arg)) > 0 { log.Println("往ES插人脸数据") for _, face := range msg.RuleResult["faces"].([]ruleserver.Arg) { for _, face := range msg.RuleResult["face"].([]ruleserver.Arg) { // 上传大图 // 解压缩并上传图片 bdata, err := util.UnCompress(msg.Data) @@ -126,9 +126,8 @@ } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) i1 := protomsg.Image{} i1 = util.Subimg(i.Data, 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.PostFormBufferData(weedfsUrl, i1, uuid.NewV4().String()) 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 { log.Println("上传小图出错") } @@ -137,12 +136,12 @@ msg.Cid, msg.Caddr, time.Now().Format("2006-01-02 15:04:05"), bigPhotoUrl["fileUrl"].(string), strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1], msg.Tasklab.Taskid, msg.Tasklab.Taskname, face.SdkName, "人脸", "", "", // 只检测，没有比对时间 time.Now().Format("2006-01-02 15:04:05"), // 只检测，没有比对时间 face.ThftRes.Gender, face.ThftRes.Age, "", @@ -150,7 +149,7 @@ face.ThftRes.Smile, face.ThftRes.Beauty, "不是每个人脸算法都有", strings.Split(resp["fileUrl"].(string), "/")[1], []string{strings.Split(resp["fileUrl"].(string), "/")[1]}, "暂无集群", localConfig.ServerId, localConfig.ServerName, @@ -160,7 +159,7 @@ "", "", 0, Base{}, []Base{}, } requstbody, err := json.Marshal(pervideo) @@ -170,13 +169,9 @@ } err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) if err != nil { log.Println("es can not execute right.") } else { log.Println("___________________________________往ES插入人脸数据成功") } }else { log.Println("跟底库的相似数据---------：",face.Liker) log.Println(err) } else { log.Println("跟底库的相似数据---------：", face.Liker) } } } @@ -201,10 +196,10 @@ i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String()) resp, err =util.DrawPolygonOnImage(msg.Cid,i) resp, err = util.DrawPolygonOnImage(msg.Cid, i) if err != nil { log.Println("画框或上传图片服务器出错",err) }else{ log.Println("画框或上传图片服务器出错", err) } else { log.Println("已报警并上传改帧图片到服务器") } } else { @@ -215,7 +210,7 @@ // log.Println("图片上传返回值：", resp) // 查询本机信息 localConfig, err := cache.GetServerInfo() if err !=nil { if err != nil { log.Println("查询本机信息失败！") } // 查询cameraName @@ -255,7 +250,7 @@ } err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) if err != nil { log.Println("往ES插入数据失败",err) log.Println("往ES插入数据失败", err) } else { log.Println("__________________________________________往ES插入数据yolo成功") } @@ -291,7 +286,7 @@ } // 把报警等级转化成汉字 func ChangeToString(i int32) string{ func ChangeToString(i int32) string { alarm := "" if i == 1 { alarm = "一级" @@ -309,4 +304,4 @@ alarm = "五级" } return alarm } }
