派生自 ruleprocess
insertdata/insertDataToEs.go
@@ -2,18 +2,19 @@ import ( "errors" "fmt" "encoding/json" "log" "net" "time" "basic.com/pubsub/protomsg.git" "basic.com/dbapi.git" "github.com/golang/protobuf/proto" "ruleprocess/ruleserver" "ruleprocess/util" "github.com/golang/protobuf/proto" "github.com/satori/go.uuid" ) var weedfsUrl = "http://192.168.1.182:6333/submit" // 人脸的数据结构 type PerVideoPicture struct { Id string `json:"id"` @@ -86,138 +87,159 @@ type AlarmRule struct { GroupId string `json:"groupId"` AlarmLevel int32 `json:"alarmLevel"` AlarmLevel int32 `json:"alarmLevel"` RuleText string `json:"ruleText"` } // 往ES插数据 func InsertToEs(msg ruleserver.ResultMsg) { fmt.Println("往ES插数据") log.Println("往ES插数据") for _, sdkinfo := range msg.Tasklab.Sdkinfos { if sdkinfo.Sdktype == "FaceDetect" { faceParam := protomsg.ParamFacePos{} err1 := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam) if err1 != nil { fmt.Println("解析FACE sdk有误", err1) if len(sdkinfo.Sdkdata) > 1 { faceParam := protomsg.ParamFacePos{} err1 := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam) if err1 != nil { log.Println("解析FACE sdk有误", err1) continue } for _, face := range faceParam.Faces { pervideo := PerVideoPicture{ uuid.NewV4().String(), msg.Cid, msg.Caddr, time.Now().Format("2006-01-02 15:04:05"), "", msg.Tasklab.Taskid, msg.Tasklab.Taskname, sdkinfo.SdkName, "", "", face.Result.Gender, face.Result.Age, "", face.Result.Race, face.Result.Smile, face.Result.Beauty, "不是每个人脸算法都有", "---", "", "", "", "", "", "", "", "", 0, Base{ "是每个人脸算法都有吗", "", 0, "", "", "", "", "", "", "", "", }, } requstbody, err := json.Marshal(pervideo) if err != nil { log.Println("json parse error ", err) return } err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) if err != nil { log.Println("es can not execute right.") } } } else { continue } for _, face := range faceParam.Faces { pervideo := PerVideoPicture{ } if sdkinfo.Sdktype == "Yolo" { if len(sdkinfo.Sdkdata) > 1 { yoloObj := protomsg.ParamYoloObj{} err1 := proto.Unmarshal(sdkinfo.Sdkdata, &yoloObj) if err1 != nil { log.Println("解析YOLO sdk有误", err1) //continue } alarmRules := []AlarmRule{} for _,result := range msg.RuleResult { alarmRules = append(alarmRules,AlarmRule{result.RuleGroupId,result.AlarmLevel,result.RuleText}) } isAlarm := "" resp := make(map[string]interface{}) if len(alarmRules) > 0 { isAlarm = "1" // 解压缩并上传图片 bdata, err := util.UnCompress(msg.Data) if err != nil { panic("解压缩图片时出现错误") } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) resp,err = util.PostFormBufferData(weedfsUrl,i.Data,uuid.NewV4().String()) log.Println("已报警并上传改帧图片到服务器") }else { isAlarm = "0" //continue } log.Println("图片上传返回值：",resp) // 查询本机信息 flag,localConfig := dbapi.SysSetApi{}.GetServerInfo() if !flag { log.Println("查询本机信息失败！") } // 查询cameraName camera,err := dbapi.CameraApi{}.GetCameraById(msg.Cid) if err == nil { log.Println("查询摄像机信息失败") } serverIp,err := GetLocalIP() peraction := Personaction{ uuid.NewV4().String(), msg.Cid, camera.Name, msg.Caddr, time.Now().Format("2006-01-02 15:04:05"), "", msg.Tasklab.Taskid, msg.Tasklab.Taskname, sdkinfo.SdkName, "", alarmRules, localConfig.ServerId, localConfig.ServerName, serverIp, "", face.Result.Gender, face.Result.Age, resp["fileUrl"].(string), time.Now().Format("2006-01-02 15:04:05"), "", face.Result.Race, face.Result.Smile, face.Result.Beauty, "不是每个人脸算法都有", "---", "", "", "", "", "", "", isAlarm, "", "", 0, Base{ "是每个人脸算法都有吗", "", 0, "", "", "", "", "", "", "", "", }, } requstbody, err := json.Marshal(pervideo) requstbody, err := json.Marshal(peraction) if err != nil { fmt.Println("json parse error ", err) log.Println("json parse error ", err) return } err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody) err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) if err != nil { fmt.Println("es can not execute right.") log.Println("es can not execute right.") } } } if sdkinfo.Sdktype == "Yolo" { yoloObj := protomsg.ParamYoloObj{} err1 := proto.Unmarshal(sdkinfo.Sdkdata, &yoloObj) if err1 != nil { fmt.Println("解析YOLO sdk有误", err1) //continue } alarmRules := []AlarmRule{} for _,result := range msg.RuleResult { alarmRules = append(alarmRules,AlarmRule{result.RuleGroupId,result.AlarmLevel,result.RuleText}) } isAlarm := "" if len(alarmRules) > 0 { isAlarm = "1" }else { isAlarm = "0" } else { continue } // 查询本机信息 flag,localConfig := dbapi.SysSetApi{}.GetServerInfo() if !flag { fmt.Println("查询本机信息失败！") } // 查询cameraName camera,err := dbapi.CameraApi{}.GetCameraById(msg.Cid) if err == nil { fmt.Println("查询摄像机信息失败") } serverIp,err := GetLocalIP() peraction := Personaction{ uuid.NewV4().String(), msg.Cid, camera.Name, msg.Caddr, msg.Tasklab.Taskid, msg.Tasklab.Taskname, sdkinfo.SdkName, "", alarmRules, localConfig.ServerId, localConfig.ServerName, serverIp, "", "", time.Now().Format("2006-01-02 15:04:05"), "", isAlarm, "", "", 0, } requstbody, err := json.Marshal(peraction) if err != nil { fmt.Println("json parse error ", err) return } err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody) if err != nil { fmt.Println("es can not execute right.") } }
