派生自 ruleprocess
insertdata/insertDataToEs.go
@@ -4,9 +4,9 @@ "encoding/json" "errors" "fmt" "log" "net" "ruleprocess/cache" "ruleprocess/logger" "strings" "time" @@ -43,9 +43,9 @@ AnalyServerName string `json:"analyServerName"` AnalyServerIp string `json:"analyServerIp"` ClusterId string `json:"clusterId"` IsAlarm string `json:"isAlarm"` IsAckAlarm string `json:"isAckAlarm"` IsCollect string `json:"isCollect"` IsAlarm int `json:"isAlarm"` IsAckAlarm int `json:"isAckAlarm"` IsCollect int `json:"isCollect"` IsDelete int `json:"isDelete"` BaseInfo []Base `json:"baseInfo"` } @@ -82,9 +82,9 @@ PicSmUrl []string `json:"picSmUrl"` PicDate string `json:"picDate"` VideoUrl string `json:"videoUrl"` IsAlarm string `json:"isAlarm"` IsAckAlarm string `json:"isAckAlarm"` IsCollect string `json:"isCollect"` IsAlarm int `json:"isAlarm"` IsAckAlarm int `json:"isAckAlarm"` IsCollect int `json:"isCollect"` IsDelete int `json:"isDelete"` } @@ -99,7 +99,7 @@ // 直接从规则的标签数据里拿符合规则的人脸结果 if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.Arg)) > 0 { log.Println("往ES插人脸数据") logger.Info("往ES插人脸数据") for _, face := range msg.RuleResult["face"].([]ruleserver.Arg) { // 上传大图 // 解压缩并上传图片 @@ -116,7 +116,7 @@ // 人脸检测，没有相似的底库人员 localConfig, err := cache.GetServerInfo() if err != nil { log.Println("查询本机信息失败！") logger.Info("查询本机信息失败！") } serverIp, err := GetLocalIP() // 解压缩并上传图片 @@ -127,15 +127,15 @@ // 查询cameraName camera, err := cache.GetCameraById(msg.Cid) if err != nil { log.Println("查询摄像机信息失败") logger.Info("查询摄像机信息失败") } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) log.Println("-------------------------------------------看下宽和高",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 { log.Println("上传小图出错") logger.Info("上传小图出错") } sex := "" if face.ThftRes.Gender == 1 { @@ -170,28 +170,28 @@ localConfig.ServerName, serverIp, "", "", "", "", 1, 0, 0, 0, []Base{}, } requstbody, err := json.Marshal(pervideo) if err != nil { log.Println("json parse error ", err) logger.Info("json parse error ", err) return } err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) log.Println(err) logger.Info(err.Error()) } else { log.Println("跟底库的相似数据---------：", face.Liker) logger.Info("跟底库的相似数据---------：", face.Liker) } } } if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]ruleserver.Result)) > 0 { log.Println("往ES插yolo数据") logger.Info("往ES插yolo数据") var sdkNames string = "" alarmRules := []AlarmRule{} for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) { @@ -199,10 +199,10 @@ alarm := ChangeToString(yoloResult.AlarmLevel) alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText}) } isAlarm := "" isAlarm := 0 resp := make(map[string]interface{}) if len(alarmRules) > 0 { isAlarm = "1" isAlarm = 1 // 解压缩并上传图片 bdata, err := util.UnCompress(msg.Data) if err != nil { @@ -211,27 +211,27 @@ 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,msg.RuleResult["yolo"].([]ruleserver.Result)) if err != nil { log.Println("画框或上传图片服务器出错", err) logger.Info("画框或上传图片服务器出错", err) } else { log.Println("已报警并上传改帧图片到服务器") logger.Info("已报警并上传改帧图片到服务器") } } else { isAlarm = "0" isAlarm = 0 // 不是报警数据不存 return } // log.Println("图片上传返回值：", resp) // logger.Println("图片上传返回值：", resp) // 查询本机信息 localConfig, err := cache.GetServerInfo() if err != nil { log.Println("查询本机信息失败！") logger.Info("查询本机信息失败！") } // 查询cameraName camera, err := cache.GetCameraById(msg.Cid) if err != nil { log.Println("查询摄像机信息失败") logger.Info("查询摄像机信息失败") } serverIp, err := GetLocalIP() peraction := Personaction{ @@ -252,22 +252,22 @@ time.Now().Format("2006-01-02 15:04:05"), "", isAlarm, "", "", 0, 0, 0, } requstbody, err := json.Marshal(peraction) if err != nil { log.Println("json parse error ", err) logger.Info("json parse error ", err) return } err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) if err != nil { log.Println("往ES插入数据失败", err) logger.Info("往ES插入数据失败", err) } else { log.Println("__________________________________________往ES插入数据yolo成功") logger.Info("__________________________________________往ES插入yolo数据成功") } } } @@ -322,7 +322,7 @@ } //获取年龄描述 func getDescription (age int32) string{ func getDescription(age int32) string { ageInfo := "青年" if age > 0 && age < 7 { ageInfo = "童年" @@ -338,14 +338,14 @@ return ageInfo } func getRaceString (i int32) string{ func getRaceString(i int32) string { race := "" if i == 1 { race = "white" } else if i == 2{ race = "yellow" race = "白人" } else if i == 2 { race = "黄人" } else { race = "black" race = "黑人" } return race } }
